open-source/MallChat
1
0
Fork 0
mirror of https://github.com/zongzibinbin/MallChat.git synced 2026-03-13 21:53:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

解决默认不传token出现NPE

Browse Source
This commit is contained in:
limeng
2023-07-06 21:51:16 +08:00
parent f63a6974d7
commit 0e23f887bd
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/service/LoginService.java
View File

@@ -38,4 +38,5 @@ public interface LoginService {
* @return * @return
*/ */
Long getValidUid(String token); Long getValidUid(String token);
} }

8
mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/websocket/HttpHeadersHandler.java
View File

@@ -8,6 +8,7 @@ import io.netty.handler.codec.http.HttpHeaders;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import java.net.InetSocketAddress; import java.net.InetSocketAddress;
import java.util.Objects;
public class HttpHeadersHandler extends ChannelInboundHandlerAdapter { public class HttpHeadersHandler extends ChannelInboundHandlerAdapter {
@@ -18,8 +19,11 @@ public class HttpHeadersHandler extends ChannelInboundHandlerAdapter {
UrlBuilder urlBuilder = UrlBuilder.ofHttp(request.uri()); UrlBuilder urlBuilder = UrlBuilder.ofHttp(request.uri());
// 获取token参数 // 获取token参数
String token = urlBuilder.getQuery().get("token").toString(); CharSequence sequence = urlBuilder.getQuery().get("token");
NettyUtil.setAttr(ctx.channel(), NettyUtil.TOKEN, token); if (Objects.nonNull(sequence)) {
String token = sequence.toString();
NettyUtil.setAttr(ctx.channel(), NettyUtil.TOKEN, token);
}
// 获取请求路径 // 获取请求路径
request.setUri(urlBuilder.getPath().toString()); request.setUri(urlBuilder.getPath().toString());