code.google.nfs.rpc.netty4.client.Netty4ClientHandler.java Source code

Introduction

Here is the source code for code.google.nfs.rpc.netty4.client.Netty4ClientHandler.java

Source

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

/**
 * nfs-rpc
 *   Apache License
 *   
 *   http://code.google.com/p/nfs-rpc (c) 2011
 */
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;

import java.io.IOException;
import java.util.List;

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

import code.google.nfs.rpc.ResponseWrapper;

/**
 * Netty4 Client Handler
 * 
 * @author <a href="mailto:coderplay@gmail.com">Min Zhou</a>
 */
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)) {
            // only log
            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, client);
    }

    @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");
        }

    }

}