在Golang中构建小型聊天室，核心是利用WebSocket实现客户端和服务端的实时双向通信，同时借助goroutine和channel处理多客户端的并发连接与消息分发，整体逻辑清晰且实现成本较低。
核心实现思路
小型聊天室需要解决三个核心问题：一是管理所有在线的客户端连接，二是接收单个客户端发送的消息，三是将消息广播给所有在线客户端。Golang的goroutine可以轻松处理每个客户端的独立读写协程，channel则可以作为消息传递的中间载体，避免并发操作连接集合时的锁竞争问题。
基础结构定义
首先定义聊天室的核心结构，包括客户端连接封装和服务端管理结构：
package main
import (
"fmt"
"net/http"
"github.com/gorilla/websocket"
)
// 客户端结构，封装单个WebSocket连接和对应的消息发送通道
type Client struct {
conn *websocket.Conn
send chan []byte
}
// 聊天室服务端结构，管理所有在线客户端和广播消息通道
type ChatRoom struct {
clients map[*Client]bool
broadcast chan []byte
register chan *Client
unregister chan *Client
}
// 初始化聊天室实例
func NewChatRoom() *ChatRoom {
return &ChatRoom{
clients: make(map[*Client]bool),
broadcast: make(chan []byte),
register: make(chan *Client),
unregister: make(chan *Client),
}
}
聊天室核心运行逻辑
聊天室需要有一个持续运行的协程来处理客户端的注册、注销和消息广播，这部分逻辑放在ChatRoom的Run方法中：
func (c *ChatRoom) Run() {
for {
select {
// 处理新客户端注册
case client := <-c.register:
c.clients[client] = true
fmt.Println("新客户端加入，当前在线人数:", len(c.clients))
// 处理客户端注销
case client := <-c.unregister:
if _, ok := c.clients[client]; ok {
delete(c.clients, client)
close(client.send)
fmt.Println("客户端离开，当前在线人数:", len(c.clients))
}
// 处理广播消息
case message := <-c.broadcast:
for client := range c.clients {
select {
case client.send <- message:
// 如果发送失败，说明客户端连接异常，直接注销
default:
close(client.send)
delete(c.clients, client)
}
}
}
}
}
客户端读写协程实现
每个客户端连接建立后，需要启动两个协程分别处理读操作和写操作：
读协程：接收客户端消息并推送到广播通道
func (c *Client) ReadPump(chatRoom *ChatRoom) {
defer func() {
chatRoom.unregister <- c
c.conn.Close()
}()
for {
// 读取客户端发送的消息
_, message, err := c.conn.ReadMessage()
if err != nil {
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
fmt.Printf("连接异常关闭: %vn", err)
}
break
}
// 将消息推送到聊天室的广播通道
chatRoom.broadcast <- message
}
}
写协程：从发送通道取消息推送给客户端
func (c *Client) WritePump() {
defer c.conn.Close()
for {
select {
case message, ok := <-c.send:
if !ok {
// 通道关闭，说明客户端已经注销
c.conn.WriteMessage(websocket.CloseMessage, []byte{})
return
}
// 将消息发送给客户端
err := c.conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
fmt.Printf("消息发送失败: %vn", err)
return
}
}
}
}
HTTP服务与WebSocket升级
最后需要实现HTTP接口，处理客户端的WebSocket连接请求，完成协议升级并启动对应的读写协程：
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
func serveWs(chatRoom *ChatRoom, w http.ResponseWriter, r *http.Request) {
// 升级HTTP连接为WebSocket连接
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
fmt.Printf("WebSocket升级失败: %vn", err)
return
}
// 创建客户端实例
client := &Client{
conn: conn,
send: make(chan []byte, 256),
}
// 注册客户端到聊天室
chatRoom.register <- client
// 启动读写协程
go client.WritePump()
go client.ReadPump(chatRoom)
}
func main() {
chatRoom := NewChatRoom()
// 启动聊天室核心协程
go chatRoom.Run()
// 注册WebSocket接口
http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
serveWs(chatRoom, w, r)
})
// 启动HTTP服务
fmt.Println("聊天室服务启动，监听端口8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
fmt.Printf("服务启动失败: %vn", err)
}
}
运行与测试
运行程序前需要先安装gorilla/websocket依赖，执行命令：
go get github.com/gorilla/websocket
启动程序后，可以使用任意WebSocket客户端工具连接ws://127.0.0.1:8080/ws，多个客户端连接后发送消息，就能看到所有在线客户端都能收到广播的消息，小型聊天室功能就基本实现了。如果需要扩展功能，还可以在消息中添加客户端标识、实现私聊、消息持久化等逻辑。