Skip to content

Commit a6ab90f

Browse files
committed
lnd: make sure we track goroutine and wait for it to be finished
1 parent 39369e3 commit a6ab90f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

server.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5192,7 +5192,12 @@ func (s *server) DisconnectPeer(pubKey *btcec.PublicKey) error {
51925192
//
51935193
// NOTE: We call it in a goroutine to avoid blocking the main server
51945194
// goroutine because we might hold the server's mutex.
5195-
go peer.Disconnect(fmt.Errorf("server: DisconnectPeer called"))
5195+
s.wg.Add(1)
5196+
go func() {
5197+
defer s.wg.Done()
5198+
5199+
peer.Disconnect(fmt.Errorf("server: DisconnectPeer called"))
5200+
}()
51965201

51975202
return nil
51985203
}

0 commit comments

Comments
 (0)