fix: agent ws connect panic

2026-01-01 13:58:27 +08:00
parent c29fb178a2
commit ac44bc450b
3 changed files with 6 additions and 23 deletions
@@ -410,7 +410,7 @@ func (c *AgentController) wsReadPump(ac *services.AgentConnection, agent *models
 		if r := recover(); r != nil {
 			logger.Errorf("[AgentWS] Agent #%d wsReadPump panic: %v", agent.ID, r)
 		}
-		c.wsManager.Unregister(agent.ID)
+		c.wsManager.Unregister(agent.ID, ac)
 	}()

 	// 检查连接是否有效（可能是旧连接被新连接替换）