Golang WebSocket聊天系统开发实践-Golang-PHP中文网

Golang WebSocket聊天系统开发实践

P粉602998670

发布： 2025-09-29 19:26:01

原创

615人浏览过

使用Go语言开发WebSocket聊天系统需借助gorilla/websocket库处理连接升级；2. 通过Upgrader实现HTTP到WebSocket协议的切换并处理消息收发；3. 利用Hub结构体管理客户端连接，通过broadcast通道实现消息广播，register和unregister通道维护连接状态，确保并发安全。

golang websocket聊天系统开发实践

用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监听事件：

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推送给所有客户端。

Linux加PHP加MySQL案例教程

通过大量实例系统全面地介绍了Linux+PHP+MySQL环境下的网络后台开发技术，详尽分析了近30个典型案例。本书以培养高级网站建设与管理人才为目标，内容循序渐进，由浅入深，通过大量的实例系统全面地介绍了Linux+PHP+MySQL环境下的网络后台开发技术。本书详尽分析了近30个典型案例。包括计数器、网站流量统计、留言扳、论坛系统、聊天室、投票与调查、用户管理、新闻发布系统、广告轮播

447

查看详情

在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进行并发控制，避免数据竞争。生产环境中还需考虑心跳、重连、认证和消息持久化等问题，但核心模型不变。

以上就是Golang WebSocket聊天系统开发实践的详细内容，更多请关注php中文网其它相关文章！