package chat.server.netty;
|
|
import chat.server.Configer;
|
import chat.server.im.IMDispatcher;
|
import chat.server.moquette.BytesMetricsCollector;
|
import chat.server.moquette.MessageMetricsCollector;
|
import chat.server.moquette.MqttBytesMetricsHandler;
|
import chat.server.moquette.MqttDecodHandler;
|
import chat.server.moquette.MqttEncodHandler;
|
import chat.server.moquette.MqttIdleTimeoutHandler;
|
import chat.server.moquette.MqttMessageLoggerHandler;
|
import chat.server.moquette.MqttMessageMetricsHandler;
|
import io.netty.channel.ChannelHandler;
|
import io.netty.channel.ChannelInitializer;
|
import io.netty.channel.ChannelPipeline;
|
import io.netty.channel.socket.SocketChannel;
|
import io.netty.handler.timeout.IdleStateHandler;
|
|
public class NettySSLChannelInitializer extends ChannelInitializer<SocketChannel> {
|
|
private MessageMetricsCollector messageMetricsCollector;
|
private BytesMetricsCollector bytesMetricsCollector;
|
|
|
@Override
|
protected void initChannel(SocketChannel channel) throws Exception {
|
int timeoutSeconds = Configer.getInteger("timeout", 10);
|
|
ChannelPipeline pipeline = channel.pipeline();
|
|
ChannelHandler bytesMetricsHandler = new MqttBytesMetricsHandler(bytesMetricsCollector);
|
ChannelHandler idleStateHandler = new IdleStateHandler(timeoutSeconds, 0, 0);
|
ChannelHandler timeoutHandler = new MqttIdleTimeoutHandler();
|
// ChannelHandler sslHandler = new SSLHandler();
|
ChannelHandler decoder = new MqttDecodHandler();
|
ChannelHandler encoder = MqttEncodHandler.getInstance();
|
ChannelHandler messageMetricsHandler = new MqttMessageMetricsHandler(messageMetricsCollector);
|
ChannelHandler messageLogger = new MqttMessageLoggerHandler();
|
ChannelHandler dispatcher = new IMDispatcher();
|
|
pipeline.addFirst("bytemetrics", bytesMetricsHandler);
|
pipeline.addLast("idleStateHandler", idleStateHandler);
|
pipeline.addAfter("idleStateHandler", "idleEventHandler", timeoutHandler);
|
// pipeline.addLast("ssl", sslHandler);
|
pipeline.addLast("decoder", decoder);
|
pipeline.addLast("encoder", encoder);
|
pipeline.addLast("metrics", messageMetricsHandler);
|
pipeline.addLast("messageLogger", messageLogger);
|
pipeline.addLast("handler", dispatcher);
|
}
|
|
public void setMessageMetricsCollector(MessageMetricsCollector messageMetricsCollector) {
|
this.messageMetricsCollector = messageMetricsCollector;
|
}
|
|
public void setBytesMetricsCollector(BytesMetricsCollector bytesMetricsCollector) {
|
this.bytesMetricsCollector = bytesMetricsCollector;
|
}
|
}
|
