一個經試用效果非常不錯的資料庫連接池--JAVA

標籤：

前言：

    雖說現在許多企業級的應用伺服器均自己帶有資料庫連接池功能，就連 Tomcat 也支援了這種功能。然而在許多時候，我們還是要使用資料庫連接池，如：訪問資料庫的 Java 傳統型應用程式等。這個資料庫連接池是我根據《 Inside Servlets 》一書中的樣本改寫而成，經過試用，效果非常不錯。特發布共用。

 

　原始碼

 

//ConnectionPool.java package com.abner.dbconnector; import java.sql.*; import java.util.*; /** * ConnectionPool 類建立了一個對特定資料庫指定大小的串連池。串連池對象 * 允許用戶端指定 JDBC 驅動程式，資料庫，使用資料庫的使用者名稱和密碼。而且， * 用戶端能指定串連池的在初始建立是產生資料庫連接的數量，和指定當串連 * 不夠時每次自動增加串連的數量及串連池最多的資料庫連接的數量。 * * 對外提供的方法有： ConnectionPool ：建構函式 * getInitialConnections: 返回串連池初始化大小 * setInitialConnections: 設定串連池初始化大小 * getIncrementalConnections: 返回串連池自動增加的增量 * setIncrementalConnections: 設定串連池自動增加的大小 * getMaxConnections ：獲得串連池的最大可允許的串連數 * setMaxConnections ：設定串連池的最大可允許的串連數 * getTestTable ：獲得測試表的名字 * setTestTable ：設定測試表的名字 * createPool: 建立串連池 , 線程己同步 * getConnection: 從串連池中獲得一個資料庫連接 * returnConnection: 返回一個串連到串連池中 * refreshConnections: 重新整理串連池 * closeConnectionPool: 關閉串連池 * * * @author abnerchai Email: [email protected] * @version 1.0.0 * */ public class ConnectionPool {    private String jdbcDriver = ""; // 資料庫驅動    private String dbUrl = ""; // 資料 URL    private String dbUsername = ""; // 資料庫使用者名稱    private String dbPassword = ""; // 資料庫使用者密碼    private String testTable = ""; // 測試連接是否可用的測試表名，預設沒有測試表    private int initialConnections = 10; // 串連池的初始大小    private int incrementalConnections = 5;// 串連池自動增加的大小    private int maxConnections = 50; // 串連池最大的大小    private Vector connections = null; // 存放串連池中資料庫連接的向量 , 初始時為 null // 它中存放的對象為 PooledConnection 型 /** * 建構函式 * * @param jdbcDriver String JDBC 驅動類串 * @param dbUrl String 資料庫 URL * @param dbUsername String 串連資料庫使用者名稱 * @param dbPassword String 串連資料庫使用者的密碼 * */ public ConnectionPool(String jdbcDriver,String dbUrl,String dbUsername,String dbPassword) {    this.jdbcDriver = jdbcDriver;    this.dbUrl = dbUrl;    this.dbUsername = dbUsername;    this.dbPassword = dbPassword; } /** * 返回串連池的初始大小 * * @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; } /** * * 建立一個資料庫連接池，串連池中的可用串連的數量採用類成員 * initialConnections 中設定的值 */ public synchronized void createPool() throws Exception { // 確保串連池沒有建立 // 如果串連池己經建立了，儲存串連的向量 connections 不會為空白 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);     System.out.println(" 資料庫連接池建立成功！ "); } /** * 建立由 numConnections 指定數目的資料庫連接 , 並把這些串連 * 放入 connections 向量中 * * @param numConnections 要建立的資料庫連接的數目 */ private void createConnections(int numConnections) throws SQLException { // 迴圈建立指定數目的資料庫連接 for (int x = 0; x < numConnections; x++) { // 是否串連池中的資料庫連接的數量己經達到最大？最大值由類成員 maxConnections // 指出，如果 maxConnections 為 0 或負數，表示串連數量沒有限制。 // 如果串連數己經達到最大，即退出。    if (this.maxConnections > 0 && this.connections.size() >= this.maxConnections) {       break;    } //add a new PooledConnection object to connections vector // 增加一個串連到串連池中（向量 connections 中） try{     connections.addElement(new PooledConnection(newConnection())); }catch(SQLException e){     System.out.println(" 建立資料庫連接失敗！ "+e.getMessage());     throw new SQLException(); }     System.out.println(" 資料庫連接己建立 ......"); } } /** * 建立一個新的資料庫連接並返回它 * * @return 返回一個新建立的資料庫連接 */ private Connection newConnection() throws SQLException { // 建立一個資料庫連接 Connection conn = DriverManager.getConnection(dbUrl, dbUsername, dbPassword); // 如果這是第一次建立資料庫連接，即檢查資料庫，獲得此資料庫允許支援的 // 最大客戶串連數目 //connections.size()==0 表示目前沒有串連己被建立 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; // 返回建立的新的資料庫連接 } /** * 通過調用 getFreeConnection() 函數返回一個可用的資料庫連接 , * 如果當前沒有可用的資料庫連接，並且更多的資料庫連接不能創 * 建（如串連池大小的限制），此函數等待一會再嘗試擷取。 * * @return 返回一個可用的資料庫連接對象 */ public synchronized Connection getConnection() throws SQLException { // 確保串連池己被建立 if (connections == null) {     return null; // 串連池還沒建立，則返回 null } Connection conn = getFreeConnection(); // 獲得一個可用的資料庫連接 // 如果目前沒有可以使用的串連，即所有的串連都在使用中 while (conn == null){ // 等一會再試     wait(250);     conn = getFreeConnection(); // 重新再試，直到獲得可用的串連，如果 //getFreeConnection() 返回的為 null // 則表明建立一批串連後也不可獲得可用串連 } 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 en = connections.elements(); // 遍曆所有的對象，看是否有可用的串連 while (en.hasMoreElements()) {    pConn = (PooledConnection) en.nextElement();    if (!pConn.isBusy()) {     // 如果此對象不忙，則獲得它的資料庫連接並把它設為忙     conn = pConn.getConnection();     pConn.setBusy(true);     // 測試此串連是否可用     if (!testConnection(conn)) {      // 如果此串連不可再用了，則建立一個新的串連，      // 並替換此停用連線物件，如果建立失敗，返回 null      try{        conn = newConnection();      }catch(SQLException e){        System.out.println(" 建立資料庫連接失敗！ "+e.getMessage());        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 {// 有測試表的時候使用測試表測試      //check if this connection is valid      Statement stmt = conn.createStatement();      stmt.execute("select count(*) from " + testTable); } } catch (SQLException e) { // 上面拋出異常，此串連己不可用，關閉它，並返回 false; closeConnection(conn); return false; } // 串連可用，返回 true return true; } /** * 此函數返回一個資料庫連接到串連池中，並把此串連置為空白閑。 * 所有使用串連池獲得的資料庫連接均應在不使用此串連時返回它。 * * @param 需返回到串連池中的連線物件 */ public void returnConnection(Connection conn) { // 確保串連池存在，如果串連沒有建立（不存在），直接返回 if (connections == null) { System.out.println(" 串連池不存在，無法返回此串連到串連池中 !"); return; } PooledConnection pConn = null; Enumeration en = connections.elements(); // 遍曆串連池中的所有串連，找到這個要返回的連線物件 while (en.hasMoreElements()) { pConn = (PooledConnection) en.nextElement(); // 先找到串連池中的要返回的連線物件 if (conn == pConn.getConnection()) { // 找到了 , 設定此串連為空白閑狀態 pConn.setBusy(false); break; } } } /** * 重新整理串連池中所有的連線物件 * */ public synchronized void refreshConnections() throws SQLException { // 確保串連池己創新存在 if (connections == null) { System.out.println(" 串連池不存在，無法重新整理 !"); return; } PooledConnection pConn = null; Enumeration en = connections.elements(); while (en.hasMoreElements()) { // 獲得一個連線物件 pConn = (PooledConnection) en.nextElement(); // 如果對象忙則等 5 秒 ,5 秒後直接重新整理 if (pConn.isBusy()) { wait(5000); // 等 5 秒 } // 關閉此串連，用一個新的串連代替它。 closeConnection(pConn.getConnection()); pConn.setConnection(newConnection()); pConn.setBusy(false); } } /** * 關閉串連池中所有的串連，並清空串連池。 */ public synchronized void closeConnectionPool() throws SQLException { // 確保串連池存在，如果不存在，返回 if (connections == null) {    System.out.println(" 串連池不存在，無法關閉 !");    return; } PooledConnection pConn = null; Enumeration en = connections.elements(); while (en.hasMoreElements()) {    pConn = (PooledConnection) en.nextElement(); // 如果忙，等 5 秒    if (pConn.isBusy()) {       wait(5000); // 等 5 秒     } //5 秒後直接關閉它 closeConnection(pConn.getConnection()); // 從串連池向量中刪除它 connections.removeElement(pConn); } // 置串連池為空白 connections = null; } /** * 關閉一個資料庫連接 * * @param 需要關閉的資料庫連接 */ private void closeConnection(Connection conn) { try { conn.close(); }catch (SQLException e) { System.out.println(" 關閉資料庫連接出錯： "+e.getMessage()); } } /** * 使程式等待給定的毫秒數 * * @param 給定的毫秒數 */ private void wait(int mSeconds) { try { Thread.sleep(mSeconds); } catch (InterruptedException e) { } } /** * * 內部使用的用於儲存串連池中連線物件的類 * 此類中有兩個成員，一個是資料庫的串連，另一個是指示此串連是否 * 正在使用的標誌。 */ class PooledConnection {    Connection connection = null;// 資料庫連接    boolean busy = false; // 此串連是否正在使用的標誌，預設沒有正在使用    // 建構函式，根據一個 Connection 構告一個 PooledConnection 對象    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; } } } 這個程式所有的解釋我都詳細地寫在了來源程式中，我想就不必多解說了吧。

如果你有改進或修改，請保留原作者的聲明

一個經試用效果非常不錯的資料庫連接池--JAVA