我的客服系统使用的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，后面再详细说明
唯一在线客服系统
https://gofly.v1kf.com
发表评论