hd/hdws/ws/ws.go

package ws

import (
	"encoding/json"
	"git.ouxuan.net/tommy/osmanthuswine/src/core"
	"gopkg.in/olahol/melody.v1"
	"math"
)

type WsCtl struct {
	core.WebSocket
	node *Node
}

func (ws *WsCtl) HandleConnect(s *melody.Session) {
	if ws.node == nil {
		ws.node = N
		ws.GetMelody().Config.MaxMessageSize = math.MaxInt16 * 10
		ws.GetMelody().Config.MessageBufferSize = math.MaxInt8
	}
	ws.node.rwmux.Lock()
	defer ws.node.rwmux.Unlock()
	query := s.Request.URL.Query()
	roomId := query.Get("activity_id")
	if _, ok := ws.node.rooms[roomId]; !ok { // 创建房间
		ws.node.rooms[roomId] = NewRoom(roomId)
	}

	client := NewClient(s, ws.node, roomId)

	var res = map[string]interface{}{}
	for k, v := range query {
		res[k] = v[0]
	}
	LoginMid(client, &Message{Data: res}) // 登录设置
	ws.node.conns[s] = client
	ws.node.handleMelodyConnect(client)
}

func (ws *WsCtl) HandleMessage(s *melody.Session, buf []byte) {
	msg := new(Message)
	err := json.Unmarshal(buf, msg)
	if err != nil {
		s.Write([]byte(err.Error()))
		return
	}

	client, ok := ws.node.conns[s] // 鉴定某个链接是否登录
	if !ok {
		WriteJsonWithSession(s, map[string]interface{}{
			"msg":  "链接无效",
			"code": 505,
		})
		//s.Write([]byte("链接无效"))
		return
	}

	ws.node.handleMelodyMessage(client, msg)
}

func (ws *WsCtl) HandleDisconnect(s *melody.Session) {
	if client, ok := ws.node.conns[s]; ok {
		ws.node.handleMelodyDisconnect(client)
	}
	ws.node.rwmux.Lock()
	defer ws.node.rwmux.Unlock()
	delete(ws.node.conns, s)
}

func (ws *WsCtl) HandleError(s *melody.Session, err error) {

}

func (ws *WsCtl) HandleMessageBinary(s *melody.Session, msg []byte) {

}

func (ws *WsCtl) HandlePong(s *melody.Session) {

}

func (ws *WsCtl) HandleSentMessage(s *melody.Session, msg []byte) {

}

func (ws *WsCtl) HandleSentMessageBinary(*melody.Session, []byte) {

}