客服系统即时通讯IM开发（二）Golang + Gin框架实现服务端WebScoket【唯一客服】网站在线客服系统

2023年5月10日 118点热度 0人点赞 0条评论

我的客服系统使用的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