MainIOHandler.java

package com.farsunset.ichat.nio.handler;

import java.util.HashMap;

import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoHandlerAdapter;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;

import com.farsunset.ichat.nio.constant.CIMConstant;
import com.farsunset.ichat.nio.mutual.ReplyBody;
import com.farsunset.ichat.nio.mutual.SentBody;
import com.farsunset.ichat.nio.session.CIMSession;

/**
 * 
 * 客户端请求的入口，所有请求都首先经过它分发处理
 * 
 * @author farsunset (3979434@qq.com)
 */
public class MainIOHandler extends IoHandlerAdapter {

	protected final Logger logger = Logger.getLogger(MainIOHandler.class);

	private HashMap<String, CIMRequestHandler> handlers = new HashMap<String, CIMRequestHandler>();

	public void sessionCreated(IoSession session) throws Exception {
		logger.warn("sessionCreated()... from "
				+ session.getRemoteAddress().toString());
	}

	public void sessionOpened(IoSession session) throws Exception {

	}

	public void messageReceived(IoSession ios, Object message) throws Exception {
		logger.debug("message: " + message.toString());

		/**
		 * flex 客户端安全策略请求，需要返回特定报文
		 */
		if (CIMConstant.FLEX_POLICY_REQUEST.equals(message)) {
			ios.write(CIMConstant.FLEX_POLICY_RESPONSE);
			return;
		}

		CIMSession cimSession = new CIMSession(ios);
		ReplyBody reply = new ReplyBody();
		SentBody body = (SentBody) message;
		String key = body.getKey();

		CIMRequestHandler handler = handlers.get(key);
		if (handler == null) {
			reply.setCode(CIMConstant.ReturnCode.CODE_405);
			reply.setCode("KEY [" + key + "] 服务端未定义");
		} else {
			reply = handler.process(cimSession, body);
		}

		if (reply != null) {
			reply.setKey(key);
			cimSession.write(reply);
			logger.debug("-----------------------process done. reply: "
					+ reply.toString());
		}

		// 设置心跳时间
		cimSession.setAttribute(CIMConstant.HEARTBEAT_KEY,
				System.currentTimeMillis());
	}

	/**
	 */
	public void sessionClosed(IoSession ios) throws Exception {

		CIMSession cimSession = new CIMSession(ios);
		try {
			logger.warn("sessionClosed()... from "
					+ cimSession.getRemoteAddress());
			CIMRequestHandler handler = handlers.get("sessionClosedHander");
			if (handler != null
					&& cimSession.containsAttribute(CIMConstant.SESSION_KEY)) {
				handler.process(cimSession, null);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 */
	public void sessionIdle(IoSession session, IdleStatus status)
			throws Exception {
		logger.warn("sessionIdle()... from "
				+ session.getRemoteAddress().toString());
		if (!session.containsAttribute(CIMConstant.SESSION_KEY)) {
			session.close(true);
		} else {
			// 如果5分钟之内客户端没有发送心态，则可能客户端断网，关闭连接
			Object heartbeat = session.getAttribute(CIMConstant.HEARTBEAT_KEY);
			if (heartbeat != null
					&& System.currentTimeMillis()
							- Long.valueOf(heartbeat.toString()) >= 300000) {
				session.close(false);
			}
		}
	}

	/**
	 */
	public void exceptionCaught(IoSession session, Throwable cause)
			throws Exception {
		logger.error("exceptionCaught()... from " + session.getRemoteAddress());
		logger.error(cause);
		cause.printStackTrace();
	}

	/**
	 */
	public void messageSent(IoSession session, Object message) throws Exception {

		
		
		// 设置心跳时间
		session.setAttribute(CIMConstant.HEARTBEAT_KEY,
				System.currentTimeMillis());
	}

	public HashMap<String, CIMRequestHandler> getHandlers() {
		return handlers;
	}

	public void setHandlers(HashMap<String, CIMRequestHandler> handlers) {
		this.handlers = handlers;
	}

}