Don't panic when callbacks aren't set

neilalexander · Jul 27, 2021 · 54ae7b1 · 54ae7b1
1 parent 52626cd
commit 54ae7b1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/conn.go b/conn.go
@@ -616,5 +616,7 @@ func (c *Conn) detach() {
 	s.lazyDestroy()
 	s.onMutex.RLock()
 	defer s.onMutex.RUnlock()
-	go s.onDetach(c.remoteSocketAddr)
+	if s.onDetach != nil {
+		go s.onDetach(c.remoteSocketAddr)
+	}
 }
diff --git a/socket.go b/socket.go
@@ -445,7 +445,9 @@ func (s *Socket) registerConn(recvID uint16, remoteAddr resolvedAddrStr, c *Conn
 	s.conns[key] = c
 	s.onMutex.RLock()
 	defer s.onMutex.RUnlock()
-	go s.onAttach(c.remoteSocketAddr)
+	if s.onAttach != nil {
+		go s.onAttach(c.remoteSocketAddr)
+	}
 	return true
 }