95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package network
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/gobwas/ws"
|
|
"github.com/gobwas/ws/wsutil"
|
|
"github.com/kercylan98/minotaur/server/internal/v2"
|
|
"github.com/kercylan98/minotaur/utils/log"
|
|
"github.com/panjf2000/gnet/v2"
|
|
"time"
|
|
)
|
|
|
|
func newWebsocketHandler(core *websocketCore) *websocketHandler {
|
|
return &websocketHandler{
|
|
core: core,
|
|
}
|
|
}
|
|
|
|
type websocketHandler struct {
|
|
engine *gnet.Engine
|
|
upgrader ws.Upgrader
|
|
core *websocketCore
|
|
}
|
|
|
|
func (w *websocketHandler) OnBoot(eng gnet.Engine) (action gnet.Action) {
|
|
w.engine = &eng
|
|
w.initUpgrader()
|
|
return
|
|
}
|
|
|
|
func (w *websocketHandler) OnShutdown(eng gnet.Engine) {
|
|
|
|
}
|
|
|
|
func (w *websocketHandler) OnOpen(c gnet.Conn) (out []byte, action gnet.Action) {
|
|
wrapper := newWebsocketWrapper(w.core.ctx, c)
|
|
c.SetContext(wrapper)
|
|
w.core.core.OnConnectionOpened(wrapper.ctx, c, func(message server.server) error {
|
|
return wsutil.WriteServerMessage(c, message.GetContext().(ws.OpCode), message.GetBytes())
|
|
})
|
|
return
|
|
}
|
|
|
|
func (w *websocketHandler) OnClose(c gnet.Conn, err error) (action gnet.Action) {
|
|
wrapper := c.Context().(*websocketWrapper)
|
|
wrapper.cancel()
|
|
return
|
|
}
|
|
|
|
func (w *websocketHandler) OnTraffic(c gnet.Conn) (action gnet.Action) {
|
|
wrapper := c.Context().(*websocketWrapper)
|
|
|
|
// read to buffer
|
|
if err := wrapper.readToBuffer(); err != nil {
|
|
log.Error("websocket", log.Err(err))
|
|
return gnet.Close
|
|
}
|
|
|
|
// check or upgrade
|
|
if err := wrapper.upgrade(w.upgrader); err != nil {
|
|
log.Error("websocket", log.Err(err))
|
|
return gnet.Close
|
|
}
|
|
wrapper.active = time.Now()
|
|
|
|
// decode
|
|
messages, err := wrapper.decode()
|
|
if err != nil {
|
|
log.Error("websocket", log.Err(err))
|
|
}
|
|
|
|
for _, message := range messages {
|
|
packet := w.core.core.GeneratePacket(message.Payload)
|
|
packet.SetContext(message.OpCode)
|
|
w.core.core.OnReceivePacket(packet)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (w *websocketHandler) OnTick() (delay time.Duration, action gnet.Action) {
|
|
return
|
|
}
|
|
|
|
func (w *websocketHandler) initUpgrader() {
|
|
w.upgrader = ws.Upgrader{
|
|
OnRequest: func(uri []byte) (err error) {
|
|
if string(uri) != w.core.pattern {
|
|
err = errors.New("bad request")
|
|
}
|
|
return
|
|
},
|
|
}
|
|
}
|