Golang中处理WebSocket写入错误的解决方案

请展示地图的定义…

要检查一个项是否已从映射中删除，请执行以下操作

if c, ok := sockets["123456"]; ok {
    // 仍然存在

问题出现在并发访问和竞态条件上。虽然你使用了 if c.sockets[mr.socket] != nil 检查，但在检查之后、调用 WriteMessage 之前，连接可能已经被注销并从映射中删除。

以下是解决方案：

case mr := <-c.broadcast:
    msg := mr.message
    
    // 获取连接时需要加锁
    c.mu.RLock()
    conn := c.sockets[mr.socket]
    c.mu.RUnlock()
    
    if conn != nil {
        if err := conn.WriteMessage(websocket.TextMessage, msg); err != nil {
            // 写入失败时也需要加锁
            c.mu.Lock()
            if c.sockets[mr.socket] == conn { // 再次验证连接是否仍然相同
                delete(c.sockets, mr.socket)
            }
            c.mu.Unlock()
            
            // 关闭连接
            conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
            conn.WriteMessage(websocket.CloseMessage, []byte{})
            conn.Close()
        }
    }

同时修改注销函数，确保在删除连接后不会留下悬空指针：

case connection := <-c.unregister:
    uuid := connection.uuid
    
    c.mu.Lock()
    conn := c.sockets[uuid]
    if conn != nil {
        delete(c.sockets, uuid)
    }
    c.mu.Unlock()
    
    // 如果找到了连接，关闭它
    if conn != nil {
        conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
        conn.WriteMessage(websocket.CloseMessage, []byte{})
        conn.Close()
    }

对于广播消息发送函数，建议改为：

func (mr MessageRequest) Send(msg []byte) {
    mr.mu.RLock()
    defer mr.mu.RUnlock()
    
    // 收集所有需要发送的连接
    connections := make([]*websocket.Conn, 0, len(mr.sockets))
    for _, socket := range mr.sockets {
        connections = append(connections, socket)
    }
    
    // 发送到通道
    for _, conn := range connections {
        mr.chat.broadcast <- Message{
            socket:  conn,  // 这里应该存储连接标识符，而不是连接本身
            message: msg,
            chat:    mr.chat,
        }
    }
}

注意：在 Message 结构中，socket 字段应该存储连接的唯一标识符（如UUID），而不是 *websocket.Conn 指针，这样可以避免在通道中传递可能已经失效的连接指针。