客服系统即时通讯IM开发（三）访客实现一对一聊天-访客生成唯一id标识存储到全局变量【唯一客服】网站在线客服系统

package main

import (
    "fmt"

    "github.com/satori/go.uuid"
)

func main() {
    u := uuid.NewV4()
    fmt.Printf("UUID: %s\n", u)
}

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[string]*websocket.Conn)

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
        }

        // 获取访客 ID
        visitorID := c.Query("visitor_id")

        // 将连接保存到全局变量中
        connections[visitorID] = conn

        for {
            // 读取消息
            _, message, err := conn.ReadMessage()
            if err != nil {
                delete(connections, visitorID)
                break
            }

            // 发送消息
            err = conn.WriteMessage(websocket.TextMessage, message)
            if err != nil {
                delete(connections, visitorID)
                break
            }
        }
    })

    

var connections = make(map[string]*websocket.Conn)