优化proto编码

2023-02-09 17:47:25 +08:00 · 2023-02-09 17:47:25 +08:00 · e8ba87e806
parent d4eeef991c
commit e8ba87e806
2 changed files with 8 additions and 9 deletions
--- a/tcp/gnetserver.go
+++ b/tcp/gnetserver.go
@ -82,13 +82,12 @@ func (s *GnetServer) OnClose(c gnet.Conn, err error) (action gnet.Action) {
 }

 func (s *GnetServer) OnTraffic(c gnet.Conn) (action gnet.Action) {
-
-	var (
-		err error
-		pkg YHProto.EchoPackage
-	)
-
 	for {
+		var (
+			err error
+			pkg YHProto.EchoPackage
+		)
+
 		err = pkg.DecodeGnet(c)
 		if err == YHProto.ErrNotEnoughStream {
 			break
--- a/ws/gnetserver.go
+++ b/ws/gnetserver.go
@ -104,11 +104,11 @@ func (s *GnetServer) OnTraffic(c gnet.Conn) (action gnet.Action) {
 	if messages == nil {
 		return gnet.None
 	}
-	var (
-		pkg YHProto.EchoPackage
-	)

 	for _, message := range messages {
+		var (
+			pkg YHProto.EchoPackage
+		)
 		switch message.OpCode {
 		case ws.OpContinuation:
 		case ws.OpText: