Golang如何开发简单聊天程序基于websocket实现双向通信-Golang-PHP中文网

Golang如何开发简单聊天程序基于websocket实现双向通信

P粉602998670

发布： 2025-07-22 13:01:01

原创

766人浏览过

实现简单聊天程序需搭建golang服务端与前端websocket连接。首先使用gorilla/websocket库升级http连接建立websocket通信；其次通过维护客户端连接池和广播机制实现消息分发；接着编写前端页面监听消息并展示；最后注意并发控制和连接管理以确保稳定性。

Golang如何开发简单聊天程序基于websocket实现双向通信

实现一个简单的聊天程序是学习 Golang 和 WebSocket 的好方式。用 Golang 搭个服务端，配合前端的 WebSocket 连接，就可以实现实时双向通信。这个过程并不复杂，只要掌握基本结构和流程就能快速上手。

准备工作：环境与依赖

在开始写代码之前，确保你已经安装好了 Go 环境，并且熟悉基本语法。另外，为了使用 WebSocket，推荐使用 gorilla/websocket 这个第三方库，它封装得很好，使用起来方便。

你可以通过下面命令安装：

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

go get github.com/gorilla/websocket

登录后复制

同时，准备一个简单的 HTML 页面来测试前端连接和服务端交互。

核心步骤：建立 WebSocket 连接

WebSocket 是基于 HTTP 协议升级而来的一种全双工通信协议。在 Golang 中，使用 gorilla/websocket 可以很方便地处理升级请求并建立连接。

示例代码如下：

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域，测试时可以用，生产环境应谨慎设置
    },
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, _ := upgrader.Upgrade(w, r, nil)
    // 后续处理消息逻辑
}

登录后复制

然后注册路由：

http.HandleFunc("/ws", handleWebSocket)
http.ListenAndServe(":8080", nil)

登录后复制

这样，服务端就准备好接受 WebSocket 连接了。

百度虚拟主播

百度智能云平台的一站式、灵活化的虚拟主播直播解决方案

126

查看详情

实现消息广播机制

单个连接能收发消息还不够，我们希望多个用户能互相看到彼此的消息。这就需要一个“广播”机制：当某个客户端发送消息时，服务端把这个消息转发给所有已连接的客户端。

实现思路大致如下：

维护一个客户端连接池（比如 map[*websocket.Conn]bool）
收到消息后遍历连接池，向每个连接发送消息
使用 goroutine 处理并发读写

关键点在于并发安全，所以需要加锁或者使用 channel 来协调读写操作。

例如，可以定义一个全局的 clients map 和一个 broadcast channel：

var (
    clients   = make(map[*websocket.Conn]bool)
    broadcast = make(chan Message)
)

// 接收消息的 goroutine
for {
    var msg Message
    err := conn.ReadJSON(&msg)
    if err != nil {
        break
    }
    broadcast <- msg
}

登录后复制

然后有一个单独的 goroutine 负责监听 broadcast channel 并将消息推送给所有客户端。

前端连接 WebSocket 并展示消息

前端部分其实也很简单，只需要创建 WebSocket 实例，监听消息事件，并把收到的内容显示在页面上。

HTML 示例：

<script>
const ws = new WebSocket("ws://localhost:8080/ws");

ws.onmessage = function(event) {
    const msg = document.createElement("div");
    msg.textContent = event.data;
    document.body.appendChild(msg);
};

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

<input id="input" />
<button onclick="sendMessage()">发送</button>

登录后复制

这样前后端就打通了，可以进行实时通信了。

基本上就这些。整个过程不难，但要注意连接管理和并发控制。如果你只是做个 Demo 或者练手项目，这套结构完全够用了。

以上就是Golang如何开发简单聊天程序基于websocket实现双向通信的详细内容，更多请关注php中文网其它相关文章！