Golang WebSocket实时通信项目实战-Golang-PHP中文网

Golang WebSocket实时通信项目实战

P粉602998670

发布： 2025-10-11 19:59:01

原创

464人浏览过

答案：使用Gorilla WebSocket库结合Go的并发模型实现高效实时通信，通过main.go升级连接，hub.go管理客户端与广播消息，client.go处理读写，前端HTML测试交互，构建可扩展的WebSocket服务。

golang websocket实时通信项目实战

用Golang实现WebSocket实时通信，核心在于利用gorilla/websocket库建立长连接，配合轻量高效的并发模型，实现实时消息推送和广播。下面是一个贴近实际开发的实战结构，帮助你快速搭建一个可运行、可扩展的WebSocket服务。

1. 环境准备与依赖引入

Go语言原生不支持WebSocket，需借助第三方库。推荐使用社区广泛使用的gorilla/websocket：

go get github.com/gorilla/websocke

项目结构建议如下：

/main.go – 启动服务
/hub.go – 消息中心管理连接
/client.go – 客户端连接处理
/chat.html – 前端测试页面

2. 核心：WebSocket连接升级与客户端管理

在main.go中设置HTTP路由，并将特定路径升级为WebSocket连接：

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

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域（生产环境应限制）
    },
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Printf("升级失败: %v", err)
        return
    }
    // 将新连接交给中心管理
    client := &Client{conn: conn, send: make(chan []byte, 256)}
    hub.register <- client

    // 启动读写协程
    go client.writePump()
    go client.readPump()
}

登录后复制

说明：每个连接升级后生成一个Client实例，通过readPump监听前端消息，writePump向浏览器发送数据。

FashionLabs

AI服装模特、商品图，可商用，低价提升销量神器

查看详情

3. 消息广播中心设计（Hub）

hub.go是整个系统的核心调度器，负责管理所有客户端连接和消息分发：

type Hub struct {
    clients    map[*Client]bool
    register   chan *Client
    broadcast  chan []byte
}

var hub = Hub{
    clients:  make(map[*Client]bool),
    register: make(chan *Client),
    broadcast: make(chan []byte),
}

func (h *Hub) run() {
    for {
        select {
        case client := <-h.register:
            h.clients[client] = true
        case message := <-h.broadcast:
            for client := range h.clients {
                select {
                case client.send <- message:
                default:
                    close(client.send)
                    delete(h.clients, client)
                }
            }
        }
    }
}

登录后复制

启动时运行hub.run()协程，在后台持续监听注册和广播事件，实现一对多消息推送。

4. 前端简单测试页面

创建chat.html用于连接和服务端交互：

<script>
const ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(event) {
    console.log("收到:", event.data);
};
function send() {
    const input = document.getElementById("msg");
    ws.send(input.value);
    input.value = "";
}
</script>
<input id="msg" placeholder="输入消息"/>
<button onclick="send()">发送</button>

登录后复制

访问页面后，输入内容点击发送，消息会传到服务端，再由Hub广播给所有在线用户。

基本上就这些。Golang的并发特性让每个WebSocket连接只需一个轻量协程，成千上万连接也能轻松应对。只要理解“连接升级 → 客户端封装 → 中心广播”这个流程，就能在此基础上扩展群聊、私信、心跳检测等实用功能。

以上就是Golang WebSocket实时通信项目实战的详细内容，更多请关注php中文网其它相关文章！