golang 使用nhooyr.io/websocket编写websocket 编写聊天室

当使用 nhooyr.io/websocket 编写聊天室时，您可以使用以下步骤：

1. 创建一个 WebSocket 服务器：

http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
    conn, err := websocket.Accept(w, r, nil)
    if err != nil {
        // 处理连接错误
        return
    }
    defer conn.Close(websocket.StatusInternalError, "连接关闭")

    // 将连接添加到聊天室中
    chatRoom.AddConnection(conn)

    // 处理聊天消息
    for {
        _, message, err := conn.Read(r.Context())
        if err != nil {
            // 处理读取错误
            break
        }

        // 将消息广播给所有连接
        chatRoom.Broadcast(message)
    }
})

2. 创建一个聊天室结构体：

type ChatRoom struct {
    connections map[*websocket.Conn]struct{}
    mutex       sync.Mutex
}

func NewChatRoom() *ChatRoom {
    return &ChatRoom{
        connections: make(map[*websocket.Conn]struct{}),
    }
}

func (c *ChatRoom) AddConnection(conn *websocket.Conn) {
    c.mutex.Lock()
    defer c.mutex.Unlock()

    c.connections[conn] = struct{}{}
}

func (c *ChatRoom) Broadcast(message []byte) {
    c.mutex.Lock()
    defer c.mutex.Unlock()

    for conn := range c.connections {
        err := conn.Write(r.Context(), websocket.MessageText, message)
        if err != nil {
            // 处理写入错误
            conn.Close(websocket.StatusInternalError, "写入错误")
            delete(c.connections, conn)
        }
    }
}

3. 启动服务器：

func main() {
    chatRoom := NewChatRoom()

    http.ListenAndServe(":8080", nil)
}

这是一个简单的聊天室示例，它接受来自客户端的连接并将消息广播给所有连接。您可以根据需求进行进一步的扩展，例如添加用户身份验证、管理在线用户列表等功能。

请注意，这只是一个基本示例，您还需要考虑更多的错误处理、并发安全等方面。希望对您有所帮助！如果您有任何其他问题，请随时提问。