refactor(wscodec): 重构 WebSocket 编解码逻辑
- 将 WsCodec 结构体中的 upgraded 字段改为 Upgraded,提升代码可读性 - 移除未使用的 Isinitws 字段 - 优化了 Upgrade 方法,使用新命名的字段 - 调整了 ServerEvent 中的 OnTraffic 方法,简化了 WebSocket 消息处理逻辑 - 修改了 controller 中的 Recv 方法,增加了对 WebSocket 消息的处理支持
This commit is contained in:
@@ -2,14 +2,12 @@ package socket
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"blazing/common/data/entity"
|
||||
|
||||
"github.com/gobwas/ws/wsutil"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"github.com/panjf2000/gnet/v2"
|
||||
"github.com/panjf2000/gnet/v2/pkg/logging"
|
||||
@@ -96,7 +94,7 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
|
||||
s.handleTcp(c)
|
||||
return gnet.None
|
||||
} else {
|
||||
fmt.Println(ws.Buf.Bytes())
|
||||
// fmt.Println(ws.Buf.Bytes())
|
||||
c.Read(make([]byte, len1))
|
||||
}
|
||||
|
||||
@@ -110,19 +108,12 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
|
||||
if messages == nil {
|
||||
return
|
||||
}
|
||||
for _, message := range messages {
|
||||
msgLen := len(message.Payload)
|
||||
if msgLen > 128 {
|
||||
logging.Infof("conn[%v] receive [op=%v] [msg=%v..., len=%d]", c.RemoteAddr().String(), message.OpCode, string(message.Payload[:128]), len(message.Payload))
|
||||
} else {
|
||||
logging.Infof("conn[%v] receive [op=%v] [msg=%v, len=%d]", c.RemoteAddr().String(), message.OpCode, string(message.Payload), len(message.Payload))
|
||||
}
|
||||
// This is the echo server
|
||||
err = wsutil.WriteServerMessage(c, message.OpCode, message.Payload)
|
||||
if err != nil {
|
||||
logging.Infof("conn[%v] [err=%v]", c.RemoteAddr().String(), err.Error())
|
||||
return gnet.Close
|
||||
}
|
||||
|
||||
for _, msg := range messages {
|
||||
//client := conn.RemoteAddr().String()
|
||||
_ = s.workerPool.Submit(func() {
|
||||
s.parser(c, msg.Payload)
|
||||
})
|
||||
}
|
||||
|
||||
return gnet.None
|
||||
|
||||
Reference in New Issue
Block a user