如何在golang中實現websocket通信？使用標準庫net/http配合gorilla/websocket包可快速搭建高性能websocket服務端和客戶端。1. 建立連接需定義upgrader配置并通過upgrade方法升級http連接；2. 收發消息通過readmessage和writemessage方法實現，建議分別在goroutine中處理；3. 保持連接穩定可通過添加心跳機制、設置超時時間和自動重連機制實現；4. 客戶端連接使用websocket.dial發起請求，并同樣處理收發消息邏輯。
在golang中實現WebSocket通信，實現實時雙向數據傳輸其實并不復雜。標準庫net/http配合第三方庫比如gorilla/websocket可以非常方便地搭建一個高性能的WebSocket服務端和客戶端。本文會從幾個關鍵點入手，讓你快速掌握如何用Go構建WebSocket應用。
如何建立WebSocket連接？
WebSocket是一種基于TCP協議的全雙工通信協議，它通過HTTP握手升級為WebSocket連接。在Go中，最常用的是gorilla/websocket包，它封裝了握手、消息讀寫等流程。
要建立一個WebSocket服務器，首先需要一個HTTP處理函數：
立即學習“go語言免費學習筆記（深入）”；
var upgrader = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, CheckOrigin: func(r *http.Request) bool { return true // 允許跨域，生產環境建議配置具體域名 }, } func wsHandler(w http.ResponseWriter, r *http.Request) { conn, _ := upgrader.Upgrade(w, r, nil) // conn 是 *websocket.Conn 類型，可用于后續通信 }
然后注冊路由并啟動服務：
http.HandleFunc("/ws", wsHandler) log.Fatal(http.ListenAndServe(":8080", nil))
這樣就完成了基礎的WebSocket連接建立。
如何收發消息？
一旦連接建立完成，就可以使用conn.ReadMessage()和conn.WriteMessage()進行消息的收發。這兩個方法都是阻塞式的，適合放在goroutine中運行。
接收消息
接收消息通常在一個循環中持續監聽：
for { _, message, err := conn.ReadMessage() if err != nil { log.Println("read error:", err) break } fmt.Println("收到消息：", string(message)) }
發送消息
發送消息也很簡單，只需要調用WriteMessage，并指定消息類型（文本或二進制）：
err := conn.WriteMessage(websocket.TextMessage, []byte("Hello Client!")) if err != nil { log.Println("write error:", err) }
小技巧：如果希望同時支持收發，可以用兩個goroutine分別處理讀寫邏輯，避免阻塞。
如何保持連接穩定？
WebSocket連接容易因為網絡波動、心跳缺失等問題斷開。為了提高穩定性，有幾點建議：
- 添加心跳機制：定期發送ping/pong消息維持連接活躍狀態。
- 設置超時時間：可以通過SetReadDeadline來設置讀取超時，防止長時間無響應導致程序卡住。
- 自動重連機制（客戶端）：如果是客戶端連接，建議在網絡異常后嘗試重新連接。
示例代碼（服務端心跳）：
conn.SetReadLimit(512) conn.SetReadDeadline(time.Now().Add(60 * time.Second)) conn.SetPongHandler(func(appData string) error { log.Println("收到pong") return nil }) ticker := time.NewTicker(30 * time.Second) defer ticker.Stop() go func() { for { select { case <-ticker.C: err := conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(10*time.Second)) if err != nil { log.Println("ping error:", err) conn.Close() return } } } }()
客戶端怎么連接？
除了服務端，你也可以用Go寫一個WebSocket客戶端用于測試或者集成到其他系統中。
使用websocket.Dial即可發起連接：
url := "ws://localhost:8080/ws" conn, _, err := websocket.DefaultDialer.Dial(url, nil) if err != nil { log.Fatal("dial error:", err) } // 收消息 go func() { for { _, msg, err := conn.ReadMessage() if err != nil { log.Println("client read error:", err) return } fmt.Println("客戶端收到：", string(msg)) } }() // 發消息 conn.WriteMessage(websocket.TextMessage, []byte("Hello Server!"))
這種方式非常適合做本地調試、微服務間通信或自動化測試。
基本上就這些。WebSocket在Go中實現起來不難，但細節處理到位才能保證穩定性和擴展性。只要掌握了基本結構和常見問題處理方式，就能輕松構建實時通信功能。
