You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.9 KiB
103 lines
3.9 KiB
package com.bx.imserver.netty;
|
|
|
|
import com.bx.imcommon.contant.IMRedisKey;
|
|
import com.bx.imcommon.enums.IMCmdType;
|
|
import com.bx.imcommon.model.IMSendInfo;
|
|
import com.bx.imserver.constant.ChannelAttrKey;
|
|
import com.bx.imserver.netty.processor.AbstractMessageProcessor;
|
|
import com.bx.imserver.netty.processor.ProcessorFactory;
|
|
import com.bx.imserver.util.SpringContextHolder;
|
|
import io.netty.channel.ChannelHandlerContext;
|
|
import io.netty.channel.SimpleChannelInboundHandler;
|
|
import io.netty.handler.timeout.IdleState;
|
|
import io.netty.handler.timeout.IdleStateEvent;
|
|
import io.netty.util.AttributeKey;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.data.redis.core.RedisTemplate;
|
|
|
|
|
|
/**
|
|
* WebSocket 长连接下 文本帧的处理器
|
|
* 实现浏览器发送文本回写
|
|
* 浏览器连接状态监控
|
|
*/
|
|
@Slf4j
|
|
public class IMChannelHandler extends SimpleChannelInboundHandler<IMSendInfo> {
|
|
|
|
/**
|
|
* 读取到消息后进行处理
|
|
*
|
|
* @param ctx
|
|
* @param sendInfo
|
|
* @throws Exception
|
|
*/
|
|
@Override
|
|
protected void channelRead0(ChannelHandlerContext ctx, IMSendInfo sendInfo) throws Exception {
|
|
// 创建处理器进行处理
|
|
AbstractMessageProcessor processor = ProcessorFactory.createProcessor(IMCmdType.fromCode(sendInfo.getCmd()));
|
|
processor.process(ctx,processor.transForm(sendInfo.getData()));
|
|
}
|
|
|
|
/**
|
|
* 出现异常的处理 打印报错日志
|
|
*
|
|
* @param ctx
|
|
* @param cause
|
|
* @throws Exception
|
|
*/
|
|
@Override
|
|
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
|
|
log.error(cause.getMessage());
|
|
//关闭上下文
|
|
//ctx.close();
|
|
}
|
|
|
|
/**
|
|
* 监控浏览器上线
|
|
*
|
|
* @param ctx
|
|
* @throws Exception
|
|
*/
|
|
@Override
|
|
public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
|
|
log.info(ctx.channel().id().asLongText() + "连接");
|
|
}
|
|
|
|
@Override
|
|
public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
|
|
AttributeKey<Long> userIdAttr = AttributeKey.valueOf(ChannelAttrKey.USER_ID);
|
|
Long userId = ctx.channel().attr(userIdAttr).get();
|
|
AttributeKey<Integer> terminalAttr = AttributeKey.valueOf(ChannelAttrKey.TERMINAL_TYPE);
|
|
Integer terminal = ctx.channel().attr(terminalAttr).get();
|
|
ChannelHandlerContext context = UserChannelCtxMap.getChannelCtx(userId,terminal);
|
|
// 判断一下，避免异地登录导致的误删
|
|
if(context != null && ctx.channel().id().equals(context.channel().id())){
|
|
// 移除channel
|
|
UserChannelCtxMap.removeChannelCtx(userId,terminal);
|
|
// 用户下线
|
|
RedisTemplate redisTemplate = SpringContextHolder.getBean("redisTemplate");
|
|
String key = String.join(":", IMRedisKey.IM_USER_SERVER_ID,userId.toString(), terminal.toString());
|
|
redisTemplate.delete(key);
|
|
log.info("断开连接,userId:{},终端类型:{}",userId,terminal);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
|
|
if (evt instanceof IdleStateEvent) {
|
|
IdleState state = ((IdleStateEvent) evt).state();
|
|
if (state == IdleState.READER_IDLE) {
|
|
// 在规定时间内没有收到客户端的上行数据, 主动断开连接
|
|
AttributeKey<Long> attr = AttributeKey.valueOf("USER_ID");
|
|
Long userId = ctx.channel().attr(attr).get();
|
|
AttributeKey<Integer> terminalAttr = AttributeKey.valueOf(ChannelAttrKey.TERMINAL_TYPE);
|
|
Integer ternimal = ctx.channel().attr(terminalAttr).get();
|
|
log.info("心跳超时，即将断开连接,用户id:{},终端类型:{} ",userId,ternimal);
|
|
ctx.channel().close();
|
|
}
|
|
} else {
|
|
super.userEventTriggered(ctx, evt);
|
|
}
|
|
|
|
}
|
|
}
|