com.lorne.sds.server.socket.handler.SocketHandler Maven / Gradle / Ivy

Go to download
package com.lorne.sds.server.socket.handler;


import com.lorne.sds.server.service.SocketService;
import com.lorne.sds.server.utils.SocketManager;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;

import java.util.concurrent.Executor;
import java.util.concurrent.Executors;


/**
 * Created by lorne on 2017/8/12.
 */
public class SocketHandler extends ChannelInboundHandlerAdapter {


    private Executor threadPool = Executors.newFixedThreadPool(100);

    private SocketService socketService;


    public SocketHandler(SocketService socketService) {
        this.socketService = socketService;
    }


    @Override
    public void channelRead(final ChannelHandlerContext ctx,final Object msg) { // (2)
        // Discard the received data silently.
        threadPool.execute(new Runnable() {
            @Override
            public void run() {
                String uniqueKey = ctx.channel().remoteAddress().toString();

                socketService.getSocketEventService().onReadListener(ctx,uniqueKey,msg);
            }
        });

    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        if(SocketManager.getInstance().isAllowConnection()){
            SocketManager.getInstance().addClient(ctx.channel());

            String uniqueKey = ctx.channel().remoteAddress().toString();

            //将数据存储到distribute下
            socketService.create(uniqueKey);

            socketService.getSocketEventService().onConnectionListener(ctx,uniqueKey);

        }else{
            ctx.close();
        }
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        SocketManager.getInstance().removeClient(ctx.channel());

        String uniqueKey = ctx.channel().remoteAddress().toString();

        //将数据从distribute下异常
        socketService.remove(uniqueKey);

        socketService.getSocketEventService().onDisConnectionListener(ctx,uniqueKey);

    }

    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        //心跳配置
        if (IdleStateEvent.class.isAssignableFrom(evt.getClass())&& socketService.getSocketEventService().hasOpenHeartCheck()) {
            IdleStateEvent event = (IdleStateEvent) evt;
            String uniqueKey = ctx.channel().remoteAddress().toString();
            if (event.state() == IdleState.READER_IDLE) {
                //表示已经多久没有收到数据了

                socketService.getSocketEventService().onHeartNoReadDataListener(ctx,uniqueKey);

            } else if (event.state() == IdleState.WRITER_IDLE) {
                //表示已经多久没有发送数据了

                socketService.getSocketEventService().onHeartNoWriteDataListener(ctx,uniqueKey);

            } else if (event.state() == IdleState.ALL_IDLE) {
                //表示已经多久既没有收到也没有发送数据了

            }
        }
    }
}