Netty4ClientHandler.java

package com.ustcinfo.rpc.netty4;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;

import java.io.IOException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.ustcinfo.rpc.ResponseWrapper;

public class Netty4ClientHandler extends SimpleChannelInboundHandler<Object> {
	
	private static final Log LOGGER = LogFactory.getLog(Netty4ClientHandler.class);
	
	private static final boolean isDebugEnabled = LOGGER.isDebugEnabled();
	
	private Netty4ClientFactory factory;
	
	private String key;
	
	private Netty4Client client;
	
	public Netty4ClientHandler(Netty4ClientFactory factory,String key){
		this.factory = factory;
		this.key = key;
	}
	
	public void setClient(Netty4Client client){
		this.client = client;
	}

	public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause)
			throws Exception {
		if (!(cause instanceof IOException)) {
			LOGGER.error("catch some exception not IOException", cause);
		}
	}

	public void channelInactive(ChannelHandlerContext ctx) throws Exception {
		LOGGER.warn("connection closed: " + ctx.channel().remoteAddress());
		factory.removeClient(key);
	}

	@Override
	protected void channelRead0(ChannelHandlerContext ctx, Object msg)
			throws Exception {
		if (msg instanceof ResponseWrapper) {
			ResponseWrapper response = (ResponseWrapper) msg;
			if (isDebugEnabled) {
				// for performance trace
				LOGGER.debug("receive response list from server: "
						+ ctx.channel().remoteAddress() + ",request is:"
						+ response.getRequestId());
			}
			client.putResponse(response);
		} else {
			LOGGER.error("receive message error,only support List || ResponseWrapper");
			throw new Exception(
					"receive message error,only support List || ResponseWrapper");
		}

	}

}