From 382f214ec73e9976aafe9bb68ed6d1c3f58fecbc Mon Sep 17 00:00:00 2001
From: 37Sir <827412104@qq.com>
Date: Tue, 8 Jun 2021 11:09:32 +0800
Subject: [PATCH] =?UTF-8?q?pref[net]:=E8=A7=A3=E7=A0=81=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../net/handler/codec/tcp/TcpPacketCodecHandler.java     | 8 ++++----
 .../handler/codec/websocket/WebSocketCodecHandler.java   | 9 +++++----
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/net/src/main/java/com/zfoo/net/handler/codec/tcp/TcpPacketCodecHandler.java b/net/src/main/java/com/zfoo/net/handler/codec/tcp/TcpPacketCodecHandler.java
index e3e3d96c..0ef45430 100644
--- a/net/src/main/java/com/zfoo/net/handler/codec/tcp/TcpPacketCodecHandler.java
+++ b/net/src/main/java/com/zfoo/net/handler/codec/tcp/TcpPacketCodecHandler.java
@@ -52,9 +52,8 @@ public class TcpPacketCodecHandler extends ByteToMessageCodec<EncodedPacketInfo>
 
         // 如果长度非法，则抛出异常断开连接
         if (length < 0) {
-            throw new IllegalArgumentException(StringUtils
-                    .format("[session:{}]的包头长度[length:{}]非法"
-                            , SessionUtils.sessionInfo(ctx), length));
+            throw new IllegalArgumentException(StringUtils.format("[session:{}]的包头长度[length:{}]非法"
+                    , SessionUtils.sessionInfo(ctx), length));
         }
 
         // ByteBuf里的数据太小
@@ -63,8 +62,9 @@ public class TcpPacketCodecHandler extends ByteToMessageCodec<EncodedPacketInfo>
             return;
         }
 
-        var tmpByteBuf = in.readRetainedSlice(length);
+        ByteBuf tmpByteBuf = null;
         try {
+            tmpByteBuf = in.readRetainedSlice(length);
             DecodedPacketInfo packetInfo = NetContext.getPacketService().read(tmpByteBuf);
             out.add(packetInfo);
         } catch (Exception e) {
diff --git a/net/src/main/java/com/zfoo/net/handler/codec/websocket/WebSocketCodecHandler.java b/net/src/main/java/com/zfoo/net/handler/codec/websocket/WebSocketCodecHandler.java
index b7284e44..5d08982c 100644
--- a/net/src/main/java/com/zfoo/net/handler/codec/websocket/WebSocketCodecHandler.java
+++ b/net/src/main/java/com/zfoo/net/handler/codec/websocket/WebSocketCodecHandler.java
@@ -60,9 +60,8 @@ public class WebSocketCodecHandler extends MessageToMessageCodec<WebSocketFrame,
 
         // 如果长度非法，则抛出异常断开连接
         if (length < 0) {
-            throw new IllegalArgumentException(StringUtils
-                    .format("[session:{}]的包头长度[length:{}]非法"
-                            , SessionUtils.sessionInfo(channelHandlerContext), length));
+            throw new IllegalArgumentException(StringUtils.format("[session:{}]的包头长度[length:{}]非法"
+                    , SessionUtils.sessionInfo(channelHandlerContext), length));
         }
 
         // ByteBuf里的数据太小
@@ -70,8 +69,10 @@ public class WebSocketCodecHandler extends MessageToMessageCodec<WebSocketFrame,
             in.resetReaderIndex();
             return;
         }
-        var tmpByteBuf = in.readRetainedSlice(length);
+
+        ByteBuf tmpByteBuf = null;
         try {
+            tmpByteBuf = in.readRetainedSlice(length);
             DecodedPacketInfo packetInfo = NetContext.getPacketService().read(tmpByteBuf);
             list.add(packetInfo);
         } catch (Exception e) {