stu/MallChat
1
0
Fork 0
mirror of https://github.com/zongzibinbin/MallChat.git synced 2025-12-26 04:47:53 +08:00
Code Issues Actions 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
Show Stats 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
*/
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 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());