使用Go语言开发WebSocket聊天系统需借助gorilla/websocket库处理连接升级；2. 通过Upgrader实现HTTP到WebSocket协议的切换并处理消息收发；3. 利用Hub结构体管理客户端连接，通过broadcast通道实现消息广播，register和unregister通道维护连接状态，确保并发安全。
用Go语言开发WebSocket聊天系统，核心在于高效处理并发连接和实时消息分发。Golang天生适合这类场景，得益于其轻量级的goroutine和简洁的net/http包支持。实际开发中，我们通常使用gorilla/websocket库来简化WebSocket协议的处理。
建立WebSocket连接
客户端通过HTTP升级请求切换到WebSocket协议，服务端需正确响应。使用gorilla/websocket可以快速完成握手过程。
定义一个升级器，将HTTP连接升级为WebSocket连接：
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true // 允许所有来源，生产环境应做校验
},
}
<p>func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Printf("升级失败: %v", err)
return
}
defer conn.Close()</p><pre class='brush:php;toolbar:false;'>// 连接建立后，可开始读取消息
for {
_, msg, err := conn.ReadMessage()
if err != nil {
log.Printf("读取消息失败: %v", err)
break
}
log.Printf("收到消息: %s", msg)
}}
立即学习“go语言免费学习笔记（深入）”；
注册路由即可启用：
http.HandleFunc("/ws", wsHandler)管理客户端连接
聊天系统需要维护所有活跃连接，以便广播消息。可以用一个全局的map存储连接，并用互斥锁保护并发访问。
定义结构体管理客户端：
type Client struct {
conn *websocket.Conn
send chan []byte
}
<p>type Hub struct {
clients map[<em>Client]bool
broadcast chan []byte
register chan </em>Client
unregister chan *Client
}</p><p>var hub = Hub{
clients: make(map[<em>Client]bool),
broadcast: make(chan []byte),
register: make(chan </em>Client),
unregister: make(chan *Client),
}</p>启动hub监听事件：
Golang Uber Dig
下载
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等
func (h *Hub) Run() {
for {
select {
case client := <-h.register:
h.clients[client] = true
case client := <-h.unregister:
if _, ok := h.clients[client]; ok {
delete(h.clients, client)
close(client.send)
}
case message := <-h.broadcast:
for client := range h.clients {
select {
case client.send <- message:
default:
close(client.send)
delete(h.clients, client)
}
}
}
}
}
实现消息广播
每个客户端开启两个goroutine，一个负责读，一个负责写。读取到的消息发送给hub的broadcast通道，由hub推送给所有客户端。
在wsHandler中启动客户端协程：
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
return
}
<pre class='brush:php;toolbar:false;'>client := &Client{conn: conn, send: make(chan []byte, 256)}
hub.register <- client
go client.writePump()
client.readPump()}
立即学习“go语言免费学习笔记（深入）”；
读取消息并转发给hub：
func (c *Client) readPump() {
defer func() {
hub.unregister <- c
c.conn.Close()
}()
for {
_, message, err := c.conn.ReadMessage()
if err != nil {
break
}
hub.broadcast <- message
}
}
</font><p>写入消息到客户端：</p><font face="Courier New"><pre class="brush:php;toolbar:false;">
func (c *Client) writePump() {
defer func() {
c.conn.Close()
}()
for {
select {
case message, ok := <-c.send:
if !ok {
c.conn.WriteMessage(websocket.CloseMessage, []byte{})
return
}
c.conn.WriteMessage(websocket.TextMessage, message)
}
}
}
前端简单对接
前端通过原生WebSocket API连接服务端：
const ws = new WebSocket("ws://localhost:8080/ws");
ws.onopen = () => console.log("已连接");
ws.onmessage = (event) => console.log("收到:", event.data);
ws.onclose = () => console.log("连接关闭");
<p>// 发送消息
function sendMsg() {
const input = document.getElementById("msg");
ws.send(input.value);
input.value = "";
}</p>配合简单HTML输入框即可实现基础交互。
基本上就这些。Golang的简洁性和高性能让WebSocket服务开发变得直接高效。关键点是合理利用channel和goroutine进行并发控制，避免数据竞争。生产环境中还需考虑心跳、重连、认证和消息持久化等问题，但核心模型不变。
