package ws import ( "encoding/json" "github.com/gin-gonic/gin" "github.com/gorilla/websocket" "github.com/taoshihan1991/imaptool/models" "log" ) func NewVisitorServer(c *gin.Context) { //go kefuServerBackend() conn, err := upgrader.Upgrade(c.Writer, c.Request, nil) if err != nil { log.Print("upgrade:", err) return } //获取GET参数,创建WS vistorInfo := models.FindVisitorByVistorId(c.Query("visitor_id")) if vistorInfo.VisitorId == "" { c.JSON(200, gin.H{ "code": 400, "msg": "访客不存在", }) return } user := &User{ Conn: conn, Name: vistorInfo.Name, Avator: vistorInfo.Avator, Id: vistorInfo.VisitorId, To_id: vistorInfo.ToId, } go models.UpdateVisitorStatus(vistorInfo.VisitorId, 1) go SendServerJiang(vistorInfo.Name, "来了", c.Request.Host) AddVisitorToList(user) for { //接受消息 var receive []byte messageType, receive, err := conn.ReadMessage() if err != nil { for _, visitor := range ClientList { if visitor.Conn == conn { log.Println("删除用户", visitor.Id) delete(ClientList, visitor.Id) VisitorOffline(visitor.To_id, visitor.Id, visitor.Name) } } log.Println(err) return } message <- &Message{ conn: conn, content: receive, context: c, messageType: messageType, } } } func AddVisitorToList(user *User) { //用户id对应的连接 ClientList[user.Id] = user lastMessage := models.FindLastMessageByVisitorId(user.Id) userInfo := make(map[string]string) userInfo["uid"] = user.Id userInfo["username"] = user.Name userInfo["avator"] = user.Avator userInfo["last_message"] = lastMessage.Content if userInfo["last_message"] == "" { userInfo["last_message"] = "新访客" } msg := TypeMessage{ Type: "userOnline", Data: userInfo, } str, _ := json.Marshal(msg) //新版 OneKefuMessage(user.To_id, str) } func VisitorOnline(kefuId string, visitor models.Visitor) { lastMessage := models.FindLastMessageByVisitorId(visitor.VisitorId) userInfo := make(map[string]string) userInfo["uid"] = visitor.VisitorId userInfo["username"] = visitor.Name userInfo["avator"] = visitor.Avator userInfo["last_message"] = lastMessage.Content if userInfo["last_message"] == "" { userInfo["last_message"] = "新访客" } msg := TypeMessage{ Type: "userOnline", Data: userInfo, } str, _ := json.Marshal(msg) OneKefuMessage(kefuId, str) } func VisitorOffline(kefuId string, visitorId string, visitorName string) { models.UpdateVisitorStatus(visitorId, 0) userInfo := make(map[string]string) userInfo["uid"] = visitorId userInfo["name"] = visitorName msg := TypeMessage{ Type: "userOffline", Data: userInfo, } str, _ := json.Marshal(msg) //新版 OneKefuMessage(kefuId, str) } func VisitorNotice(visitorId string, notice string) { msg := TypeMessage{ Type: "notice", Data: notice, } str, _ := json.Marshal(msg) visitor := ClientList[visitorId] visitor.Conn.WriteMessage(websocket.TextMessage, str) }