You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
go-fly/ws/user.go

149 lines
3.2 KiB

4 years ago
package ws
import (
"encoding/json"
"github.com/gin-gonic/gin"
"github.com/gorilla/websocket"
"github.com/taoshihan1991/imaptool/models"
"github.com/taoshihan1991/imaptool/tools"
"log"
4 years ago
"time"
)
func NewKefuServer(c *gin.Context) {
kefuId, _ := c.Get("kefu_id")
kefuInfo := models.FindUserById(kefuId)
if kefuInfo.ID == 0 {
c.JSON(200, gin.H{
"code": 400,
"msg": "用户不存在",
})
return
}
//go kefuServerBackend()
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
if err != nil {
log.Print("upgrade:", err)
return
}
4 years ago
//获取GET参数,创建WS
var kefu User
kefu.Id = kefuInfo.Name
kefu.Name = kefuInfo.Nickname
kefu.Avator = kefuInfo.Avator
kefu.Role_id = kefuInfo.RoleId
4 years ago
kefu.Conn = conn
AddKefuToList(&kefu)
for {
//接受消息
var receive []byte
messageType, receive, err := conn.ReadMessage()
if err != nil {
log.Println(err)
go SendPingToKefuClient()
return
}
message <- &Message{
conn: conn,
content: receive,
context: c,
messageType: messageType,
}
}
}
func AddKefuToList(kefu *User) {
var newKefuConns = []*User{kefu}
kefuConns := KefuList[kefu.Id]
if kefuConns != nil {
for _, otherKefu := range kefuConns {
msg := TypeMessage{
Type: "many pong",
}
str, _ := json.Marshal(msg)
err := otherKefu.Conn.WriteMessage(websocket.TextMessage, str)
if err == nil {
newKefuConns = append(newKefuConns, otherKefu)
}
}
}
log.Println("xxxxxxxxxxxxxxxxxxxxxxxx", newKefuConns)
KefuList[kefu.Id] = newKefuConns
}
//给超管发消息
func SuperAdminMessage(str []byte) {
return
//给超管发
for _, kefuUsers := range KefuList {
for _, kefuUser := range kefuUsers {
if kefuUser.Role_id == "2" {
kefuUser.Conn.WriteMessage(websocket.TextMessage, str)
}
}
}
}
//给指定客服发消息
func OneKefuMessage(toId string, str []byte) {
//新版
4 years ago
mKefuConns, ok := KefuList[toId]
if mKefuConns != nil && ok {
for _, kefu := range mKefuConns {
4 years ago
log.Println("OneKefuMessage lock")
kefu.Mux.Lock()
defer kefu.Mux.Unlock()
log.Println("OneKefuMessage unlock")
4 years ago
error := kefu.Conn.WriteMessage(websocket.TextMessage, str)
4 years ago
tools.Logger().Println("send_kefu_message", error, string(str))
}
}
4 years ago
SuperAdminMessage(str)
}
4 years ago
func KefuMessage(visitorId, content string, kefuInfo models.User) {
msg := TypeMessage{
Type: "message",
Data: ClientMessage{
Name: kefuInfo.Nickname,
Avator: kefuInfo.Avator,
Id: visitorId,
Time: time.Now().Format("2006-01-02 15:04:05"),
ToId: visitorId,
Content: content,
IsKefu: "yes",
},
}
str, _ := json.Marshal(msg)
OneKefuMessage(kefuInfo.Name, str)
}
//给客服客户端发送消息判断客户端是否在线
func SendPingToKefuClient() {
msg := TypeMessage{
Type: "many pong",
}
str, _ := json.Marshal(msg)
for kefuId, kfConns := range KefuList {
var newKefuConns = []*User{}
for _, kefuConn := range kfConns {
if kefuConn == nil {
continue
}
4 years ago
kefuConn.Mux.Lock()
defer kefuConn.Mux.Unlock()
err := kefuConn.Conn.WriteMessage(websocket.TextMessage, str)
if err == nil {
newKefuConns = append(newKefuConns, kefuConn)
}
}
if len(newKefuConns) > 0 {
KefuList[kefuId] = newKefuConns
} else {
delete(KefuList, kefuId)
}
}
}