Golang实现WebSocket通信详解

想要在Golang中实现WebSocket通信吗？本文将手把手教你使用强大的`gorilla/websocket`库，轻松构建实时交互应用。首先，通过`go get`命令安装依赖，接着，我们将创建一个WebSocket服务端，利用`upgrader.Upgrade`函数将HTTP连接升级为WebSocket连接，并使用`ReadMessage`和`WriteMessage`处理消息的收发。此外，文章还提供了前端测试页面的代码示例，以及如何使用Go作为客户端发起连接。最后，我们还会分享一些关于读写超时设置、并发安全管理和异步写入优化的实用技巧，助你打造稳定高效的Golang WebSocket服务。

使用gorilla/websocket库可实现Go语言中的WebSocket通信。1. 通过go get github.com/gorilla/websocket安装依赖；2. 创建服务端，利用upgrader.Upgrade将HTTP升级为WebSocket，通过ReadMessage和WriteMessage处理消息收发；3. 前端使用new WebSocket("ws://localhost:8080/ws")连接并测试通信；4. Go也可作为客户端，使用websocket.DefaultDialer.Dial发起连接；5. 注意设置读写超时、并发安全管理连接及异步写入优化。

在Golang中实现WebSocket通信，主要依赖第三方库 gorilla/websocket，这是目前最常用且稳定的选择。Go标准库本身不包含WebSocket支持，因此需要引入这个库来完成握手、消息读写等操作。

1. 安装 gorilla/websocket 库

使用 go mod 管理依赖，在项目根目录执行：

go get github.com/gorilla/websocket

这会自动下载并添加到 go.mod 文件中。

2. 创建 WebSocket 服务端

以下是一个简单的 WebSocket 服务器示例，支持客户端连接、接收消息并回显：

package main
import (
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true // 允许跨域连接，生产环境应限制具体来源
},
}
func echoHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Print("升级失败:", err)
return
}
defer conn.Close()
for {
    messageType, message, err := conn.ReadMessage()
    if err != nil {
        log.Print("读取消息失败:", err)
        break
    }

    log.Printf("收到: %s", message)
    if err := conn.WriteMessage(messageType, message); err != nil {
        log.Print("发送消息失败:", err)
        break
    }
}
}
func main() {
http.HandleFunc("/ws", echoHandler)
log.Println("服务启动在 :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}

说明：

• upgrader.Upgrade 将HTTP请求升级为WebSocket连接。
• CheckOrigin 设置为true允许所有跨域请求，实际部署时建议验证Origin头。
• ReadMessage 阻塞等待客户端消息，返回消息类型和字节数据。
• WriteMessage 向客户端发送消息，保持类型一致（如文本或二进制）。

3. 编写前端测试页面

创建一个HTML页面测试连接：

打开浏览器访问该页面，可在控制台看到来回通信的日志。

4. 实现客户端（可选）

Go也可以作为WebSocket客户端：

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 client"))
, msg,  := conn.ReadMessage()
log.Printf("响应: %s", msg)

这种方式适合微服务间通信或自动化测试。

5. 注意事项与优化建议

• 每个连接使用独立的goroutine处理，避免阻塞其他客户端。
• 设置读写超时防止连接长时间挂起：

conn.SetReadDeadline(time.Now().Add(60 * time.Second))
conn.SetWriteDeadline(time.Now().Add(10 * time.Second))

• 管理活跃连接可用全局map+互斥锁，或结合Redis做集群状态同步。
• 对高频消息场景，考虑使用缓冲通道进行异步写入，避免WriteMessage阻塞。

基本上就这些。用 gorilla/websocket 能快速搭建可靠的WebSocket服务，结构清晰，易于扩展。

到这里，我们也就讲完了《Golang实现WebSocket通信详解》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！