swarm/network: Wake the hive when a peer is registered

Signed-off-by: Lewis Marshall <lewis@lmars.net>
This commit is contained in:
Lewis Marshall 2017-05-17 21:56:30 -07:00
parent 0b28cf8544
commit 0534fa4f97

View file

@ -167,7 +167,7 @@ func (self *Hive) Stop() {
}
func (self *Hive) Run(p *bzzPeer) error {
dp := NewDiscovery(p, self.Overlay)
dp := NewDiscovery(p, self)
log.Debug(fmt.Sprintf("to add new bee %v", p))
self.On(dp)
self.wake()
@ -199,6 +199,11 @@ func (self *Hive) Healthy() bool {
return true
}
func (self *Hive) Register(peers chan OverlayAddr) error {
defer self.wake()
return self.Overlay.Register(peers)
}
// wake triggers
func (self *Hive) wake() {
select {