实现简单聊天程序需搭建golang服务端与前端websocket连接。首先使用gorilla/websocket库升级http连接建立websocket通信；其次通过维护客户端连接池和广播机制实现消息分发；接着编写前端页面监听消息并展示；最后注意并发控制和连接管理以确保稳定性。
实现一个简单的聊天程序是学习 Golang 和 WebSocket 的好方式。用 Golang 搭个服务端，配合前端的 WebSocket 连接，就可以实现实时双向通信。这个过程并不复杂，只要掌握基本结构和流程就能快速上手。
准备工作：环境与依赖
在开始写代码之前，确保你已经安装好了 Go 环境，并且熟悉基本语法。另外，为了使用 WebSocket，推荐使用 gorilla/websocket 这个第三方库，它封装得很好，使用起来方便。
你可以通过下面命令安装：
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 连接了。
实现消息广播机制
单个连接能收发消息还不够，我们希望多个用户能互相看到彼此的消息。这就需要一个“广播”机制：当某个客户端发送消息时，服务端把这个消息转发给所有已连接的客户端。
实现思路大致如下：
- 维护一个客户端连接池（比如 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 <p>然后有一个单独的 goroutine 负责监听 broadcast channel 并将消息推送给所有客户端。</p><h2>前端连接 WebSocket 并展示消息</h2><p>前端部分其实也很简单，只需要创建 WebSocket 实例，监听消息事件，并把收到的内容显示在页面上。</p><p>HTML 示例：</p><pre class="brush:html;toolbar:false;"><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免费学习笔记（深入）：立即使用
在学习笔记中，你将探索golang的核心概念和高级技巧！
