org.bei.dema.tcp.TcpServer Maven / Gradle / Ivy

Go to download
package org.bei.dema.tcp;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.TimerTask;
import java.util.Vector;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;


/**
 * author：zhaochengbei
 * date：2017/5/31
*/
public class TcpServer {

	/**
	 * 
	 */
	private ServerSocket accepter;

	/**
	 * 
	 */
	public int maxConnectionCount = 0;
	/**
	 * 
	 */
	private Runnable acceptSocketLogic = new Runnable() {
		
		public void run() {
			while(true){
				try {
					Socket socket = accepter.accept();
					
					if(socket != null){
						TcpConnection connection = new TcpConnection(socket);
						//if exceed maxConnectionCount 
						if(maxConnectionCount!=0 && getConnections().size()>= maxConnectionCount){
							connection.close(TcpConnectionCloseReason.ExceedMaxConnectionCount);
							continue;
						}
						connection.inReading = false;
						connection.lastReadTime = System.currentTimeMillis();
						connectionManager.add(connection);
						
					}
				}catch (Exception e) {
					//open too many file or accepter be close
					e.printStackTrace();
					break;
				}
			}
			
		}
	};

	/**
	 * 
	 */
	private Thread acceptSocketThread = new Thread(acceptSocketLogic,"AcceptSocket_0");
	/**
	 * 
	 */
	public int readIdleTimeoutSeconds = 0;
	/**
	 * 
	 */
	public int readIdleCheckGapSeconds = 1;
	/**
	 * 
	 */
	private Runnable checkReadIdleTimeOutLogic = new Runnable() {
		
		public void run() {
			while(true){
				try {
					long time = System.currentTimeMillis();
					for (int i = 0; i < getConnections().size(); i++) {
						TcpConnection connection = getConnections().get(i);
						if(readIdleTimeoutSeconds !=0 &&connection.isClose() == false&&time - connection.lastReadTime> readIdleTimeoutSeconds*1000){
							//use part will receive a close event;
							connection.close(TcpConnectionCloseReason.ReadIdleTimeOut);
						}
					}
				}catch (ArrayIndexOutOfBoundsException e) {
					//thread confict,need do nothing
				}
				
				try {
					Thread.sleep(readIdleCheckGapSeconds);
				} catch (InterruptedException e) {
					break;
//					// TODO Auto-generated catch block
//					e.printStackTrace();
				}
			}
			
			
			
		}
	};

	/**
	 * 
	 */
	private ScheduledExecutorService checkReadIdleTimeOutThreads = Executors.newScheduledThreadPool(1, new ThreadFactory() {
		
		public Thread newThread(Runnable r) {
			return new Thread(r,"CheckReadIdleTimeOut_0");
		}
	});

	/**
	 * 
	 */
	/**
	 * 
	 */
	private TcpConnectionManager connectionManager = new TcpConnectionManager();

	/**
	 * 
	 * @param maxConnectionCount
	 * @param readIdleTimeoutSeconds ,timeunit is millsecond
	 */
	public void config(int maxConnectionCount,int readIdleTimeoutSeconds){
		this.maxConnectionCount = maxConnectionCount;
		this.readIdleTimeoutSeconds = readIdleTimeoutSeconds;
	}
	/**
	 * 
	 */
	public void configThread(int exeIoTaskThreadCount){
		connectionManager.exeIoTaskThreadCount = exeIoTaskThreadCount;
	}
	/**
	 * 
	 */
	public void configCheckGap(int checkStatusGapMillSeconds,int readIdleCheckGapSeconds){
		connectionManager.checkStatusGapMillSeconds = checkStatusGapMillSeconds;
		this.readIdleCheckGapSeconds = readIdleCheckGapSeconds;
	}
	/**
	 * 
	 */
	public void start(int port,IoHandler ioHandler) throws IOException{
		connectionManager.start(ioHandler);
		accepter = new ServerSocket(port);
		acceptSocketThread.start();
		checkReadIdleTimeOutThreads.execute(checkReadIdleTimeOutLogic);
	}
	/**
	 * 
	 */
	public Vector getConnections(){
		return connectionManager.connections;
	}
	
	
	/**
	 * 
	 */
	public void shutdown() throws IOException{
		accepter.close();
		checkReadIdleTimeOutThreads.shutdown();
		connectionManager.shutdown();

	}
}