|
|
package websocketservers
|
|
|
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
"flx/Common"
|
|
|
"fmt"
|
|
|
"net"
|
|
|
"os"
|
|
|
"strings"
|
|
|
"sync"
|
|
|
"time"
|
|
|
|
|
|
"github.com/gorilla/websocket"
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
)
|
|
|
|
|
|
var messageList [][]byte //储存因为不在线未发送的信息
|
|
|
|
|
|
var IsMaster bool = true
|
|
|
var MasterSendID []string = []string{"meetman", "electron"}
|
|
|
|
|
|
// ClientManager is a websocketservers manager
|
|
|
// 客户端管理
|
|
|
type ClientManager struct {
|
|
|
MuLock sync.RWMutex
|
|
|
//客户端 map 储存并管理所有的长连接client，在线的为true，不在的为false
|
|
|
Clients map[*Client]bool
|
|
|
//web端发送来的的message我们用broadcast来接收，并最后分发给所有的client
|
|
|
Broadcast chan []byte
|
|
|
//新创建的长连接client
|
|
|
Register chan *Client
|
|
|
//新注销的长连接client
|
|
|
Unregister chan *Client
|
|
|
|
|
|
InvokeEventHandler func(uid string, ctype string, content string) //外部链接事件
|
|
|
|
|
|
InvokeEventHandlers func()
|
|
|
}
|
|
|
|
|
|
// Client is a websocketservers client
|
|
|
// 客户端 Client
|
|
|
type Client struct {
|
|
|
//用户id
|
|
|
Id string
|
|
|
//连接的socket
|
|
|
Socket *websocket.Conn
|
|
|
//发送的消息
|
|
|
Send chan []byte
|
|
|
}
|
|
|
|
|
|
var MClientID map[string]string
|
|
|
|
|
|
var FileData []map[string]interface{}
|
|
|
|
|
|
// Message is an object for websocketservers message which is mapped to json type
|
|
|
// 会把Message格式化成json
|
|
|
type Message struct {
|
|
|
//消息struct
|
|
|
ContentType string `json:"contenttype,omitempty"` //文本类型
|
|
|
Sender string `json:"sender,omitempty"` //发送者
|
|
|
Recipient string `json:"recipient,omitempty"` //接收者
|
|
|
Content interface{} `json:"datarecord,omitempty"` //内容
|
|
|
ContentTime string `json:"contenttime,omitempty"` //时间
|
|
|
}
|
|
|
|
|
|
// Manager define a ws server manager
|
|
|
// 创建客户端管理者
|
|
|
var Manager = ClientManager{
|
|
|
Broadcast: make(chan []byte),
|
|
|
Register: make(chan *Client),
|
|
|
Unregister: make(chan *Client),
|
|
|
Clients: make(map[*Client]bool),
|
|
|
}
|
|
|
|
|
|
func (manager *ClientManager) ManagerInit(invokeEventHandler func(uid string, ctype string, content string)) {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("ManagerInit:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
manager.InvokeEventHandler = invokeEventHandler
|
|
|
}
|
|
|
|
|
|
func (manager *ClientManager) ManagerInits(invokeEventHandler func()) {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("ManagerInits:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
manager.InvokeEventHandlers = invokeEventHandler
|
|
|
}
|
|
|
|
|
|
// Start is to start a ws server
|
|
|
func (manager *ClientManager) Start() {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("websocketserversStart:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
for {
|
|
|
select {
|
|
|
//如果有新的连接接入,就通过channel把连接传递给conn
|
|
|
case conn := <-manager.Register:
|
|
|
func() {
|
|
|
//把客户端的连接设置为true
|
|
|
manager.MuLock.Lock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("manager.MuLock:", err) // 打印错误信息
|
|
|
}
|
|
|
manager.MuLock.Unlock()
|
|
|
}()
|
|
|
manager.Clients[conn] = true
|
|
|
}()
|
|
|
|
|
|
//把返回连接成功的消息json格式化
|
|
|
// jsonMessage, _ := json.Marshal(&Message{ContentType: Common.WebSocketOpen, Content: "id:" + conn.Id + " has connected.", ContentTime: time.Now().Format("2006-01-02 15:04:05")})
|
|
|
tcpConn, ok := conn.Socket.UnderlyingConn().(*net.TCPConn)
|
|
|
if ok {
|
|
|
// log.Fatal("Failed to cast net.Conn to *net.TCPConn")
|
|
|
tcpConn.SetKeepAlive(true)
|
|
|
tcpConn.SetKeepAlivePeriod(3 * time.Second)
|
|
|
}
|
|
|
//调用客户端的send方法，发送消息
|
|
|
// manager.Send(jsonMessage, conn)
|
|
|
if manager.InvokeEventHandler != nil {
|
|
|
manager.InvokeEventHandler(conn.Id, "Online", "")
|
|
|
|
|
|
// nethelper.ServerSendSeatDataChange(conn.Id, "Online") // 测试无纸化在线离线
|
|
|
}
|
|
|
//如果连接断开了
|
|
|
case conn := <-manager.Unregister:
|
|
|
//判断连接的状态，如果是true,就关闭send，删除连接client的值
|
|
|
|
|
|
if _, ok := manager.Clients[conn]; ok {
|
|
|
if manager.InvokeEventHandler != nil {
|
|
|
manager.InvokeEventHandler(conn.Id, "Offline", "")
|
|
|
// nethelper.ServerSendSeatDataChange(conn.Id, "Offline") // 测试无纸化在线离线
|
|
|
}
|
|
|
close(conn.Send)
|
|
|
|
|
|
func() {
|
|
|
//把客户端的连接设置为true
|
|
|
manager.MuLock.Lock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("manager.MuLock:", err) // 打印错误信息
|
|
|
}
|
|
|
manager.MuLock.Unlock()
|
|
|
}()
|
|
|
delete(manager.Clients, conn)
|
|
|
}()
|
|
|
|
|
|
// jsonMessage, _ := json.Marshal(&Message{ContentType: Common.WebSocketClose, Content: "id:" + conn.Id + " has disconnected.", ContentTime: time.Now().Format("2006-01-02 15:04:05")})
|
|
|
// manager.Send(jsonMessage, conn)
|
|
|
}
|
|
|
//广播
|
|
|
case message := <-manager.Broadcast:
|
|
|
//遍历已经连接的客户端，把消息发送给他们
|
|
|
|
|
|
fmt.Println("33333333333-------")
|
|
|
func() {
|
|
|
//把客户端的连接设置为true
|
|
|
manager.MuLock.Lock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("manager.MuLock:", err) // 打印错误信息
|
|
|
}
|
|
|
manager.MuLock.Unlock()
|
|
|
}()
|
|
|
for conn := range manager.Clients {
|
|
|
|
|
|
select {
|
|
|
case conn.Send <- message:
|
|
|
default:
|
|
|
close(conn.Send)
|
|
|
delete(manager.Clients, conn)
|
|
|
}
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
fmt.Println("333333333333333333333")
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Send is to send ws message to ws client
|
|
|
// 定义客户端管理的send方法,屏蔽不需要发送的客户
|
|
|
func (manager *ClientManager) Send(message []byte, ignore *Client) {
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("WebSocketserverSend:", r)
|
|
|
// 可以在这里处理错误，比如设置 tempresultmap 之类的
|
|
|
}
|
|
|
}()
|
|
|
// fmt.Println("444444444----------")
|
|
|
manager.MuLock.RLock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("manager.MuLock:", err) // 打印错误信息
|
|
|
}
|
|
|
manager.MuLock.RUnlock()
|
|
|
}()
|
|
|
//只遍历键时，使用下面的形式：
|
|
|
for conn := range manager.Clients {
|
|
|
//不给屏蔽的连接发送消息
|
|
|
if conn != ignore {
|
|
|
conn.Send <- message
|
|
|
}
|
|
|
}
|
|
|
// fmt.Println("4444444444444444")
|
|
|
}
|
|
|
|
|
|
// 定义客户端管理的send方法,屏蔽不需要发送的客户
|
|
|
func (manager *ClientManager) SendMessageByClientId(message []byte, uid string) {
|
|
|
// fmt.Println("555555555555555-----------")
|
|
|
manager.MuLock.RLock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("manager.MuLock:", err) // 打印错误信息
|
|
|
log.Error("SendMessageByClientId:", err)
|
|
|
}
|
|
|
manager.MuLock.RUnlock()
|
|
|
// fmt.Println("55555555555555555")
|
|
|
}()
|
|
|
//只遍历键时，使用下面的形式：
|
|
|
for conn := range manager.Clients {
|
|
|
//发给指定用户的连接发送消息
|
|
|
if conn.Id == uid {
|
|
|
// // 打印消息发送前的日志，记录目标 IP 和 UID
|
|
|
// fmt.Printf("Sending message to client with UID: %s, IP: %s\n", uid, "打印消息发送前的日志")
|
|
|
|
|
|
// // 发送消息
|
|
|
// select {
|
|
|
// case conn.Send <- message:
|
|
|
// // 成功发送
|
|
|
// fmt.Println("Message sent successfully成功发送.")
|
|
|
// default:
|
|
|
// // 发送失败，管道阻塞
|
|
|
// fmt.Println("Failed to send message: Channel is blocked or client is not responsive发送失败，管道阻塞.")
|
|
|
// }
|
|
|
|
|
|
// // 发送后，确认已发送的消息
|
|
|
// fmt.Printf("发送后，确认已发送的消息Message sent to %s: %s\n", uid, string(message))
|
|
|
|
|
|
conn.Send <- message
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 定义客户端管理的send方法,屏蔽不需要发送的客户
|
|
|
func (manager *ClientManager) SendMessageByClientType(message []byte, uid string) {
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("manager.MuLock:", err) // 打印错误信息
|
|
|
log.Error("SendMessageByClientType:", err)
|
|
|
}
|
|
|
manager.MuLock.RUnlock()
|
|
|
// fmt.Println("55555555555555555")
|
|
|
}()
|
|
|
// fmt.Println("6666666666----------")
|
|
|
manager.MuLock.RLock()
|
|
|
//只遍历键时，使用下面的形式：
|
|
|
for conn := range manager.Clients {
|
|
|
//发给指定用户的连接发送消息
|
|
|
if strings.Contains(conn.Id, uid) {
|
|
|
conn.Send <- message
|
|
|
}
|
|
|
}
|
|
|
// fmt.Println("666666666666666666666")
|
|
|
}
|
|
|
|
|
|
// 定义客户端结构体的read方法
|
|
|
func (c *Client) Read() {
|
|
|
defer func() {
|
|
|
//触发关闭
|
|
|
Manager.Unregister <- c
|
|
|
c.Socket.Close()
|
|
|
if err := recover(); err != nil {
|
|
|
log.WithFields(log.Fields{
|
|
|
"function": "Read",
|
|
|
"error": err,
|
|
|
}).Error("Read")
|
|
|
log.Error("websocketserversRead:", err)
|
|
|
// log.Printf("OpenDevice: %v", err) // 记录错误信息到日志
|
|
|
fmt.Println("Read:", err) // 打印错误信息
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
for {
|
|
|
// c.Socket.SetReadDeadline(time.Now().Add(10 * time.Second)) // 设置读取超时为10秒
|
|
|
//读取消息
|
|
|
_, message, err := c.Socket.ReadMessage()
|
|
|
//如果有错误信息，就注销这个连接然后关闭
|
|
|
if err != nil {
|
|
|
Manager.Unregister <- c
|
|
|
c.Socket.Close()
|
|
|
break
|
|
|
}
|
|
|
//如果没有错误信息就把信息放入broadcast
|
|
|
// var jsonMessage []byte
|
|
|
// if message != nil && string(message) == Common.WebSocketHeartBeat {
|
|
|
// jsonMessage, _ = json.Marshal(&Message{ContentType: Common.WebSocketHeartBeat, Sender: c.Id, Content: string(message), ContentTime: time.Now().Format("2006-01-02 15:04:05")})
|
|
|
// } else {
|
|
|
// jsonMessage, _ = json.Marshal(&Message{ContentType: Common.WebSocketBroadCast, Sender: c.Id, Content: string(message), ContentTime: time.Now().Format("2006-01-02 15:04:05")})
|
|
|
// }
|
|
|
|
|
|
tempMap := make(map[string]interface{})
|
|
|
json.Unmarshal(message, &tempMap)
|
|
|
if tempMap["ContentType"] == Common.WebSocketChat {
|
|
|
jsonMessage, _ := json.Marshal(&Message{ContentType: Common.WebSocketChat, Sender: c.Id, Recipient: tempMap["Recipient"].(string), Content: tempMap["Content"].(string), ContentTime: time.Now().Format("2006-01-02 15:04:05")})
|
|
|
//Content 有要发送给谁
|
|
|
//发给李四
|
|
|
fmt.Println("77777777777----------")
|
|
|
|
|
|
func() {
|
|
|
//把客户端的连接设置为true
|
|
|
Manager.MuLock.RLock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("Manager.MuLock:", err) // 打印错误信息
|
|
|
}
|
|
|
Manager.MuLock.RUnlock()
|
|
|
}()
|
|
|
for k, v := range Manager.Clients {
|
|
|
if k.Id == tempMap["Recipient"].(string) {
|
|
|
for true {
|
|
|
if v == true {
|
|
|
messageList = append(messageList, jsonMessage)
|
|
|
for _, m := range messageList {
|
|
|
k.Send <- m
|
|
|
}
|
|
|
break
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
fmt.Println("777777777777777777777")
|
|
|
} else if tempMap["ContentType"] == Common.WebSocketRefresh {
|
|
|
onlineList := make([]string, 0)
|
|
|
|
|
|
fmt.Println("88888888888888----------")
|
|
|
func() {
|
|
|
//把客户端的连接设置为true
|
|
|
Manager.MuLock.RLock()
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
fmt.Println("Manager.MuLock:", err) // 打印错误信息
|
|
|
}
|
|
|
Manager.MuLock.RUnlock()
|
|
|
}()
|
|
|
for k, v := range Manager.Clients {
|
|
|
cli := k
|
|
|
if v {
|
|
|
onlineList = append(onlineList, cli.Id)
|
|
|
}
|
|
|
jsonMessage, _ := json.Marshal(&Message{ContentType: Common.WebSocketRefresh, Sender: c.Id, Content: onlineList, ContentTime: time.Now().Format("2006-01-02 15:04:05")})
|
|
|
c.Send <- jsonMessage
|
|
|
}
|
|
|
}()
|
|
|
fmt.Println("888888888888888888888888")
|
|
|
} else if tempMap["ContentType"] == Common.WebSocketCloseExe {
|
|
|
//浏览器关闭时调用退出会议
|
|
|
// localcache.DisposeContent()
|
|
|
// ServerSendElectronScreenClose()
|
|
|
if Manager.InvokeEventHandlers != nil {
|
|
|
Manager.InvokeEventHandlers()
|
|
|
}
|
|
|
time.Sleep(time.Second * 2)
|
|
|
os.Exit(0)
|
|
|
} else if tempMap["ContentType"] == Common.WebSocketCloseElectron {
|
|
|
//浏览器关闭时调用退出会议
|
|
|
// localcache.DisposeContent()
|
|
|
ServerSendElectronScreenClose()
|
|
|
if Manager.InvokeEventHandlers != nil {
|
|
|
Manager.InvokeEventHandlers()
|
|
|
}
|
|
|
} else if tempMap["ContentType"] == Common.WebSocketFileRate {
|
|
|
|
|
|
FileData = append(FileData, tempMap)
|
|
|
}
|
|
|
|
|
|
// log.Println(string(jsonMessage))
|
|
|
//触发消息发送
|
|
|
//Manager.Broadcast <- jsonMessage
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (c *Client) Write() {
|
|
|
defer func() {
|
|
|
c.Socket.Close()
|
|
|
|
|
|
if err := recover(); err != nil {
|
|
|
log.WithFields(log.Fields{
|
|
|
"function": "Write",
|
|
|
"error": err,
|
|
|
}).Error("Write")
|
|
|
log.Error("websocketserversWrite:", err)
|
|
|
|
|
|
// log.Printf("OpenDevice: %v", err) // 记录错误信息到日志
|
|
|
fmt.Println("Write:", err) // 打印错误信息
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
for {
|
|
|
select {
|
|
|
//从send里读消息
|
|
|
case message, ok := <-c.Send:
|
|
|
//如果没有消息
|
|
|
if !ok {
|
|
|
c.Socket.WriteMessage(websocket.CloseMessage, []byte{})
|
|
|
fmt.Println("发送信息错误")
|
|
|
return
|
|
|
}
|
|
|
//有消息就写入，发送给web端
|
|
|
// log.Println(string(message))
|
|
|
|
|
|
if IsMaster {
|
|
|
c.Socket.WriteMessage(websocket.TextMessage, message)
|
|
|
//fmt.Println("发送信息111：", MasterSendID, string(message))
|
|
|
} else {
|
|
|
for i := 0; i < len(MasterSendID); i++ {
|
|
|
if MasterSendID[i] == c.Id {
|
|
|
c.Socket.WriteMessage(websocket.TextMessage, message)
|
|
|
//fmt.Println("发送信息222：", MasterSendID, string(message))
|
|
|
break
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
// c.Socket.WriteMessage(websocket.TextMessage, message)
|
|
|
}
|
|
|
}
|
|
|
}
|
