package chat.server.netty; 

import org.apache.log4j.Logger;

import chat.server.Configer;
import chat.server.moquette.BytesMetricsCollector;
import chat.server.moquette.MessageMetricsCollector;
import chat.server.moquette.MqttBytesMetricsHandler;
import chat.server.moquette.MqttDecodHandler;
import chat.server.moquette.MqttDispatcher;
import chat.server.moquette.MqttEncodHandler;
import chat.server.moquette.MqttIdleTimeoutHandler;
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 NettyTCPChannelInitializer extends ChannelInitializer<SocketChannel> {

//    private MessageMetricsCollector messageMetricsCollector;
    private BytesMetricsCollector bytesMetricsCollector;
    
    protected Logger logger = Logger.getLogger(NettyTCPChannelInitializer.class);
    
    
	@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 decoder = new MqttDecodHandler();
		ChannelHandler encoder = new MqttEncodHandler();
		//ChannelHandler messageMetricsHandler = new MqttMessageMetricsHandler(messageMetricsCollector);
		//ChannelHandler messageLogger = new MqttMessageLoggerHandler();
		ChannelHandler dispatcher = new MqttDispatcher();
		
		pipeline.addFirst("bytemetrics", bytesMetricsHandler);
		pipeline.addLast("idleStateHandler", idleStateHandler);
		pipeline.addAfter("idleStateHandler", "idleEventHandler", timeoutHandler);
		
		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;
	}

}