新增websocket服务器

helper/websocketclient.go

@@ -1,4 +1,4 @@
-package sysmodule
+package helper
 
 import (
 	"fmt"

helper/websocketserver.go

@@ -0,0 +1,123 @@
+package helper
+
+import (
+	"fmt"
+	"net/http"
+	"sync"
+
+	"github.com/gorilla/websocket"
+)
+
+type IWebsocketServer interface {
+	SendMsg(clientid uint64, messageType int, msg []byte) bool
+}
+
+type IMessageReceiver interface {
+	OnConnected(webServer IWebsocketServer, clientid uint64)
+	OnDisconnect(webServer IWebsocketServer, clientid uint64, err error)
+	OnRecvMsg(webServer IWebsocketServer, clientid uint64, msgtype int, data []byte)
+}
+
+type WSClient struct {
+	clientid  uint64
+	conn      *websocket.Conn
+	bwritemsg chan WSMessage
+}
+
+type WSMessage struct {
+	msgtype   int
+	bwritemsg []byte
+}
+
+type WebsocketServer struct {
+	wsUri       string
+	maxClientid uint64 //记录当前最新clientid
+	mapClient   map[uint64]*WSClient
+
+	pattern            string
+	port               uint16
+	bEnableCompression bool
+	locker             sync.Mutex
+	messageReciver     IMessageReceiver
+}
+
+func (slf *WebsocketServer) wsHandler(w http.ResponseWriter, r *http.Request) {
+	/*if r.Header.Get("Origin") != "http://"+r.Host {
+		http.Error(w, "Origin not allowed", 403)
+		return
+	}
+	*/
+	conn, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)
+	if err != nil {
+		http.Error(w, "Could not open websocket connection", http.StatusBadRequest)
+	}
+
+	slf.maxClientid++
+	pclient := &WSClient{slf.maxClientid, conn, make(chan WSMessage, 1024)}
+	slf.mapClient[pclient.clientid] = pclient
+	slf.messageReciver.OnConnected(slf, pclient.clientid)
+	go pclient.startSendMsg()
+	go slf.OnConnected(pclient)
+}
+
+func (slf *WebsocketServer) OnConnected(pclient *WSClient) {
+	for {
+		msgtype, message, err := pclient.conn.ReadMessage()
+		if err != nil {
+			pclient.conn.Close()
+			slf.locker.Lock()
+			defer slf.locker.Unlock()
+			delete(slf.mapClient, pclient.clientid)
+			slf.messageReciver.OnDisconnect(slf, pclient.clientid, err)
+			return
+		}
+
+		slf.messageReciver.OnRecvMsg(slf, pclient.clientid, msgtype, message)
+	}
+}
+
+func (slf *WebsocketServer) Init(pattern string, port uint16, messageReciver IMessageReceiver, bEnableCompression bool) {
+	slf.pattern = pattern
+	slf.port = port
+	slf.bEnableCompression = bEnableCompression
+	slf.mapClient = make(map[uint64]*WSClient)
+	slf.messageReciver = messageReciver
+
+	http.HandleFunc("/ws", slf.wsHandler)
+}
+
+func (slf *WebsocketServer) startListen() {
+	address := fmt.Sprintf(":%d", slf.port)
+	err := http.ListenAndServe(address, nil)
+	if err != nil {
+		fmt.Printf("%v", err)
+	}
+}
+
+func (slf *WSClient) startSendMsg() {
+	for {
+		msgbuf := <-slf.bwritemsg
+		slf.conn.WriteMessage(msgbuf.msgtype, msgbuf.bwritemsg)
+	}
+}
+
+func (slf *WebsocketServer) Start() {
+
+	go slf.startListen()
+}
+
+func (slf *WebsocketServer) SendMsg(clientid uint64, messageType int, msg []byte) bool {
+	slf.locker.Lock()
+	defer slf.locker.Unlock()
+	value, ok := slf.mapClient[clientid]
+	if ok == false {
+		return false
+	}
+
+	value.bwritemsg <- WSMessage{messageType, msg}
+	return true
+}
+
+func (slf *WebsocketServer) Stop() {
+
+}