import com.mchange.v2.c3p0.ComboPooledDataSource;
import java.beans.PropertyVetoException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/**
* c3p0實作的一個資料源工廠類
*/
public class DataSourceFactory {
// 儲存所有建立的DataSource,程序停止時統一銷毀
private static final Map<String, ComboPooledDataSource> dsPool = new ConcurrentHashMap<>();
static synchronized ComboPooledDataSource getDataSource(String driver, String url, String user, String pass) {
String dataSourceKey = driver + "_" + url + "_" + user;
ComboPooledDataSource dataSource = dsPool.get(dataSourceKey);
if (dataSource == null) {
dataSource = makeDataSource(driver, url, user, pass);
dsPool.put(dataSourceKey, dataSource);
}
return dataSource;
}
private static ComboPooledDataSource makeDataSource(String driver, String url, String user, String pass) {
ComboPooledDataSource dataSource = new ComboPooledDataSource();
try {
dataSource.setDriverClass(driver);
System.out.println("JDBC Driver Class load completion:" + driver);
} catch (PropertyVetoException e) {
System.out.println("JDBC Driver Class ["+driver+"] is not Found!");
throw new RuntimeException("Please check your application.conf!", e);
}
dataSource.setJdbcUrl(url);
dataSource.setUser(user);
dataSource.setPassword(pass);
// 連接配接池中最小連接配接數
dataSource.setMinPoolSize(1);
// 連接配接池中最大連接配接數
dataSource.setMaxPoolSize(20);
// 檢查連接配接池空閑連接配接的時間間隔,機關:秒
dataSource.setIdleConnectionTestPeriod(60);
// 最大空閑時間,指定時間中連接配接未被使用則會被丢棄,機關:秒
dataSource.setMaxIdleTime(3600);
return dataSource;
}
/**
* 關閉所有資料源并清空緩存
* 如果正在建立DataSource,closeAll将等待建立完成後執行
*/
public static void closeAll() {
rwLock.writeLock().lock();
try {
for (ComboPooledDataSource dataSource : dsPool.values()) {
dataSource.close();
}
dsPool.clear();
} finally {
rwLock.writeLock().unlock();
}
}
}