使用golang实现websocket服务的关键在于选择合适的库并理解基本流程。一、推荐使用成熟的第三方库gorilla/websocket，可通过go get命令安装；二、通过创建http服务器并在指定路由升级连接实现服务端搭建，每个连接由独立协程处理，示例代码展示了消息的接收与回传；三、客户端同样利用该库建立连接并实现消息收发；四、实际应用中需引入连接池和广播机制管理并发连接，如使用全局map维护连接及通过channel实现消息广播。结构清晰、注重细节是成功实现的关键。
用Golang实现WebSocket服务，其实并不难。Go语言本身的并发模型非常适合处理这种长连接、高并发的实时通信场景。只要掌握了基本结构和流程，就能快速搭建出一个稳定可靠的WebSocket服务。
一、选择合适的WebSocket库
虽然标准库中没有原生支持WebSocket，但Go生态里有几个非常成熟的第三方库可以使用。最常用的是gorilla/websocket，它功能完整、文档齐全，社区活跃度也很高。
你可以通过下面这个命令安装：
立即学习“go语言免费学习笔记（深入）”；
go get github.com/gorilla/websocket
安装好之后就可以开始写服务端代码了。相比自己从头实现协议解析，使用这个库能节省大量时间，也更稳定可靠。
二、搭建基础的WebSocket服务端
要启动一个WebSocket服务，首先要创建一个HTTP服务器，然后在某个路由上“升级”连接到WebSocket。
以下是一个简单的示例：
package main
import (
"fmt"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool {
return true // 简单起见先允许所有跨域请求
},
}
func handleWebSocket(conn *websocket.Conn) {
for {
messageType, p, err := conn.ReadMessage()
if err != nil {
fmt.Println("Read error:", err)
return
}
fmt.Printf("Received: %s\n", p)
if err := conn.WriteMessage(messageType, p); err != nil {
fmt.Println("Write error:", err)
return
}
}
}
func main() {
http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
conn, _ := upgrader.Upgrade(w, r, nil)
go handleWebSocket(conn)
})
fmt.Println("Starting server on :8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
panic(err)
}
}上面这段代码实现了：
- 监听
/ws路径上的WebSocket连接； - 每个连接都会被单独处理（得益于
go handleWebSocket(conn)）； - 收到消息后原样返回，适合测试用。
三、实现客户端连接与消息收发
除了服务端，你可能也需要写一个简单的客户端来测试。同样可以用gorilla/websocket包。
示例代码如下：
package main
import (
"fmt"
"github.com/gorilla/websocket"
)
func main() {
c, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)
if err != nil {
panic(err)
}
msg := []byte("Hello WebSocket!")
c.WriteMessage(websocket.TextMessage, msg)
_, resp, _ := c.ReadMessage()
fmt.Printf("Response: %s\n", resp)
}运行起来后你应该能看到服务端打印收到的消息，同时客户端也收到了回声。
四、优化并发与连接管理
当你要构建实际应用时，比如聊天室或通知系统，就需要考虑如何管理多个连接，并进行广播或定向推送。
常见做法是维护一个全局的连接池，比如：
var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan Message)
type Message struct {
Conn *websocket.Conn
Data []byte
}然后在主函数中启动一个广播协程：
func broadcaster() {
for {
select {
case msg := <-broadcast:
for client := range clients {
if err := client.WriteMessage(websocket.TextMessage, msg.Data); err != nil {
delete(clients, client)
}
}
}
}
}这样每当有消息进入broadcast通道，就会自动推送给所有在线用户。你也可以根据需要扩展成按用户ID分组发送。
基本上就这些。用Golang写WebSocket服务不复杂，但细节需要注意，比如连接超时、错误处理、并发安全等。不过只要结构清晰、逻辑明确，很快就能跑通。
