133 lines
3.1 KiB
Go
133 lines
3.1 KiB
Go
|
package ws
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/gorilla/websocket"
|
||
|
|
"net/http"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Message struct {
|
||
|
|
conn *websocket.Conn
|
||
|
|
context *gin.Context
|
||
|
|
content []byte
|
||
|
|
messageType int
|
||
|
|
Mux sync.Mutex
|
||
|
|
}
|
||
|
|
type TypeMessage struct {
|
||
|
|
Type interface{} `json:"type"`
|
||
|
|
Data interface{} `json:"data"`
|
||
|
|
}
|
||
|
|
type ClientMessage struct {
|
||
|
|
MsgId uint `json:"msg_id"`
|
||
|
|
Name string `json:"name"`
|
||
|
|
Avator string `json:"avator"`
|
||
|
|
Id string `json:"id"`
|
||
|
|
VisitorId string `json:"visitor_id"`
|
||
|
|
Group string `json:"group"`
|
||
|
|
Time string `json:"time"`
|
||
|
|
ToId string `json:"to_id"`
|
||
|
|
Content string `json:"content"`
|
||
|
|
City string `json:"city"`
|
||
|
|
ClientIp string `json:"client_ip"`
|
||
|
|
Refer string `json:"refer"`
|
||
|
|
IsKefu string `json:"is_kefu"`
|
||
|
|
}
|
||
|
|
type SimpleMessage struct {
|
||
|
|
From string `json:"from"`
|
||
|
|
To string `json:"to"`
|
||
|
|
Content string `json:"content"`
|
||
|
|
}
|
||
|
|
|
||
|
|
//var Room = NewRoom()
|
||
|
|
var ClientList = make(map[string]*VisitorConnection)
|
||
|
|
|
||
|
|
var upgrader = websocket.Upgrader{
|
||
|
|
ReadBufferSize: 1024,
|
||
|
|
WriteBufferSize: 1024,
|
||
|
|
// 解决跨域问题
|
||
|
|
CheckOrigin: func(r *http.Request) bool {
|
||
|
|
return true
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
//var message = make(chan *Message, 100)
|
||
|
|
|
||
|
|
//var Mux sync.RWMutex
|
||
|
|
|
||
|
|
//func init() {
|
||
|
|
// upgrader = websocket.Upgrader{
|
||
|
|
// ReadBufferSize: 1024,
|
||
|
|
// WriteBufferSize: 1024,
|
||
|
|
// // 解决跨域问题
|
||
|
|
// CheckOrigin: func(r *http.Request) bool {
|
||
|
|
// return true
|
||
|
|
// },
|
||
|
|
// }
|
||
|
|
//}
|
||
|
|
|
||
|
|
//func SendServerJiang(title string, content string, domain string) string {
|
||
|
|
// noticeServerJiang, err := strconv.ParseBool(models.FindConfig("NoticeServerJiang"))
|
||
|
|
// serverJiangAPI := models.FindConfig("ServerJiangAPI")
|
||
|
|
// if err != nil || !noticeServerJiang || serverJiangAPI == "" {
|
||
|
|
// log.Println("do not notice serverjiang:", serverJiangAPI, noticeServerJiang)
|
||
|
|
// return ""
|
||
|
|
// }
|
||
|
|
// sendStr := fmt.Sprintf("%s%s", title, content)
|
||
|
|
// desp := title + ":" + content + "[登录](http://" + domain + "/main)"
|
||
|
|
// url := serverJiangAPI + "?text=" + sendStr + "&desp=" + desp
|
||
|
|
// //log.Println(url)
|
||
|
|
// res := tools.Get(url)
|
||
|
|
// return res
|
||
|
|
//}
|
||
|
|
|
||
|
|
//定时给更新数据库状态
|
||
|
|
func UpdateVisitorStatusCron() {
|
||
|
|
for {
|
||
|
|
SendPingToKefuClient()
|
||
|
|
time.Sleep(61 * time.Second)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//后端广播发送消息
|
||
|
|
//func WsServerBackend() {
|
||
|
|
// for {
|
||
|
|
// message := <-message
|
||
|
|
// var typeMsg TypeMessage
|
||
|
|
// json.Unmarshal(message.content, &typeMsg)
|
||
|
|
// conn := message.conn
|
||
|
|
// if typeMsg.Type == nil || typeMsg.Data == nil {
|
||
|
|
// continue
|
||
|
|
// }
|
||
|
|
// msgType := typeMsg.Type.(string)
|
||
|
|
//
|
||
|
|
// switch msgType {
|
||
|
|
// //心跳
|
||
|
|
// case "ping":
|
||
|
|
// msg := TypeMessage{
|
||
|
|
// Type: "pong",
|
||
|
|
// }
|
||
|
|
// str, _ := json.Marshal(msg)
|
||
|
|
// message.Mux.Lock()
|
||
|
|
// defer message.Mux.Unlock()
|
||
|
|
// conn.WriteMessage(websocket.TextMessage, str)
|
||
|
|
// case "inputing":
|
||
|
|
// data := typeMsg.Data.(map[string]interface{})
|
||
|
|
// from := data["from"].(string)
|
||
|
|
// to := data["to"].(string)
|
||
|
|
// //限流
|
||
|
|
// if tools.LimitFreqSingle("inputing:"+from, 1, 2) {
|
||
|
|
// OneKefuMessage(to, message.content)
|
||
|
|
// }
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// }
|
||
|
|
//}
|
||
|
|
func UpdateVisitorUser(visitorId string, toId string) {
|
||
|
|
if guest, ok := ClientList[visitorId]; ok {
|
||
|
|
guest.ToId = toId
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|