From e8ba87e806a5e5273e996b6253ee3696565659b5 Mon Sep 17 00:00:00 2001
From: mic <562710164>
Date: Thu, 9 Feb 2023 17:47:25 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96proto=E7=BC=96=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tcp/gnetserver.go | 11 +++++------
 ws/gnetserver.go  |  6 +++---
 2 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/tcp/gnetserver.go b/tcp/gnetserver.go
index 69680b3..62b7bc9 100644
--- 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
diff --git a/ws/gnetserver.go b/ws/gnetserver.go
index c49f95d..274a34f 100644
--- 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: