设为首页 加入收藏

TOP

客服系统即时通讯IM开发(二)Golang + Gin框架实现服务端WebScoket【唯一客服】网站在线客服系统
2023-07-23 13:32:56 】 浏览:35
Tags:时通讯 开发 Golang Gin WebScoket

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

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

  1. 安装 gin-gonic/websocket 库。

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

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

】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇客服系统即时通讯IM开发(三)访.. 下一篇Go编程实战:博客备份

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目