com.mengweifeng.nettysupport.ServerHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mengwefieng.support Show documentation
mengweifeng.support
The newest version!
package com.mengweifeng.nettysupport;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.ChannelStateEvent;
import org.jboss.netty.channel.ExceptionEvent;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelUpstreamHandler;
import org.jboss.netty.channel.WriteCompletionEvent;
import org.jboss.netty.handler.codec.http.HttpRequest;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.stereotype.Component;

@Component("serverHandler")
public class ServerHandler extends SimpleChannelUpstreamHandler implements ApplicationListener , ApplicationContextAware {

	private static final Log log = LogFactory.getLog(ServerHandler.class);

	private ApplicationContext applicationContext;
	private Map httpHanlderMap = new HashMap();

    /** (non-Javadoc)
     * @see org.jboss.netty.channel.SimpleChannelUpstreamHandler#messageReceived(org.jboss.netty.channel.ChannelHandlerContext, org.jboss.netty.channel.MessageEvent)
     */
    @Override
    public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
        Object msg = e.getMessage();
        if (msg instanceof HttpRequest) {
        	HttpRequest req = (HttpRequest) msg;
        	String uri = req.getUri();
    		if(uri == null) {
    			ctx.getChannel().close();
    			return;
    		}
    		int i = uri.indexOf("?");
    		if(i != -1) uri = uri.substring(0, i);
    		HttpHandler handler = httpHanlderMap.get(uri);
    		if(handler != null) {
    			handler.process(ctx, req);
    		} else {
    			ctx.getChannel().close();
    		}
        } else {
        	e.getChannel().close();
        }
    }

	/**
	 * (non-Javadoc)
	 * 
	 * @see org.jboss.netty.channel.SimpleChannelUpstreamHandler#exceptionCaught(org.jboss.netty.channel.ChannelHandlerContext,
	 *      org.jboss.netty.channel.ExceptionEvent)
	 */
	@Override
	public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) throws Exception {
		ctx.getChannel().close();
		Throwable t = e.getCause();
		if (t != null && ("Connection reset by peer".equalsIgnoreCase(t.getMessage()) || "Broken pipe".equalsIgnoreCase(t.getMessage()))) {
			return;
		}
		log.error("exceptionCaught", e.getCause());
	}

	/**
	 * Invoked when a {@link Channel} is open, but not bound nor connected. 

	 * 
	 * Be aware that this event is fired from within the Boss-Thread so
	 * you should not execute any heavy operation in there as it will block the
	 * dispatching to other workers!
	 */
	public void channelOpen(ChannelHandlerContext ctx, ChannelStateEvent e)
			throws Exception {
	}

	/**
	 * Invoked when a {@link Channel} is open and bound to a local address, but
	 * not connected. 

	 * 
	 * Be aware that this event is fired from within the Boss-Thread so
	 * you should not execute any heavy operation in there as it will block the
	 * dispatching to other workers!
	 */
	public void channelBound(ChannelHandlerContext ctx, ChannelStateEvent e)
			throws Exception {
	}

	/**
	 * Invoked when a {@link Channel} is open, bound to a local address, and
	 * connected to a remote address. 

	 * 
	 * Be aware that this event is fired from within the Boss-Thread so
	 * you should not execute any heavy operation in there as it will block the
	 * dispatching to other workers!
	 */
	public void channelConnected(ChannelHandlerContext ctx, ChannelStateEvent e)
			throws Exception {
	}

	/**
	 * Invoked when a {@link Channel}'s {@link Channel#getInterestOps()
	 * interestOps} was changed.
	 */
	public void channelInterestChanged(ChannelHandlerContext ctx,
			ChannelStateEvent e) throws Exception {
	}

	/**
	 * Invoked when a {@link Channel} was disconnected from its remote peer.
	 */
	public void channelDisconnected(ChannelHandlerContext ctx,
			ChannelStateEvent e) throws Exception {
	}

	/**
	 * Invoked when a {@link Channel} was unbound from the current local
	 * address.
	 */
	public void channelUnbound(ChannelHandlerContext ctx, ChannelStateEvent e)
			throws Exception {
	}

	/**
	 * Invoked when a {@link Channel} was closed and all its related resources
	 * were released.
	 */
	public void channelClosed(ChannelHandlerContext ctx, ChannelStateEvent e)
			throws Exception {
	}

	/**
	 * Invoked when something was written into a {@link Channel}.
	 */
	public void writeComplete(ChannelHandlerContext ctx, WriteCompletionEvent e)
			throws Exception {
	}

	@Override
	public void setApplicationContext(ApplicationContext applicationContext)
			throws BeansException {
		this.applicationContext = applicationContext;
	}

	@Override
	public void onApplicationEvent(ApplicationEvent event) {
		if(event instanceof ContextRefreshedEvent) {
			Map httpHandlers = applicationContext.getBeansWithAnnotation(Handler.class);
			for(Iterator