代码说明、参考文章及运行环境请参考系列文章(五)
/*
* DBConnectionManager.java
* Created on 2005-1-29
*/
package db;
import java.io.*;
import java.sql.*;
import java.util.*;
import org.apache.log4j.Logger;
//import org.apache.log4j.PropertyConfigurator;
/**
* 管理类DBConnectionManager支持对一个或多个由属性文件定义的数据库连接
* 池的访问.客户程序可以调用getInstance()方法访问本类的唯一实例.
*
* @author Yewberry
*/
public class DBConnectionManager {
private static DBConnectionManager instance; // 唯一实例
private static int clients;
private Vector drivers = new Vector();
//private PrintWriter log;
private Hashtable pools = new Hashtable();
private static Logger log;
static {
try {
log = Logger.getLogger(DBConnectionManager.class.getName());
log.info("Log4j 日志类成功加载.");
// 指定 log4j.properties 文件所在的位置
// PropertyConfigurator.configure("C://test//log4j.properties");
} catch (NoClassDefFoundError ee) {
System.out.println("Log4j 日志类加载失败。");
}
}
/**
* 返回唯一实例.如果是第一次调用此方法,则创建实例
*
* @return DBConnectionManager 唯一实例
*/
static synchronized public DBConnectionManager getInstance() {
if (instance == null) {
instance = new DBConnectionManager();
}
clients++;
return instance;
}
/**
* 建构函数私有以防止其它对象创建本类实例
*/
private DBConnectionManager() {
init();
}
/**
* 将连接对象返回给由名字指定的连接池
*
* @param name 在属性文件中定义的连接池名字
* @param con 连接对象
*/
public void freeConnection(String name, Connection con) {
DBConnectionPool pool = (DBConnectionPool) pools.get(name);
if (pool != null) {
pool.freeConnection(con);
}
}
/**
* 获得一个可用的(空闲的)连接.如果没有可用连接,且已有连接数小于最大连接数
* 限制,则创建并返回新连接
*
* @param name 在属性文件中定义的连接池名字
* @return Connection 可用连接或null
*/
public Connection getConnection(String name) {
DBConnectionPool pool = (DBConnectionPool) pools.get(name);
if (pool != null) {
return pool.getConnection();
}
return null;
}
/**
* 获得一个可用连接.若没有可用连接,且已有连接数小于最大连接数限制,
* 则创建并返回新连接.否则,在指定的时间内等待其它线程释放连接.
*
* @param name 连接池名字
* @param time 以毫秒计的等待时间
* @return Connection 可用连接或null
*/
public Connection getConnection(String name, long time) {
DBConnectionPool pool = (DBConnectionPool) pools.get(name);
if (pool != null) {
return pool.getConnection(time);
}
return null;
}
/**
* 关闭所有连接,撤销驱动程序的注册
*/
public synchronized void release() {
// 等待直到最后一个客户程序调用
if (--clients != 0) {
return;
}
Enumeration allPools = pools.elements();
while (allPools.hasMoreElements()) {
DBConnectionPool pool = (DBConnectionPool) allPools.nextElement();
pool.release();
}
Enumeration allDrivers = drivers.elements();
while (allDrivers.hasMoreElements()) {
Driver driver = (Driver) allDrivers.nextElement();
try {
DriverManager.deregisterDriver(driver);
if (log.isInfoEnabled()) {
log.info("撤销JDBC驱动程序 " +
driver.getClass().getName()+ " 的注册。");
}
//log("撤销JDBC驱动程序 " + driver.getClass().getName()+"的注册///");
} catch (SQLException e) {
if (log.isInfoEnabled()) {
log.info("无法撤销下列JDBC驱动程序的注册: " +
driver.getClass().getName(), e);
}
//log(e, "无法撤销下列JDBC驱动程序的注册: " + driver.getClass().getName());
}
}
}
/**
* 读取属性完成初始化
*/
private void init() {
InputStream is = getClass().getResourceAsStream("/db.properties");
Properties dbProps = new Properties();
try {
dbProps.load(is);
// 将 dbProps 中的元素全部列出
// dbProps.list(System.out);
} catch (Exception e) {
System.err.println("不能读取属性文件。" +
"请确保db.properties在CLASSPATH指定的路径中。");
return;
}
/*
String logFile = dbProps.getProperty("logfile", "DBConnectionManager.log");
try {
//log = new PrintWriter(new FileWriter(logFile, true), true);
} catch (IOException e) {
System.err.println("无法打开日志文件: " + logFile);
//log = new PrintWriter(System.err);
}
*/
loadDrivers(dbProps);
createPools(dbProps);
}
/**
* 装载和注册所有JDBC驱动程序
*
* @param props 属性
*/
private void loadDrivers(Properties props) {
String driverClasses = props.getProperty("drivers");
StringTokenizer st = new StringTokenizer(driverClasses);
while (st.hasMoreElements()) {
String driverClassName = st.nextToken().trim();
try {
Driver driver = (Driver)
Class.forName(driverClassName).newInstance();
DriverManager.registerDriver(driver);
drivers.addElement(driver);
//System.out.println("成功注册JDBC驱动程序: " + driverClassName);
if (log.isInfoEnabled()) {
log.info("成功注册JDBC驱动程序: " + driverClassName);
}
//log("成功注册JDBC驱动程序///" + driverClassName);
} catch (Exception e) {
//System.out.println("无法注册JDBC驱动程序:" + driverClassName);
if (log.isInfoEnabled()) {
log.info("无法注册JDBC驱动程序: " +
driverClassName, e);
}
//log("无法注册JDBC驱动程序: " +
// driverClassName + ", 错误: " + e);
}
}
}
/**
* 根据指定属性创建连接池实例.
*
* @param props 连接池属性
*/
private void createPools(Properties props) {
Enumeration propNames = props.propertyNames();
while (propNames.hasMoreElements()) {
String name = (String) propNames.nextElement();
if (name.endsWith(".url")) {
String poolName = name.substring(0, name.lastIndexOf("."));
String url = props.getProperty(poolName + ".url");
if (url == null) {
//System.out.println("没有为连接池" + poolName + "指定URL");
if (log.isInfoEnabled()) {
log.info("没有为连接池 " + poolName + " 指定URL");
}
//log("没有为连接池" + poolName + "指定URL");
continue;
}
String user = props.getProperty(poolName + ".user");
String password = props.getProperty(poolName + ".password");
String maxconn = props.getProperty(poolName + ".maxconn", "0");
int max;
try {
max = Integer.valueOf(maxconn).intValue();
} catch (NumberFormatException e) {
//System.out.println("错误的最大连接数限制: " +
// maxconn + " .连接池: " + poolName);
if (log.isInfoEnabled()) {
log.info("错误的最大连接数限制: " + maxconn +
" .连接池: " + poolName);
}
//log("错误的最大连接数限制: " + maxconn + " .连接池: " + poolName);
max = 0;
}
DBConnectionPool pool =
new DBConnectionPool(poolName, url, user, password, max);
pools.put(poolName, pool);
//System.out.println("成功创建连接池 " + poolName);
//System.out.println(pools.toString());
if (log.isInfoEnabled()) {
log.info("成功创建连接池 " + poolName);
}
//log("成功创建连接池" + poolName);
}
}
}
}