我的客服系统使用的Golang+ Gin作为后端服务，所以下面的代码是演示demo
在 Go 语言中使用 Gin 框架实现 WebSocket 的方法如下：
-
安装 gin-gonic/websocket 库。
-
在 Gin 的路由处理函数中使用
ws.Upgrade函数将 HTTP 连接升级为 WebSocket 连接。 -
在路由处理函数中使用
conn.ReadMessage和conn.WriteMessage函数接收和发送消息。
例如：
package main import ( "net/http" "github.com/gin-gonic/gin" "github.com/gin-gonic/websocket" ) var upgrader = websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { return true }, } func main() { r := gin.Default() r.GET("/ws", func(c *gin.Context) { conn, err := upgrader.Upgrade(c.Writer, c.Request, nil) if err != nil { http.NotFound(c.Writer, c.Request) return } for { // 读取消息 _, message, err := conn.ReadMessage() if err != nil { break } // 发送消息 err = conn.WriteMessage(websocket.TextMessage, message) if err != nil { break } } }) r.Run() }
上面的代码中，我们在 /ws 路由处理函数中使用 upgrader.Upgrade 函数将 HTTP 连接升级为 WebSocket 连接。然后，我们使用 conn.ReadMessage 函数读取消息，使用 conn.WriteMessage 函数发送消息。
前端连接Conn保存
我们需要把前端链接过来的WebSocket Conn保存下来，这样才可以在其他地方去调用这个Conn发送消息
可以使用一个全局的变量来保存前端连接。
例如，你可以使用一个全局的 map 类型变量来保存前端连接，这样就可以在其他地方访问连接了。
例如：
package main import ( "net/http" "github.com/gin-gonic/gin" "github.com/gin-gonic/websocket" ) var upgrader = websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { return true }, } // 定义一个全局的 map 类型变量来保存前端连接 var connections = make(map[*websocket.Conn]bool) func main() { r := gin.Default() r.GET("/ws", func(c *gin.Context) { conn, err := upgrader.Upgrade(c.Writer, c.Request, nil) if err != nil { http.NotFound(c.Writer, c.Request) return } // 将连接保存到全局变量中 connections[conn] = true for { // 读取消息 _, message, err := conn.ReadMessage() if err != nil { delete(connections, conn) break } // 发送消息 err = conn.WriteMessage(websocket.TextMessage, message) if err != nil { delete(connections, conn) break } } }) r.Run() }
上面的代码中，我们在每个前端连接建立时将连接保存到全局变量中，在连接断开时将连接从全局变量中删除。
在其他地方访问保存的前端连接，可以使用下面的代码：
for conn := range connections { err := conn.WriteMessage(websocket.TextMessage, []byte("消息内容")) if err != nil { delete(connections, conn) } }
当然我们map全局变量，需要使用一个访客id标识来作为key，后面再详细说明