com.centit.support.database.DbcpConnectPools Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of centit-database Show documentation
数据库操作通用方法和函数，从以前的util包中分离出来，并且整合了部分sys-module中的函数
There is a newer version: 5.3.2302
package com.centit.support.database;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.dbcp2.BasicDataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DbcpConnectPools {
	
	protected static final Logger logger = LoggerFactory.getLogger(DbcpConnectPools.class);
	private static final
		Map dbcpDataSourcePools
		 = new HashMap();
	
	private static synchronized BasicDataSource addDataSource(DataSourceDescription dsDesc){
		
		//BasicDataSourceFactory.createDataSource(properties)
		BasicDataSource ds = new BasicDataSource();  
		ds.setDriverClassName(dsDesc.getDriver());  
		ds.setUsername(dsDesc.getUsername());  
		ds.setPassword(dsDesc.getPassword());  
		ds.setUrl(dsDesc.getConnUrl());  
		ds.setInitialSize(dsDesc.getInitialSize()); // 初始的连接数；  
		ds.setMaxTotal(dsDesc.getMaxTotal());  
		ds.setMaxIdle(dsDesc.getMaxIdle());  
		ds.setMaxWaitMillis(dsDesc.getMaxWaitMillis());  
		ds.setMinIdle(dsDesc.getMinIdle());  
		//ds.setTestOnReturn(false);
		dbcpDataSourcePools.put(dsDesc, ds);
		return ds;
	}
	
	public static synchronized BasicDataSource getDataSource(DataSourceDescription dsDesc){
		BasicDataSource ds = dbcpDataSourcePools.get(dsDesc);
		if(ds==null)
			ds = addDataSource(dsDesc);
		return ds;
	}
	
	public static synchronized DbcpConnect getDbcpConnect(DataSourceDescription dsDesc) throws SQLException{
		BasicDataSource ds = dbcpDataSourcePools.get(dsDesc);
		if(ds==null)
			ds = addDataSource(dsDesc);
		Connection conn = null;
		conn = ds.getConnection();  
        conn.setAutoCommit(false);  
        ///*dsDesc.getUsername(),dsDesc.getDbType(),*/
        return new DbcpConnect(
        		dsDesc.getDatabaseCode(),conn);
	}
	
	/* 获得数据源连接状态 */
    public static Map getDataSourceStats(DataSourceDescription dsDesc){  
        BasicDataSource bds = dbcpDataSourcePools.get(dsDesc);
        if(bds==null)
        	return null;
        Map map = new HashMap(2);  
        map.put("active_number", bds.getNumActive());  
        map.put("idle_number", bds.getNumIdle());  
        return map;  
    }  
  
    /** 关闭数据源 */  
    public static synchronized void shutdownDataSource(){  
    	for(Map.Entry dbs : dbcpDataSourcePools.entrySet()){
    		try {
				dbs.getValue().close();
			} catch (SQLException e) {
				logger.error(e.getMessage(),e);//e.printStackTrace();
			}
    	}
    }  
    
    public static synchronized boolean testDataSource(DataSourceDescription dsDesc){
		BasicDataSource ds = new BasicDataSource();  
		ds.setDriverClassName(dsDesc.getDriver());  
		ds.setUsername(dsDesc.getUsername());  
		ds.setPassword(dsDesc.getPassword());  
		ds.setUrl(dsDesc.getConnUrl());  
		ds.setInitialSize(dsDesc.getInitialSize()); // 初始的连接数；  
		ds.setMaxTotal(dsDesc.getMaxTotal());  
		ds.setMaxIdle(dsDesc.getMaxIdle());  
		ds.setMaxWaitMillis(dsDesc.getMaxWaitMillis());  
		ds.setMinIdle(dsDesc.getMinIdle());
		boolean connOk = false;
		try {
			//Class.forName(dsDesc.getDriver());
			//Connection conn = DriverManager.getConnection(dsDesc.getConnUrl(),
					//dsDesc.getUsername(),dsDesc.getPassword());
			Connection conn = ds.getConnection();
			if(conn != null){
				connOk = true;
				conn.close();
			}
			ds.close();
		} catch (SQLException e) {
			try {
				ds.close();
			} catch (SQLException e1) {
				//e1.printStackTrace();
			}
			logger.error(e.getMessage(),e);//e.printStackTrace();
		}
		return connOk;
	}
    
    public static void closeConnect(Connection conn){
    	if(conn!=null){			
			try {
				conn.close();				
			} catch (SQLException e) {
				logger.error(e.getMessage(),e);//e.printStackTrace();
			}
		}
    }
}