首先使用 gorilla/websocket 库实现 WebSocket 服务，定义 Upgrader 并处理连接请求，在 wsHandler 中读取和回显消息，最后通过 http.ListenAndServe 启动服务，并用 javaScript 或 Go 客户端测试通信。

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，相比传统的 HTTP 轮询，它能显著降低延迟并提升实时性。golang 凭借其高并发和轻量级 Goroutine 的优势，非常适合构建高性能的 WebSocket 服务。本文将详细介绍如何使用 Go 实现 WebSocket 通信，并提供完整可运行的实践示例。

使用 gorilla/websocket 库实现 WebSocket 服务

Go 标准库不直接支持 WebSocket，但社区广泛使用的 gorilla/websocket 库提供了完整的实现。首先安装该库：

go get github.com/gorilla/websocket

接下来创建一个基础的 WebSocket 服务器：

定义升级器（Upgrader），用于将 HTTP 连接升级为 WebSocket 连接：

立即学习“go语言免费学习笔记（深入）”；

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 {
        log.Println(“Upgrade Error:”, err)
        return
    }
    defer conn.Close()

        err = conn.WriteMessage(websocket.TextMessage, msg)
        if err != nil {
            log.Println(“Write error:”, err)
            break
        }
    }
}

启动 HTTP 服务：

func main() {
    http.HandleFunc(“/ws”, wsHandler)
    log.Println(“Server started on :8080”)
    log.Fatal(http.ListenAndServe(“:8080”, nil))
}

客户端连接与消息收发

使用 javascript 浏览器客户端测试服务：

<script>
const ws = new WebSocket(“ws://localhost:8080/ws”);
ws.onopen = () => {
  ws.send(“Hello, Go WebSocket!”);
};
ws.onmessage = (Event) => {
  console.log(“Received:”, event.data);
};
</script>

你也可以用 Go 编写客户端：

微信 WeLM

WeLM不是一个直接的对话机器人，而是一个补全用户输入信息的生成模型。

33

查看详情

func main() {
    conn, _, err := websocket.DefaultDialer.Dial(“ws://localhost:8080/ws”, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer conn.Close()

    conn.WriteMessage(websocket.TextMessage, []byte(“Hi from Go client”))
    _, msg, _ := conn.ReadMessage()
    fmt.Printf(“Response: %sn”, msg)
}

实现广播机制与连接管理

实际项目中通常需要管理多个连接并支持广播。可以使用 map 和互斥锁来保存客户端连接：

var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan []byte)
var mutex = sync.Mutex{}

在处理函数中注册连接：

mutex.Lock()
clients[conn] = true
mutex.Unlock()

启动广播监听协程：

go func() {
    for {
        msg :=         mutex.Lock()
        for client := range clients {
            err := client.WriteMessage(websocket.TextMessage, msg)
            if err != nil {
                client.Close()
                delete(clients, client)
            }
        }
        mutex.Unlock()
    }
}()

收到消息后推送到广播通道：

broadcast <- msg

心跳检测与连接健康维护

长时间空闲可能导致连接中断。通过 Ping/Pong 机制维持活跃：

conn.SetReadDeadline(time.Now().Add(60 * time.Second))
conn.SetPongHandler(func(string) error {
    conn.SetReadDeadline(time.Now().Add(60 * time.Second))
    return nil
})

服务端定期发送 Ping 消息：

go func() {
    ticker := time.NewTicker(54 * time.Second)
    defer ticker.Stop()
    for {
        select {
        case <-ticker.C:
            if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
                return
            }
        }
    }
}()

基本上就这些。通过 gorilla/websocket，Go 能轻松实现稳定高效的 WebSocket 服务。关键在于连接管理、并发安全和心跳机制的设计。实际部署时建议结合 TLS 使用 wss 协议保障安全。不复杂但容易忽略细节，比如关闭连接和资源清理。