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

我的客服系统使用的Golang+ Gin作为后端服务，所以下面的代码是演示demo

在 Go 语言中使用 Gin 框架实现 WebSocket 的方法如下：

1. 安装 gin-gonic/websocket 库。

2. 在 Gin 的路由处理函数中使用 ws.Upgrade 函数将 HTTP 连接升级为 WebSocket 连接。

3. 在路由处理函数中使用 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

十年开发经验程序员，离职全心创业中，历时三年开发出的产品《唯一客服系统》

一款基于Golang+Vue开发的在线客服系统，软件著作权编号：2021SR1462600。一套可私有化部署的网站在线客服系统，编译后的二进制文件可直接使用无需搭开发环境，下载zip解压即可，仅依赖MySQL数据库，是一个开箱即用的全渠道在线客服系统，致力于帮助广大开发者/公司快速部署整合私有化客服功能。
开源地址：唯一客服（开源学习版）
官网地址：唯一客服官网

来源：https://www.cnblogs.com/taoshihan/p/17032575.html
本站部分图文来源于网络，如有侵权请联系删除。