From 0e23f887bd86d40abd4a7e6a6dc12fc8fcd4d2d6 Mon Sep 17 00:00:00 2001 From: limeng Date: Thu, 6 Jul 2023 21:51:16 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E9=BB=98=E8=AE=A4=E4=B8=8D?= =?UTF-8?q?=E4=BC=A0token=E5=87=BA=E7=8E=B0NPE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../abin/mallchat/custom/user/service/LoginService.java | 1 + .../custom/user/websocket/HttpHeadersHandler.java | 8 ++++++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/service/LoginService.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/service/LoginService.java index ed28384..3f57542 100644 --- a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/service/LoginService.java +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/service/LoginService.java @@ -38,4 +38,5 @@ public interface LoginService { * @return */ Long getValidUid(String token); + } diff --git a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/websocket/HttpHeadersHandler.java b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/websocket/HttpHeadersHandler.java index b6cdd43..5434151 100644 --- a/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/websocket/HttpHeadersHandler.java +++ b/mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/websocket/HttpHeadersHandler.java @@ -8,6 +8,7 @@ import io.netty.handler.codec.http.HttpHeaders; import org.apache.commons.lang3.StringUtils; import java.net.InetSocketAddress; +import java.util.Objects; public class HttpHeadersHandler extends ChannelInboundHandlerAdapter { @@ -18,8 +19,11 @@ public class HttpHeadersHandler extends ChannelInboundHandlerAdapter { UrlBuilder urlBuilder = UrlBuilder.ofHttp(request.uri()); // 获取token参数 - String token = urlBuilder.getQuery().get("token").toString(); - NettyUtil.setAttr(ctx.channel(), NettyUtil.TOKEN, token); + CharSequence sequence = urlBuilder.getQuery().get("token"); + if (Objects.nonNull(sequence)) { + String token = sequence.toString(); + NettyUtil.setAttr(ctx.channel(), NettyUtil.TOKEN, token); + } // 获取请求路径 request.setUri(urlBuilder.getPath().toString());