Golang如何处理WebSocket客户端与服务器_Golang WebSocket通信实践-Golang-PHP中文网

Golang如何处理WebSocket客户端与服务器_Golang WebSocket通信实践

P粉602998670

发布： 2025-11-24 18:20:03

原创

261人浏览过

答案：本文介绍如何使用Golang和gorilla/websocket库实现WebSocket通信。首先通过Upgrader将HTTP连接升级为WebSocket，服务器监听/ws路径处理连接，读取并回显消息；客户端使用Dial连接服务器，启用Goroutine接收消息，主线程发送用户输入。实际应用中可用map和sync.Mutex管理多个连接，结合broadcast channel实现消息广播，充分发挥Go的高并发优势，适用于实时通信场景。

golang如何处理websocket客户端与服务器_golang websocket通信实践

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 客户端连接到服务器，适用于测试或集成其他服务。

Vheer

AI图像处理平台

125

查看详情

示例客户端代码：

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 } } }

说明：