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

feat:ws握手同时认证

Browse Source
This commit is contained in:
zbzbzzz
2023-07-01 20:37:43 +08:00
parent d1ad40cadf
commit 46c0d4b771
2 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,7 +1,10 @@
package com.abin.mallchat.custom.user.websocket; package com.abin.mallchat.custom.user.websocket;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter; import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.handler.codec.http.DefaultFullHttpRequest;
import io.netty.handler.codec.http.FullHttpRequest; import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpHeaders; import io.netty.handler.codec.http.HttpHeaders;
import io.netty.util.AttributeKey; import io.netty.util.AttributeKey;
@@ -14,16 +17,27 @@ public class HttpHeadersHandler extends ChannelInboundHandlerAdapter {
@Override @Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
if (msg instanceof FullHttpRequest) { if (msg instanceof FullHttpRequest) {
HttpHeaders headers = ((FullHttpRequest) msg).headers(); FullHttpRequest request = (FullHttpRequest) msg;
HttpHeaders headers = request.headers();
String ip = headers.get("X-Real-IP"); String ip = headers.get("X-Real-IP");
if (StringUtils.isEmpty(ip)) {//如果没经过nginx，就直接获取远端地址 if (StringUtils.isEmpty(ip)) {//如果没经过nginx，就直接获取远端地址
InetSocketAddress address = (InetSocketAddress) ctx.channel().remoteAddress(); InetSocketAddress address = (InetSocketAddress) ctx.channel().remoteAddress();
ip = address.getAddress().getHostAddress(); ip = address.getAddress().getHostAddress();
} }
String[] uriSplit = request.uri().split("[?]");
FullHttpRequest modifiedRequest = new DefaultFullHttpRequest(request.protocolVersion(), request.method(), uriSplit[0]);
modifiedRequest.headers().add(headers);
modifiedRequest.content().writeBytes(request.content());
NettyUtil.setAttr(ctx.channel(), NettyUtil.IP, ip); NettyUtil.setAttr(ctx.channel(), NettyUtil.IP, ip);
String token = headers.get("token"); if (uriSplit.length>=2) {
NettyUtil.setAttr(ctx.channel(), NettyUtil.TOKEN, token); NettyUtil.setAttr(ctx.channel(), NettyUtil.TOKEN, uriSplit[1]);
}
ctx.fireChannelRead(modifiedRequest);
}else {
super.channelRead(ctx, msg);
} }
ctx.fireChannelRead(msg);
} }
} }

5
mallchat-custom-server/src/main/java/com/abin/mallchat/custom/user/websocket/NettyWebSocketServerHandler.java
View File

@@ -93,7 +93,10 @@ public class NettyWebSocketServerHandler extends SimpleChannelInboundHandler<Tex
log.info("请求二维码 = " + msg.text()); log.info("请求二维码 = " + msg.text());
break; break;
case HEARTBEAT: case HEARTBEAT:
log.info("心跳"); break;
case AUTHORIZE:
getService().authorize(ctx.channel(), JSONUtil.toBean(wsBaseReq.getData(), WSAuthorize.class));
log.info("主动认证 = " + msg.text());
break; break;
default: default:
log.info("未知类型"); log.info("未知类型");