feat：完善信息，增加socket

app/service/websocket.go

@@ -0,0 +1,138 @@
+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),
+	}
+}