move onDisconect before close (#637)

googollee · May 29, 2024 · 65235b2 · 65235b2
1 parent b044011
commit 65235b2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/connection.go b/connection.go
@@ -61,11 +61,10 @@ func (c *conn) Close() error {
 	c.closeOnce.Do(func() {
 		// for each namespace, leave all rooms, and call the disconnect handler.
 		c.namespaces.Range(func(ns string, nc *namespaceConn) {
-			nc.LeaveAll()
-
 			if nh, _ := c.handlers.Get(ns); nh != nil && nh.onDisconnect != nil {
 				nh.onDisconnect(nc, clientDisconnectMsg)
 			}
+			nc.LeaveAll()
 		})
 		err = c.Conn.Close()