139 lines
2.6 KiB
Go
139 lines
2.6 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"SciencesServer/app/logic"
|
|||
|
|
"SciencesServer/serve/logger"
|
|||
|
|
"encoding/json"
|
|||
|
|
"github.com/gorilla/websocket"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Websocket struct {
|
|||
|
|
ID string
|
|||
|
|
conn *websocket.Conn
|
|||
|
|
send chan []byte
|
|||
|
|
time *time.Ticker
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Hub struct {
|
|||
|
|
clients map[string]*Websocket
|
|||
|
|
broadcast chan logic.INotice
|
|||
|
|
emit chan *HubEmit
|
|||
|
|
register chan *Websocket
|
|||
|
|
unregister chan *Websocket
|
|||
|
|
locker *sync.RWMutex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type HubEmit struct {
|
|||
|
|
ID string
|
|||
|
|
Msg logic.INotice
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Message struct {
|
|||
|
|
Prefix string `json:"prefix"`
|
|||
|
|
logic.INotice `json:"data"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var HubMessage *Hub
|
|||
|
|
|
|||
|
|
func (this *Hub) Run() {
|
|||
|
|
for {
|
|||
|
|
select {
|
|||
|
|
case client := <-this.register:
|
|||
|
|
this.clients[client.ID] = client
|
|||
|
|
logger.InfoF("客户端【%s】发起链接", client.ID)
|
|||
|
|
//for {
|
|||
|
|
// select {
|
|||
|
|
// case <-client.time.C:
|
|||
|
|
// fmt.Println("重置了")
|
|||
|
|
// client.time.Stop()
|
|||
|
|
// }
|
|||
|
|
//}
|
|||
|
|
case client := <-this.unregister:
|
|||
|
|
this.locker.Lock()
|
|||
|
|
|
|||
|
|
if _, ok := this.clients[client.ID]; ok {
|
|||
|
|
delete(this.clients, client.ID)
|
|||
|
|
close(client.send)
|
|||
|
|
}
|
|||
|
|
this.locker.Unlock()
|
|||
|
|
case message := <-this.broadcast:
|
|||
|
|
_message, _ := json.Marshal(&Message{
|
|||
|
|
Prefix: message.Prefix(),
|
|||
|
|
INotice: message,
|
|||
|
|
})
|
|||
|
|
for _, client := range this.clients {
|
|||
|
|
client.send <- _message
|
|||
|
|
}
|
|||
|
|
case iMsg := <-this.emit:
|
|||
|
|
client, has := this.clients[iMsg.ID]
|
|||
|
|
|
|||
|
|
if has {
|
|||
|
|
_message, _ := json.Marshal(&Message{
|
|||
|
|
Prefix: iMsg.Msg.Prefix(),
|
|||
|
|
INotice: iMsg.Msg,
|
|||
|
|
})
|
|||
|
|
client.send <- _message
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Hub) EmitHandle(iMsg *HubEmit) {
|
|||
|
|
this.emit <- iMsg
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Hub) BroadcastHandle(msg logic.INotice) {
|
|||
|
|
this.broadcast <- msg
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Hub) RegisterHandle(ws *Websocket) {
|
|||
|
|
this.register <- ws
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Hub) UnregisterHandle(ws *Websocket) {
|
|||
|
|
this.unregister <- ws
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewHub() *Hub {
|
|||
|
|
HubMessage = &Hub{
|
|||
|
|
clients: make(map[string]*Websocket),
|
|||
|
|
broadcast: make(chan logic.INotice),
|
|||
|
|
emit: make(chan *HubEmit),
|
|||
|
|
register: make(chan *Websocket),
|
|||
|
|
unregister: make(chan *Websocket),
|
|||
|
|
locker: new(sync.RWMutex),
|
|||
|
|
}
|
|||
|
|
return HubMessage
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Websocket) Write() {
|
|||
|
|
defer func() {
|
|||
|
|
this.conn.Close()
|
|||
|
|
}()
|
|||
|
|
for {
|
|||
|
|
select {
|
|||
|
|
case message, ok := <-this.send:
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
logger.InfoF("发送到客户端【%s】信息：%s", this.ID, string(message))
|
|||
|
|
_ = this.conn.WriteMessage(websocket.TextMessage, message)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Websocket) Read() {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewWebsocket(id string, conn *websocket.Conn) *Websocket {
|
|||
|
|
return &Websocket{
|
|||
|
|
ID: id,
|
|||
|
|
conn: conn,
|
|||
|
|
send: make(chan []byte),
|
|||
|
|
time: time.NewTicker(10 * time.Second),
|
|||
|
|
}
|
|||
|
|
}
|