Netty4ServerHandler.java

package code.google.nfs.rpc.netty4.server;

/**
 * nfs-rpc
 *   Apache License
 *   
 *   http://code.google.com/p/nfs-rpc (c) 2011
 */
import java.io.IOException;

import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.SimpleChannelInboundHandler;

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

import code.google.nfs.rpc.ProtocolFactory;
import code.google.nfs.rpc.RequestWrapper;
import code.google.nfs.rpc.ResponseWrapper;

/**
 * Netty4 Server Handler
 * 
 * @author <a href="mailto:coderplay@gmail.com">Min Zhou</a>
 */
public class Netty4ServerHandler extends SimpleChannelInboundHandler<Object> {

	private static final Log LOGGER = LogFactory
			.getLog(Netty4ServerHandler.class);

    /**
     * Calls {@link ChannelHandlerContext#fireExceptionCaught(Throwable)} to forward
     * to the next {@link ChannelHandler} in the {@link ChannelPipeline}.
     *
     * Sub-classes may override this method to change behavior.
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause)
            throws Exception {
		if(!(cause instanceof IOException)){
			// only log
			LOGGER.error("catch some exception not IOException", cause);
		}
    }
  
	@Override
	protected void channelRead0(ChannelHandlerContext ctx, Object msg)
			throws Exception {
		RequestWrapper request = (RequestWrapper) msg;
		long beginTime = System.currentTimeMillis();
		ResponseWrapper responseWrapper = ProtocolFactory.getServerHandler(
				request.getProtocolType()).handleRequest(request);
		final int id = request.getId();
		// already timeout,so not return
		if ((System.currentTimeMillis() - beginTime) >= request.getTimeout()) {
			LOGGER.warn("timeout,so give up send response to client,requestId is:"
					+ id
					+ ",client is:"
					+ ctx.channel().remoteAddress()
					+ ",consumetime is:"
					+ (System.currentTimeMillis() - beginTime)
					+ ",timeout is:"
					+ request.getTimeout());
			return;
		}
		ChannelFuture wf = ctx.channel().writeAndFlush(responseWrapper);
		wf.addListener(new ChannelFutureListener() {
			public void operationComplete(ChannelFuture future)
					throws Exception {
				if (!future.isSuccess()) {
					LOGGER.error("server write response error,request id is: "
							+ id);
				}
			}
		});
	}

}