如何使用Golang开发WebSocket聊天室-Golang-PHP中文网

如何使用Golang开发WebSocket聊天室

P粉602998670

发布： 2025-10-01 17:19:01

原创

197人浏览过

使用Golang开发WebSocket聊天室需先建立持久连接，通过net/http和gorilla/websocket库升级HTTP连接；2. 管理客户端状态采用全局clients映射存储活跃连接，结合broadcast通道传递消息；3. 实现广播机制，在handleMessages循环中读取broadcast通道内容并发送给所有客户端。

如何使用golang开发websocket聊天室

用Golang开发WebSocket聊天室，核心是建立持久连接、管理客户端状态、实现消息广播。Go语言的高效并发模型特别适合这种实时通信场景。下面一步步说明如何构建一个基础但完整的WebSocket聊天室。

搭建WebSocket服务器

使用标准库net/http结合第三方库gorilla/websocket可以快速实现WebSocket服务。先安装依赖：

go get github.com/gorilla/websocket

创建HTTP处理器，将普通请求升级为WebSocket连接：

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

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool { return true }, // 允许跨域
}

func handleConnections(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
        return
    }
    defer conn.Close()

    // 连接成功后，等待接收消息
    for {
        var msg string
        err := conn.ReadJSON(&msg)
        if err != nil {
            log.Printf("读取消息错误: %v", err)
            break
        }
        // 后续将消息推送给所有用户
    }
}

登录后复制

管理客户端与广播消息

聊天室的关键是维护所有活跃连接，并在收到新消息时通知所有人。可以用全局变量保存客户端集合：

type Client struct {
    conn *websocket.Conn
}

var clients = make(map[*Client]bool)
var broadcast = make(chan string)

func handleMessages() {
    for {
        msg := <-broadcast
        for client := range clients {
            err := client.conn.WriteJSON(msg)
            if err != nil {
                client.conn.Close()
                delete(clients, client)
            }
        }
    }
}

登录后复制

在main函数中启动消息广播协程：

go handleMessages()

AliGenie 天猫精灵开放平台

天猫精灵开放平台

查看详情

当新连接建立时，将其加入clients，并将收到的消息发送到broadcast通道。

前端页面接入WebSocket

写一个简单的HTML页面，通过JavaScript建立WebSocket连接：

const ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(event) {
    const chatBox = document.getElementById("chat");
    const msg = document.createElement("div");
    msg.textContent = event.data;
    chatBox.appendChild(msg);
};

function send() {
    const input = document.getElementById("input");
    ws.send(input.value);
    input.value = "";
}

登录后复制

配合一个基础表单，用户就能输入并发送消息。

启动服务与测试

在main函数中注册路由并启动服务器：

func main() {
    http.HandleFunc("/ws", handleConnections)
    http.Handle("/", http.FileServer(http.Dir("./static")))
    log.Println("服务器运行在 :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

登录后复制

把HTML文件放在static目录下，运行程序后访问http://localhost:8080即可打开聊天页面。多个浏览器窗口间能实时收发消息。

基本上就这些。不复杂但容易忽略的是错误处理和连接清理——断开的客户端要及时从clients中移除，避免写入失效连接导致panic。可以加锁保护map操作，或改用sync.Map提升并发安全。功能扩展如私聊、房间、用户名等，可在结构体中添加字段并设计消息格式支持。

以上就是如何使用Golang开发WebSocket聊天室的详细内容，更多请关注php中文网其它相关文章！