Golang如何搭建WebSocket服务_Golang WebSocket实时通信实践-Golang-PHP中文网

Golang如何搭建WebSocket服务_Golang WebSocket实时通信实践

P粉602998670

发布： 2025-11-13 18:40:03

原创

181人浏览过

答案：使用 gorilla/websocket 库可快速搭建 Go 语言 WebSocket 服务，通过 Hub 管理连接实现广播机制，配合前端测试验证通信，注意心跳、并发安全与资源清理等优化点。

golang如何搭建websocket服务_golang websocket实时通信实践

搭建 WebSocket 服务在 Go 语言中非常直观，得益于其强大的标准库和轻量级的并发模型。WebSocket 能实现客户端与服务端之间的全双工通信，非常适合实时聊天、通知推送、在线协作等场景。下面介绍如何用 Golang 快速实现一个可用的 WebSocket 服务。

使用 gorilla/websocket 库搭建基础服务

Go 标准库没有原生支持 WebSocket，但社区广泛使用 gorilla/websocket 这个第三方库。它功能完整、文档清晰，是构建 WebSocket 服务的首选。

安装方式：

go get github.com/gorilla/websocket

创建一个最简单的 WebSocket 服务端：

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

package main

import ( "log" "net/http" "github.com/gorilla/websocket" )

var upgrader = websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { return true // 允许跨域连接，生产环境应做更严格校验 }, }

func echoHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Print("升级失败:", err) return } defer conn.Close()

for {
    messageType, message, err := conn.ReadMessage()
    if err != nil {
        log.Print("读取消息失败:", err)
        break
    }

    log.Printf("收到消息: %s", message)
    if err := conn.WriteMessage(messageType, message); err != nil {
        log.Print("发送消息失败:", err)
        break
    }
}

登录后复制

}

func main() { http.HandleFunc("/ws", echoHandler) log.Println("服务启动在 :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }

这段代码实现了一个“回声”服务：客户端发送什么，服务端就原样返回。

实现广播机制支持多客户端通信

真实场景中通常需要多个客户端之间通信，比如群聊。这就需要维护连接池，并实现消息广播。

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

定义一个中心化的 Hub 来管理所有连接：

type Hub struct { clients map[websocket.Conn]bool broadcast chan []byte register chan websocket.Conn unregister chan *websocket.Conn }

func newHub() Hub { return &Hub{ clients: make(map[websocket.Conn]bool), broadcast: make(chan []byte), register: make(chan websocket.Conn), unregister: make(chan websocket.Conn), } }

func (h *Hub) run() { for { select { case conn := <-h.register: h.clients[conn] = true case conn := <-h.unregister: delete(h.clients, conn) close(conn) case message := <-h.broadcast: for conn := range h.clients { if err := conn.WriteMessage(websocket.TextMessage, message); err != nil { log.Print("广播失败:", err) conn.Close() delete(h.clients, conn) } } } } }

然后在 HTTP 处理函数中接入 Hub：

var hub = newHub()

func wsHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { return } defer conn.Close()

hub.register <- conn
defer func() { hub.unregister <- conn }()

for {
    _, message, err := conn.ReadMessage()
    if err != nil {
        break
    }
    hub.broadcast <- message
}

登录后复制

}

启动时运行 Hub：

func main() { go hub.run() http.HandleFunc("/ws", wsHandler) log.Println("服务启动在 :8080") log.Fatal(http.ListenAndServe(":8080", nil)) }

这样任意客户端发来的消息都会被广播给所有已连接的客户端。