check channel close in listenNKN

Signed-off-by: Yilun <[email protected]>
nknorg · Nov 19, 2024 · f6231de · f6231de
1 parent 3e6e2ef
commit f6231de
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -359,7 +359,10 @@ func (c *TunaSessionClient) GetPubAddrs() *PubAddrs {
 
 func (c *TunaSessionClient) listenNKN() {
 	for {
-		msg := <-c.multiClient.OnMessage.C
+		msg, ok := <-c.multiClient.OnMessage.C
+		if !ok {
+			return
+		}
 		if !c.shouldAcceptAddr(msg.Src) {
 			continue
 		}