websocket优化

xiangpei

2024-04-23 d4ef4a8eebc8b115bacd9b3170618fd024938042

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package com.ycl.websocket.handler;
 
import com.ycl.websocket.NettyConnect;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;
import io.netty.util.AttributeKey;
 
import java.util.Objects;
 
/**
 * @author：xp
 * @date：2024/4/19 14:41
 */
public class HeartBeatHandler extends ChannelInboundHandlerAdapter {
 
    private int lossConnectCount = 0;
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        if (evt instanceof IdleStateEvent){
            IdleStateEvent event = (IdleStateEvent)evt;
            if (event.state()== IdleState.READER_IDLE){
                lossConnectCount ++;
                if (lossConnectCount > 2){
                    AttributeKey<Long> userIdKey = AttributeKey.valueOf("userId");
                    Long userId = ctx.channel().attr(userIdKey).get();
                    if (Objects.nonNull(userId)) {
                        NettyConnect.getUserChannelMap().remove(userId);
                    }
                    NettyConnect.getChannelGroup().remove(ctx.channel());
                    ctx.channel().close();
                }
            }
        }else {
            super.userEventTriggered(ctx,evt);
        }
    }
 
}