package openminer;

import java.util.*;
import java.sql.*;
import java.net.*;
import java.io.*;

import org.jdom.*;
import openminer.util.*;
import openminer.util.db.*;

public class OpenMiner {	
	private Hashtable m_DBServerAccounts = null;
	private Hashtable m_UserAccountsTable = null;
	private Hashtable m_MinerThreadPoolsTable = null;
	private Listener m_Listener = null;
	private int m_ListenPort;
	private static String m_ModelTableFullName;
	public  final static boolean TRACE_ENABLE = true;
	
	public static void main(String[] args) {
		OpenMiner server = new OpenMiner();
		try{
			// 开启服务
			server.startService();
			// 用户命令输入
			server.commands();
		}catch(Exception e){
			e.printStackTrace();
		}
	}
	
	/**
	 * 开启openminer服务
	 *
	 */
	public void startService() throws Exception {
		System.out.println("OpenMiner Service Start...");
		loadDBServerXmlFile("dbserver.xml");
		loadPropertyXmlFile("property.xml");
		createThreadPools();
		startListener();
		System.out.println("Service start sucessfully! ");
	}
	
	/**
	 * 关闭openminer服务
	 * @throws Exception
	 */
	public void exitService() throws Exception {
		System.out.println("OpenMiner Service exit...");
		m_Listener.stopListener();
		releaseThreadPools();
		System.out.println("Service exit sucessfully!");
		System.exit(0);
	}
	
	/**
	 * 服务端用户命令输入控制
	 * @throws Exception
	 */
	private void commands() throws Exception {
		InputStreamReader isr = new InputStreamReader(System.in);
		BufferedReader br = new BufferedReader(isr);
		while(true) {
			System.out.print(":>");
			String cmd = br.readLine();
			if(cmd.equals("exit")) {
				br.close();
				isr.close();
				exitService();
				break;
			}
			else if(cmd.equals("startup")) {
				startService();
			}
			else if(cmd.equals("restart")) {
				exitService();
				startService();
			}
			else {
				System.out.println("Unknown command: "+cmd);
			}
		}
	}
	
	
	/**
	 * 装载数据库的配置文件
	 * @param fileName 数据库配置文件名
	 * @throws Exception
	 */
	private void loadDBServerXmlFile(String fileName) throws Exception {
		m_DBServerAccounts = new Hashtable();
		Element dbserverElement = XMLFileLoader.loadXMLDOM(fileName);
		// 获取数据库名字
		String dbName = dbserverElement.getAttributeValue("database");
		// 获取模型Model保存的数据表
		m_ModelTableFullName = dbserverElement.getAttributeValue("modeltable");
		// 获取数据库驱动名字
		String driverName = dbserverElement.getAttributeValue("driver");
		// 装载数据库和驱动
		loadDatabaseNameAndDrivers(dbName,driverName);
		// 装数据库用户帐号信息
		List userList = dbserverElement.getChildren();
		for(int i=0; i<userList.size(); i++) {
			Element userElement = (Element)userList.get(i);
			DBUserAccount account = new DBUserAccount();
			account.m_UserName = userElement.getChildText("username");
			account.m_Password = userElement.getChildText("password");
			account.m_URL = userElement.getChildText("url");
			m_DBServerAccounts.put(account.m_UserName,account);
			// 打印用户的信息
			println("Add DB Account user : "+account.m_UserName);
		}
	}
	
	/**
	 * 建立数据库名字和驱动
	 * @param dbName 数据库名字
	 * @param driverName 数据库驱动名字
	 * @throws Exception
	 */
	private void loadDatabaseNameAndDrivers(String dbName,
			String driverName) throws Exception {
		// 设置数据库
		DatabaseProperty.setDatabaseName(dbName);
		println("Database setting OK!");
		// 注册数据库驱动
		Class.forName(driverName).newInstance();
		println("Database driver register OK!");
	}
	
	/**
	 * 装载openminer的配置文件
	 * @param fileName 配置信息文件名
	 * @throws Exception
	 */
	private void loadPropertyXmlFile(String fileName) throws Exception {
		m_UserAccountsTable = new Hashtable();
		Element propertyElement = XMLFileLoader.loadXMLDOM(fileName);
		// 读取TCP端口号
		m_ListenPort = Integer.parseInt(propertyElement.getAttributeValue("port"));
		// 读取用户帐号信息
		List userList = propertyElement.getChildren();
		for(int i=0; i<userList.size(); i++) {
			Element userElement = (Element)userList.get(i);
			UserAccount account = new UserAccount();
			account.m_MaxConnection = 
				Integer.parseInt(userElement.getAttributeValue("maxconnection"));
			account.m_UserName = userElement.getChildText("username");
			account.m_Password = userElement.getChildText("password");
			account.m_DBUserName = userElement.getChildText("dbusername");
			DBUserAccount dbAccount = 
				(DBUserAccount)m_DBServerAccounts.get(account.m_DBUserName);
			if(dbAccount == null) 
				throw new Exception("Cannot find the database user : "+account.m_DBUserName);
			account.m_DBPassword = dbAccount.m_Password;
			account.m_DBURL = dbAccount.m_URL;
			// 增加帐号到表内
			m_UserAccountsTable.put(account.m_UserName, account);
			// 打印o帐户信息
			println("Add OpenMiner Account user : "+account.m_UserName);
		}
	}
	
	/**
	 * 创建数据库连接池
	 * @throws Exception
	 */
	private void createThreadPools() throws Exception {
		m_MinerThreadPoolsTable = new Hashtable();
		Enumeration userEnum =  m_UserAccountsTable.elements();
		UserAccount account;
		Connection conn;
		MinerThreadPool pool;
		while(userEnum.hasMoreElements()) {
			account = (UserAccount)userEnum.nextElement();
			pool = new MinerThreadPool(account.m_MaxConnection);
			for(int i=0;i<account.m_MaxConnection; i++) {
				// 创建数据库连接对象
				conn = DriverManager.getConnection(account.m_DBURL, 
						account.m_DBUserName,
						account.m_DBPassword);
				// 给线程对象设置数据库连接
				pool.connect(i, conn);
				//	打印数据连接成功
				println("Database Connection :"+account.m_DBURL+" "+i+" OK!");		
			}
			// 增加线程池到表内
			m_MinerThreadPoolsTable.put(account.m_UserName, pool);
		}
	}
	
	private void releaseThreadPools() throws Exception {
		Enumeration poolEnum = m_MinerThreadPoolsTable.elements();
		while(poolEnum.hasMoreElements()) {
			MinerThreadPool pool = (MinerThreadPool)poolEnum.nextElement();
			pool.destroyAll();
		}
	}
		
	/**
	 * 开始监听客户端socket连接
	 *
	 */
	private void startListener() throws Exception  {
		m_Listener = new Listener(m_ListenPort,m_UserAccountsTable,m_MinerThreadPoolsTable);
		m_Listener.startListener();
	}
	
	/**
	 * 打印调试信息
	 * @param s
	 */
	public static void println(String s) {
		if(TRACE_ENABLE) {
			System.out.println(s);
		}
	}
	
	/**
	 * 打印不换行的调试信息
	 * @param s
	 */
	public void print(String s) {
		if(TRACE_ENABLE) {
			System.out.print(s);
		}
	}
	
	/**
	 * 获取Model存储表的全名（包括了schema)
	 * @return
	 */
	public synchronized static String getModelTableFullName() {
		return m_ModelTableFullName;
	}
}

class DBUserAccount {
	public String m_UserName;
	public String m_Password;
	public String m_URL;

	public DBUserAccount() {
		
	}
}