Golang WebSocket实时通信开发

我正在用Golang开发一个需要WebSocket实时通信的项目，遇到几个问题想请教大家：

  1. 如何高效处理大量并发WebSocket连接？Golang的goroutine虽然轻量，但连接数上万时内存占用会不会很高？
  2. 有没有推荐的生产级WebSocket库？标准库的gorilla/websocket和nhooyr.io/websocket各有什么优缺点？
  3. 客户端异常断开时，服务端如何可靠地检测并清理资源？心跳机制的最佳实践是什么？
  4. 需要同时支持JSON和Protobuf协议传输，有什么优雅的实现方案？
    求有实战经验的大佬分享建议！
2 回复

使用Golang开发WebSocket实时通信很简单，推荐使用gorilla/websocket库。

核心步骤：

  1. 安装依赖
go get github.com/gorilla/websocket
  1. 升级HTTP连接为WebSocket
var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool { return true },
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        return
    }
    defer conn.Close()
}
  1. 消息处理循环
for {
    messageType, p, err := conn.ReadMessage()
    if err != nil {
        break
    }
    
    // 处理消息
    err = conn.WriteMessage(messageType, p)
    if err != nil {
        break
    }
}
  1. 广播消息 维护连接池，遍历所有连接发送消息。

优化建议：

  • 使用goroutine处理每个连接
  • 添加连接超时和心跳机制
  • 使用通道进行消息队列管理
  • 考虑使用Redis Pub/Sub实现分布式

适用场景： 聊天室、实时数据推送、在线协作等。

更多关于Golang WebSocket实时通信开发的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


在Golang中开发WebSocket实时通信，推荐使用gorilla/websocket库，这是最流行的WebSocket实现方案。

1. 安装依赖

go get github.com/gorilla/websocket

2. 基础服务器实现

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 生产环境应验证来源
    },
}

var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan Message)

type Message struct {
    Username string `json:"username"`
    Message  string `json:"message"`
}

func handleConnections(w http.ResponseWriter, r *http.Request) {
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer ws.Close()
    
    clients[ws] = true
    
    for {
        var msg Message
        err := ws.ReadJSON(&msg)
        if err != nil {
            log.Printf("error: %v", err)
            delete(clients, ws)
            break
        }
        broadcast <- msg
    }
}

func handleMessages() {
    for {
        msg := <-broadcast
        for client := range clients {
            err := client.WriteJSON(msg)
            if err != nil {
                log.Printf("error: %v", err)
                client.Close()
                delete(clients, client)
            }
        }
    }
}

func main() {
    http.HandleFunc("/ws", handleConnections)
    go handleMessages()
    
    log.Println("服务器启动在 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

3. 客户端HTML示例

<!DOCTYPE html>
<html>
<body>
    <div id="messages"></div>
    <input type="text" id="messageInput">
    <button onclick="sendMessage()">发送</button>

    <script>
        const ws = new WebSocket("ws://localhost:8080/ws");
        const messagesDiv = document.getElementById("messages");
        
        ws.onmessage = function(event) {
            const msg = JSON.parse(event.data);
            messagesDiv.innerHTML += `<p>${msg.username}: ${msg.message}</p>`;
        };
        
        function sendMessage() {
            const input = document.getElementById("messageInput");
            const message = {
                username: "用户",
                message: input.value
            };
            ws.send(JSON.stringify(message));
            input.value = "";
        }
    </script>
</body>
</html>

4. 关键特性说明

连接升级：通过upgrader.Upgrade()将HTTP连接升级为WebSocket连接。

并发处理：

  • 使用goroutine处理消息广播
  • 使用channel进行线程安全的消息传递

连接管理：

  • clients map跟踪所有活跃连接
  • 自动清理断开连接的客户端

5. 生产环境建议

  1. 添加认证：在连接建立前验证用户身份
  2. 错误处理：完善的连接异常处理
  3. 限流控制：防止单个客户端过度占用资源
  4. 心跳检测：定期检查连接健康状态
  5. 使用WSS：生产环境务必使用加密连接

这个基础框架可以扩展为聊天应用、实时数据推送、多人协作等场景。

回到顶部