connectionfactory.java

数据交换

package com.gemt.dataswap.dao;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import oracle.jdbc.OracleDriver;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.gemt.dataswap.config.DataBaseConfig;
import com.gemt.dataswap.resources.ReadConfig;
import com.gemt.dataswap.util.Constants;

public abstract class ConnectionFactory
{
	private static final Log log = LogFactory.getLog(ConnectionFactory.class);
	
	public static Connection connections[];
	public static Map<String, Integer> dbNameMap = new HashMap<String, Integer>();
	public static Map<String, String> dbDialectMap = new HashMap<String, String>();
	
	static {
		getAllConnection();
	}
	
	public static Connection getConnection(String dbName) {
		if(log.isDebugEnabled()) {
			log.debug("Connection getConnection(String dbName) begin");
		}

		if(log.isDebugEnabled()) {
			log.debug("Connection getConnection(String dbName) end");
		}		
		return connections[dbNameMap.get(dbName).intValue()];
		
		//ReadConfig.dataBaseConfig();
	}
	
	public static Connection[] getAllConnection() {
		if(log.isDebugEnabled()) {
			log.debug("Connection[] getAllConnection() begin");
		}
		Map<String, DataBaseConfig> dataBaseConfigs = ReadConfig.dataBaseConfig;
		if(dataBaseConfigs == null || dataBaseConfigs.size() == 0) {
			if(log.isErrorEnabled()) {
				log.error("lost database config, can not build database connection");
			}
		}
		
		if(log.isDebugEnabled()) {
			log.debug("fetch connection step one ");
		}
		connections = new Connection[dataBaseConfigs.size()];
		Set<String> dbNameSet = dataBaseConfigs.keySet();
		if(log.isDebugEnabled()) {
			log.debug("fetch connection step two " + dbNameSet);
		}		
		int i = 0;
		for(String dbName : dbNameSet) {
			dbNameMap.put(dbName, new Integer(i));
			if(log.isDebugEnabled()) {
				log.debug("fetch connection step three ");
			}
			
			DataBaseConfig dataBaseConfig = dataBaseConfigs.get(dbName);
			dbDialectMap.put(dbName, dataBaseConfig.getDbDialect());
			if(Constants.DATABASE_ORACLE.equals(dataBaseConfig.getDbDialect())) {
				try
				{
					DriverManager.registerDriver(new OracleDriver());
					final String url = Constants.ORACLE_DB_URL_PREFIX + dataBaseConfig.getHost() + ":" 
						+ dataBaseConfig.getPort() + ":" + dataBaseConfig.getDbName();
					if(log.isDebugEnabled()) {
						log.debug("oracle :: url = " + url);
					}
					connections[i] = DriverManager.getConnection(url, dataBaseConfig.getUserName(), dataBaseConfig.getPassword());
				}
				catch (SQLException e)
				{
					if(log.isErrorEnabled()) {
						log.error("cant not build connetion for database :: " + dbName + " , Please check config file || " + Constants.DATABASE_CONFIG);
					}
					//e.printStackTrace();
				}
			}
			else if(Constants.DATABASE_SQL_SERVER.equals(dataBaseConfig.getDbDialect())) {
				try
				{
					DriverManager.registerDriver(new net.sourceforge.jtds.jdbc.Driver());
					final String url = Constants.SQL_SERVER_URL_PREFIX + dataBaseConfig.getHost() + ":" 
						+ dataBaseConfig.getPort()+ "/" + dataBaseConfig.getDbName() + ";TDS=7.0";//"jdbc:jtds:sqlserver://server/db;TDS=7.0";
					if(log.isDebugEnabled()) {
						log.debug("SQL Server :: url = " + url);
					}
					connections[i] = DriverManager.getConnection(url, dataBaseConfig.getUserName(), dataBaseConfig.getPassword());
					
					
				}
				catch (SQLException e)
				{
					if(log.isErrorEnabled()) {
						log.error("cant not build connetion for database :: " + dbName + " , Please check config file || " + Constants.DATABASE_CONFIG);
					}
					//e.printStackTrace();
				}
			}
			else {
				if(log.isWarnEnabled()) {
					log.warn("please config database dialect properly for database :: " + dbName + " , Please check config file || " + Constants.DATABASE_CONFIG);
				}
			}
			//jdbc:oracle:thin:@192.168.0.16:1521:szwrb
			i++;
		}
		
		if(log.isDebugEnabled()) {
			log.debug("Connection[] getAllConnection() end");
		}		
		return connections;
	}
	
	public static void closeAllConnection() {
		if(log.isDebugEnabled())
			log.debug("closeAllConnection() begin..");
		if(connections != null && connections.length > 0) {
			for(int ci = 0; ci < connections.length; ci++) {
				
				try {
					if(connections[ci] != null && !connections[ci].isClosed()) {
						connections[ci].close();
					}
				}
				catch (SQLException e) {
                    if(log.isErrorEnabled()) {
                    	log.error("throw exception when close database connection");
                    }
				}
			}
		}
		
		if(log.isDebugEnabled())
			log.debug("closeAllConnection() end..");		
	}
	
	
	/**
	 * @param args
	 */
	public static void main(String[] args)
	{
		//getAllConnection();
		
		Connection con = getConnection("db3");
		try {
			PreparedStatement pstmt = con.prepareStatement("select * from FTD0311_SNPH_R ");
			ResultSet rs = pstmt.executeQuery();
			while(rs.next()) {
				for(int i = 1; i <= rs.getMetaData().getColumnCount(); i++) {
					log.debug(rs.getMetaData().getColumnName(i) + " " + rs.getObject(i));
				}
			}
			if(rs != null)
				rs.close();
			if(pstmt != null)
				pstmt.close();
		}
		catch (SQLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		finally {
			
		}
		closeAllConnection();
		
		// TODO Auto-generated method stub
	}
}