www.pudn.com > dbPool.rar > ConnectionPool.java
package dev.trade.common.db; import java.sql.*; import java.util.*; import org.apache.log4j.Logger; /** *Title: 福建客服支撑系统
* *Description:
* ConnectionPool 类创建了一个对特定数据库指定大小的连接池。连接池对象 * 允许客户端指定 JDBC 驱动程序，数据库，使用数据库的用户名和密码。而且， * 客户端能指定连接池的在初始创建是产生数据库连接的数量，和指定当连接 * 不够时每次自动增加连接的数量及连接池最多的数据库连接的数量。 * --- 主要方法: --- * createPool: 创建连接池 , 线程己同步 * getConnection: 从连接池中获得一个数据库连接 * returnConnection: 返回一个连接到连接池中 * refreshConnections: 刷新连接池 * closeConnectionPool: 关闭连接池 * *Copyright: Copyright (c) 2006
* *Company: Newland
* * @author ZhengYanNan * @version 1.0 */ public class ConnectionPool{ private static Logger log = Logger.getLogger(ConnectionPool.class); // 数据库驱动 private String jdbcDriver = ""; // 数据 URL private String dbUrl = ""; // 数据库用户名 private String dbUsername = ""; // 数据库用户密码 private String dbPassword = ""; // 数据库登录超时（秒） private int loginTimeout = 120; // 测试连接是否可用的测试表名，默认没有测试表 private String testTable = ""; // 连接池的初始大小 private int initialConnections = 10; // 连接池自动增加的大小 private int incrementalConnections = 5; // 连接池最大的大小 private int maxConnections = 50; // 存放连接池中数据库连接的向量 , 初始时为 null, 它中存放的对象为 PooledConnection 型 private Vector connections = null; /** * 构造函数 * @param jdbcDriver String JDBC 驱动类串 * @param dbUrl String 数据库 URL * @param dbUsername String 连接数据库用户名 * @param dbPassword String 连接数据库用户的密码 * @param timeout int 数据库登录超时(秒) */ public ConnectionPool(String jdbcDriver, String dbUrl, String dbUsername, String dbPassword, int loginTimeout){ this.jdbcDriver = jdbcDriver; this.dbUrl = dbUrl; this.dbUsername = dbUsername; this.dbPassword = dbPassword; this.loginTimeout = loginTimeout; } /** * 创建一个数据库连接池，连接池中的可用连接的数量采用类成员 * initialConnections 中设置的值 */ public synchronized void createPool() throws Exception{ if(connections != null){ return; } // 实例化 JDBC Driver 中指定的驱动类实例 Driver driver = (Driver)(Class.forName(this.jdbcDriver).newInstance()); DriverManager.registerDriver(driver); // 注册 JDBC 驱动程序 // 创建保存连接的向量 , 初始时有 0 个元素 connections = new Vector(); // 根据 initialConnections 中设置的值，创建连接。 createConnections(this.initialConnections); log.info("数据库连接池创建成功!"); } /** * 关闭连接池中所有的连接，并清空连接池。 */ public synchronized void closeConnectionPool() throws SQLException{ if(connections == null){ log.error("连接池不存在,无法关闭!"); return; } PooledConnection pConn = null; Enumeration em = connections.elements(); while(em.hasMoreElements()){ pConn = (PooledConnection)em.nextElement(); // 如果忙，等5秒,5 秒后直接关闭它 if(pConn.isBusy()){ wait(5000); } closeConnection(pConn.getConnection()); // 从连接池向量中删除它 connections.removeElement(pConn); } connections = null; log.info("所有数据库连接已断开,连接池成功关闭!"); } /** * 通过调用 getFreeConnection() 函数返回一个可用的数据库连接 , * 如果当前没有可用的数据库连接，并且更多的数据库连接不能创 * 建（如连接池大小的限制），此函数等待一会再尝试获取。 * @return 返回一个可用的数据库连接对象 */ public synchronized Connection getConnection() throws Exception{ if(connections == null){ throw new NullPointerException("连接池还没创建，无法获取连接"); } Connection conn = getFreeConnection(); // 获得一个可用的数据库连接 // 如果目前没有可以使用的连接，即所有的连接都在使用中 while(conn == null){ wait(250); conn = getFreeConnection(); // 重新再试，直到获得可用的连接，否则为忙,等待 } return conn; } /** * 此函数返回一个数据库连接到连接池中，并把此连接置为空闲。 * 所有使用连接池获得的数据库连接均应在不使用此连接时返回它。 * @param 需返回到连接池中的连接对象 */ public boolean returnConnection(Connection conn){ if(connections == null){ return true; } PooledConnection pConn = null; Enumeration em = connections.elements(); while(em.hasMoreElements()){ pConn = (PooledConnection)em.nextElement(); if(conn == pConn.getConnection()){ pConn.setBusy(false); return true; } } return false; } /** * 刷新连接池中所有的连接对象 */ public synchronized void refreshConnections() throws SQLException{ if(connections == null){ log.error("连接池不存在,无法刷新!"); return; } PooledConnection pConn = null; Enumeration em = connections.elements(); while(em.hasMoreElements()){ pConn = (PooledConnection)em.nextElement(); // 如果对象忙则等 5 秒 ,5 秒后直接刷新 if(pConn.isBusy()){ wait(5000); } // 关闭此连接，用一个新的连接代替它。 closeConnection(pConn.getConnection()); pConn.setConnection(newConnection()); pConn.setBusy(false); } } //=============================== /** * 返回连接池的初始大小 * @return 初始连接池中可获得的连接数量 */ public int getInitialConnections(){ return this.initialConnections; } /** * 设置连接池的初始大小 * @param 用于设置初始连接池中连接的数量 */ public void setInitialConnections(int initialConnections){ this.initialConnections = initialConnections; } /** * 返回连接池自动增加的大小 * @return 连接池自动增加的大小 */ public int getIncrementalConnections(){ return this.incrementalConnections; } /** * 设置连接池自动增加的大小 * @param 连接池自动增加的大小 */ public void setIncrementalConnections(int incrementalConnections){ this.incrementalConnections = incrementalConnections; } /** * 返回连接池中最大的可用连接数量 * @return 连接池中最大的可用连接数量 */ public int getMaxConnections(){ return this.maxConnections; } /** * 设置连接池中最大可用的连接数量 * @param 设置连接池中最大可用的连接数量值 */ public void setMaxConnections(int maxConnections){ this.maxConnections = maxConnections; } /** * 获取测试数据库表的名字 * @return 测试数据库表的名字 */ public String getTestTable(){ return this.testTable; } /** * 设置测试表的名字 * @param testTable String 测试表的名字 */ public void setTestTable(String testTable){ this.testTable = testTable; } //=================================== /** * 创建由 numConnections 指定数目的数据库连接 , 并把这些连接放入 connections 向量中 * @param numConnections 要创建的数据库连接的数目 */ private void createConnections(int numConnections) throws SQLException{ for(int x = 0; x < numConnections; x++){ // 如果 maxConnections 为 0 或负数，表示连接数量没有限制,如果连接数己经达到最大，即退出。 if(this.maxConnections > 0 && this.connections.size() >= this.maxConnections){ break; } // 增加一个连接到连接池中（向量 connections 中） try{ connections.addElement(new PooledConnection(newConnection())); } catch(SQLException e){ log.error("创建数据库连接失败! ",e); throw new SQLException(); } log.info("数据库连接己创建..."); } } /** * 创建一个新的数据库连接并返回它 * @return 返回一个新创建的数据库连接 */ private Connection newConnection() throws SQLException{ DriverManager.setLoginTimeout(loginTimeout); Connection conn = DriverManager.getConnection(dbUrl, dbUsername, dbPassword); conn.setAutoCommit(true); // 目前没有连接己被创建 if(connections.size() == 0){ DatabaseMetaData metaData = conn.getMetaData(); int driverMaxConnections = metaData.getMaxConnections(); // 数据库返回的 driverMaxConnections 若为 0，表示此数据库没有最大连接限制， // 或数据库的最大连接限制不知道 // driverMaxConnections 为返回的一个整数，表示此数据库允许客户连接的数目 // 如果连接池中设置的最大连接数量大于数据库允许的连接数目 , 则置连接池的最大 // 连接数目为数据库允许的最大数目 if(driverMaxConnections > 0 && this.maxConnections > driverMaxConnections){ this.maxConnections = driverMaxConnections; } } return conn; } /** * 本函数从连接池向量 connections 中返回一个可用的的数据库连接，如果 * 当前没有可用的数据库连接，本函数则根据 incrementalConnections 设置 * 的值创建几个数据库连接，并放入连接池中。 * 如果创建后，所有的连接仍都在使用中，则返回 null * @return 返回一个可用的数据库连接 */ private Connection getFreeConnection() throws SQLException{ // 从连接池中获得一个可用的数据库连接 Connection conn = findFreeConnection(); if(conn == null){ createConnections(incrementalConnections); // 重新从池中查找是否有可用连接 conn = findFreeConnection(); if(conn == null){ // 如果创建连接后仍获得不到可用的连接，则返回 null return null; } } return conn; } /** * 查找连接池中所有的连接，查找一个可用的数据库连接， * 如果没有可用的连接，返回 null * @return 返回一个可用的数据库连接 */ private Connection findFreeConnection() throws SQLException{ Connection conn = null; PooledConnection pConn = null; Enumeration em = connections.elements(); // 遍历所有的对象，看是否有可用的连接 while(em.hasMoreElements()){ pConn = (PooledConnection)em.nextElement(); if(!pConn.isBusy()){ conn = pConn.getConnection(); pConn.setBusy(true); // 测试此连接是否可用, 此功能用于恢复网络原因断开的连接 if(!testConnection(conn)){ // 如果此连接不可再用了，则创建一个新的连接, 并替换此不可用的连接对象，如果创建失败，返回 null try{ conn = newConnection(); } catch(SQLException e){ log.error("创建数据库连接失败!",e); return null; } pConn.setConnection(conn); } break; // 己经找到一个可用的连接，退出 } } return conn; // 返回找到到的可用连接 } /** * 测试一个连接是否可用，如果不可用，关掉它并返回 false,否则可用返回 true * @param conn 需要测试的数据库连接 * @return 返回 true 表示此连接可用， false 表示不可用 */ private boolean testConnection(Connection conn){ try{ // 判断测试表是否存在 if(testTable.equals("")){ // 如果测试表为空，试着使用此连接的 setAutoCommit() 方法 // 来判断连接否可用（此方法只在部分数据库可用，如果不可用 , // 抛出异常）。注意：使用测试表的方法更可靠 conn.setAutoCommit(true); } else{ // 有测试表的时候使用测试表测试 Statement stmt = conn.createStatement(); stmt.execute("select count(*) from " + testTable); stmt.close(); } } catch(SQLException e){ // 上面抛出异常，此连接己不可用，关闭它，并返回 false; closeConnection(conn); return false; } return true; } /** * 关闭一个数据库连接 * @param 需要关闭的数据库连接 */ private void closeConnection(Connection conn){ try{ conn.close(); } catch(SQLException e){ log.error("关闭数据库连接出错!",e); } } /** * 使程序等待给定的毫秒数 * @param 给定的毫秒数 */ private void wait(int mSeconds){ try{ Thread.sleep(mSeconds); } catch(InterruptedException e){ } } ////////////////////////////////////////////////////////////////////// /** * 内部使用的用于保存连接池中连接对象的类 * 此类中有两个成员，一个是数据库的连接，另一个是指示此连接是否 * 正在使用的标志。 */ class PooledConnection{ // 数据库连接 Connection connection = null; // 此连接是否正在使用的标志，默认没有正在使用 boolean busy = false; public PooledConnection(Connection connection){ this.connection = connection; } public Connection getConnection(){ return connection; } public void setConnection(Connection connection){ this.connection = connection; } public boolean isBusy(){ return busy; } public void setBusy(boolean busy){ this.busy = busy; } } }