Golang如何开发简单的聊天程序使用WebSocket实现通信-Golang-PHP中文网

Golang如何开发简单的聊天程序使用WebSocket实现通信

P粉602998670

发布： 2025-08-11 12:15:02

原创

632人浏览过

用golang实现websocket聊天程序的关键步骤如下：1. 安装go环境及配置项目结构，使用gorilla/websocket库搭建http服务并处理升级请求；2. 通过全局连接池管理客户端连接，为每个连接启动goroutine监听消息并广播给其他用户；3. 编写前端html页面利用javascript建立websocket连接并实现消息收发展示；4. 注意跨域设置、连接中断处理、并发安全及性能优化等常见问题。

Golang如何开发简单的聊天程序使用WebSocket实现通信

想用 Golang 做一个简单的聊天程序，WebSocket 是个很合适的通信方式。它支持双向实时通信，适合聊天这种需要即时响应的场景。下面我从几个关键点出发，讲讲怎么一步步实现。

1. 准备工作：搭建基础环境

要开始开发，首先得确保你的开发环境已经准备好。你需要安装 Go，并且配置好 GOPATH 和项目目录结构。然后，需要用到一个 WebSocket 的库，推荐使用

gorilla/websocket

登录后复制

，这是社区广泛使用的包。

安装方法很简单：

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

go get github.com/gorilla/websocket

登录后复制

接着你可以创建一个基本的 HTTP 服务作为入口，用于接收 WebSocket 升级请求。

示例代码片段：

http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
    conn, _ := upgrader.Upgrade(w, r, nil)
    // 处理连接逻辑
})
http.ListenAndServe(":8080", nil)

登录后复制

这里的

upgrader

登录后复制

是一个配置项，用于控制升级过程，比如跨域设置等。

2. 管理连接和广播消息

聊天程序的核心在于连接管理和消息转发。你需要维护所有当前连接的客户端，并在收到消息时广播给其他用户。

天工

昆仑万维推出的通用AI智能体平台，原天工AI

212

查看详情

可以定义一个全局的连接池，比如用

map[*websocket.Conn]bool

登录后复制

来保存连接对象。虽然不是线程安全的，但对简单场景来说够用了。

处理消息的方式通常是启动一个 goroutine 来监听每个连接的消息：

func handleConnection(conn *websocket.Conn) {
    defer conn.Close()

    for {
        _, msg, err := conn.ReadMessage()
        if err != nil {
            break
        }
        broadcastMessage(msg)
    }
}

登录后复制

broadcastMessage

登录后复制

函数负责把消息发送给所有在线的连接。

广播的时候注意加锁，避免并发写入 WebSocket 连接导致出错。

3. 实现前端页面展示和连接

后端搭好了，还需要一个简单的前端页面让用户接入。可以用 HTML + JavaScript 快速写一个测试页面。

HTML 示例：

<!DOCTYPE html>
<html>
<body>
    <div id="chat"></div>
    <input type="text" id="input" />
    <button onclick="sendMessage()">发送</button>

    <script>
        const ws = new WebSocket("ws://localhost:8080/ws");
        ws.onmessage = function(event) {
            document.getElementById('chat').innerText += '\n' + event.data;
        };

        function sendMessage() {
            const input = document.getElementById('input');
            ws.send(input.value);
            input.value = '';
        }
    </script>
</body>
</html>

登录后复制

这样就可以打开页面，输入内容后看到多个用户之间的消息互通了。