connector.java

一个java的server模版程序

package com.ec.generalserver;

import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Set;

public class Connector extends Thread
{
	private Acceptor acceptor;
	private Selector selector;
	private AppFactory app_creator;
	private int conn_timeout = -1;		//秒为单位
	
	private HashMap session_map = new HashMap();
	
	public Connector(Acceptor acptor,AppFactory appfactry)
	{
		acceptor = acptor;
		app_creator = appfactry;
	};
	
	public void setConnTimeout(int timeout)
	{
		conn_timeout = timeout;
	}
	
	public void init() throws Exception
	{
		acceptor.init();
	}
	
	public void run()
	{
		try
		{
			selector = Selector.open();
			acceptor.register(selector);
		} catch (Exception e1)
		{
			e1.printStackTrace();
			return;
		}
			
		while(true)
		{
			
			try
			{
				selector.select(1000*30);	// 半分钟check一次
			
				Set readyKeys = selector.selectedKeys();
				Iterator it = readyKeys.iterator();
				while( it.hasNext()  )
				{
					SelectionKey key = (SelectionKey) it.next();
					it.remove();
				
					if ( key.isAcceptable() )
					{
						Connection conn = acceptor.comeConnection();
						if ( conn != null ) 
						{
							AppProcessor processor = app_creator.CreateApp();
							conn.updateAccess(System.currentTimeMillis());
							conn.registor(selector);
							conn.setConnector(this);
							session_map.put(conn,processor);
						}
					}
					if ( key.isReadable() )
					{
						Connection conn = (Connection)key.attachment();
						conn.updateAccess(System.currentTimeMillis());
						AppProcessor processor = (AppProcessor)session_map.get(conn);
						processor.doProcess(conn);
					}
				}
				
				checkConnStatus();
			}
			catch(ConnectionProcessException e)
			{
				e.printStackTrace();
				try
				{
					e.connection.close();
				} catch (Exception e1)
				{
					e1.printStackTrace();
				}
			}
			catch(Exception e)
			{
				e.printStackTrace();
			}
		}	// while
	}
	
	public void checkConnStatus() throws Exception
	{
		if( conn_timeout < 0 )
			return;
		
		Set conn_set = session_map.keySet();
		Iterator it = conn_set.iterator();
		while( it.hasNext()  )
		{
			Connection conn = (Connection)it.next();
			if ( conn.isTimeout(conn_timeout*1000) )
			{
				System.out.println("connection timeout,close");
				conn.close();
			}
		}
	}
	
	public void remove(Connection conn)
	{
		session_map.remove(conn);
	}
}