如何在Golang中实现WebSocket通信-Golang-PHP中文网

如何在Golang中实现WebSocket通信

P粉602998670

发布： 2025-10-04 15:39:02

原创

375人浏览过

使用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通信

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

1. 安装 gorilla/websocket 库

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

go get github.com/gorilla/websocket

登录后复制

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

2. 创建 WebSocket 服务端

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

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

package main
<p>import (
"log"
"net/http"
"github.com/gorilla/websocket"
)</p><p>var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true // 允许跨域连接，生产环境应限制具体来源
},
}</p><p>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()</p><pre class='brush:php;toolbar:false;'>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页面测试连接：

如知AI笔记

如知笔记——支持markdown的在线笔记，支持ai智能写作、AI搜索，支持DeepseekR1满血大模型

查看详情

<!DOCTYPE html>
<html>
<body>
<script>
const ws = new WebSocket("ws://localhost:8080/ws");
<p>ws.onopen = function() {
console.log("连接建立");
ws.send("Hello, Server!");
};</p><p>ws.onmessage = function(event) {
console.log("收到:", event.data);
};
</script>
</body>
</html></p>

登录后复制

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

4. 实现客户端（可选）

Go也可以作为WebSocket客户端：

conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)
if err != nil {
    log.Fatal("连接失败:", err)
}
defer conn.Close()
<p>conn.WriteMessage(websocket.TextMessage, []byte("Hi from client"))
<em>, msg, </em> := conn.ReadMessage()
log.Printf("响应: %s", msg)</p>

登录后复制

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

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通信的详细内容，更多请关注php中文网其它相关文章！

大家都在看：

Golang处理Session怎么做内存与Redis存储方案对比 Golang操作Redis数据库实践使用go-redis客户端库连接 Golang新手怎样操作Redis 使用go-redis连接与数据存取怎样用Golang实现分布式锁对比Redis与Etcd的实现优劣 Golang如何搭建分布式锁服务集成Redis Redlock算法实现