答案：本文介绍如何使用Golang和gorilla/websocket库实现WebSocket通信。首先通过Upgrader将HTTP连接升级为WebSocket，服务器监听/ws路径处理连接，读取并回显消息；客户端使用Dial连接服务器，启用Goroutine接收消息，主线程发送用户输入。实际应用中可用map和sync.Mutex管理多个连接，结合broadcast channel实现消息广播，充分发挥Go的高并发优势，适用于实时通信场景。
WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，广泛用于实时应用如聊天系统、实时通知和在线协作工具。Golang 因其高并发和轻量级 Goroutine 的特性，非常适合构建高性能的 WebSocket 服务。本文将介绍如何使用 Golang 实现 WebSocket 客户端与服务器之间的通信，并提供实用代码示例。
使用 gorilla/websocket 包
Go 标准库不直接支持 WebSocket，但社区最常用的第三方库是 gorilla/websocket。它提供了简洁的 API 来处理 WebSocket 连接、消息读写和连接管理。
安装方式：
go get github.com/gorilla/websocket实现 WebSocket 服务器
一个典型的 WebSocket 服务器需要监听 HTTP 升级请求，将连接从 HTTP 切换到 WebSocket 协议。
立即学习“go语言免费学习笔记（深入）”；
以下是一个简单的 WebSocket 服务器示例：
package main import ( "log" "net/http" "github.com/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, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Printf("升级失败: %v", err) return } defer conn.Close() for { // 读取消息 _, msg, err := conn.ReadMessage() if err != nil { log.Printf("读取失败: %v", err) break } log.Printf("收到消息: %s", msg) // 回显消息 if err := conn.WriteMessage(websocket.TextMessage, msg); err != nil { log.Printf("发送失败: %v", err) break } } } func main() { http.HandleFunc("/ws", handleWebSocket) log.Println("服务器启动在 :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }说明：
- upgrader.Upgrade 将 HTTP 请求升级为 WebSocket 连接。
- CheckOrigin 设为 true 可接受任意来源的请求，适合开发调试。
- 使用无限循环 for {} 持续读取客户端消息并回显。
实现 WebSocket 客户端
Go 也可以作为 WebSocket 客户端连接到服务器，适用于测试或集成其他服务。
示例客户端代码：
package main import ( "bufio" "fmt" "log" "os" "strings" "github.com/gorilla/websocket" ) func main() { conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil) if err != nil { log.Fatalf("连接失败: %v", err) } defer conn.Close() // 开启一个 Goroutine 接收服务器消息 go func() { for { _, msg, err := conn.ReadMessage() if err != nil { log.Printf("接收消息失败: %v", err) return } fmt.Printf("服务器: %s\n", msg) } }() // 主线程读取用户输入 scanner := bufio.NewScanner(os.Stdin) for scanner.Scan() { text := scanner.Text() if strings.ToLower(text) == "exit" { break } err := conn.WriteMessage(websocket.TextMessage, []byte(text)) if err != nil { log.Printf("发送消息失败: %v", err) return } } }说明：
- 使用 websocket.Dial 连接到服务器。
- 开启独立 Goroutine 处理接收逻辑，避免阻塞输入。
- 通过标准输入读取用户消息并发送。
连接管理与并发处理
在实际项目中，可能需要管理多个客户端连接，比如广播消息或维护会话状态。
常见做法是使用一个全局的 map[*websocket.Conn]bool 存储连接，并配合互斥锁保护。
例如：
var ( clients = make(map[*websocket.Conn]bool) broadcast = make(chan []byte) mu sync.Mutex )当新连接建立时加入 map，断开时删除。通过 broadcast channel 将消息推送给所有客户端。
基本上就这些。使用 gorilla/websocket 能快速搭建稳定可靠的 WebSocket 服务，结合 Go 的并发模型可轻松应对高并发场景。
