fixed RegisterConnectHandler function

99fb481c · suyanlong · 5eefc718 · 99fb481c
Commit 99fb481c authored Dec 21, 2021 by suyanlong
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 5 deletions

handler.go internal/peermgr/handler.go +5 -5

No files found.
--- a/internal/peermgr/handler.go
+++ b/internal/peermgr/handler.go
@@ -42,9 +42,9 @@ func (swarm *Swarm) registerMsgHandler(messageType pb.Message_Type, handler inte
 }

 func (swarm *Swarm) RegisterConnectHandler(handler ConnectHandler) error {
-	//swarm.lock.Lock()
-	//defer swarm.lock.Unlock()
-	//swarm.connectHandlers = append(swarm.connectHandlers, handler)
-	//return nil
-	panic("Unimplemented!")
+	swarm.lock.Lock()
+	defer swarm.lock.Unlock()
+	swarm.connectHandlers = append(swarm.connectHandlers, handler)
+	return nil
+	//panic("Unimplemented!")
 }