fix: agent ws connect panic

internal/controllers/agent_controller.go

@@ -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)
 	}()
 
 	// 检查连接是否有效（可能是旧连接被新连接替换）

internal/services/agent_ws_service.go

@@ -161,12 +161,13 @@ func (m *AgentWSManager) Register(agentID uint, conn *websocket.Conn, ip string)
 	return ac
 }
 
-// Unregister 注销连接
-func (m *AgentWSManager) Unregister(agentID uint) {
+// Unregister 注销连接（只注销指定的连接实例）
+func (m *AgentWSManager) Unregister(agentID uint, ac *AgentConnection) {
 	m.mu.Lock()
 	defer m.mu.Unlock()
 
-	if conn, exists := m.connections[agentID]; exists {
+	// 只有当前连接和 map 中的连接是同一个实例时才删除
+	if conn, exists := m.connections[agentID]; exists && conn == ac {
 		// 减少 IP 连接计数
 		if conn.IP != "" {
 			if count, ok := m.ipConnections[conn.IP]; ok && count > 0 {