1. 程式人生 > >實現數據庫連接池

實現數據庫連接池

tinc port 一次 通過 所有 name 不能 註冊 失敗

  文件ConnectionParam.java

package exam.Util.DButil;

public class ConnectionParam
{
    private String driver; // 數據庫連接驅動
    private String url; // 數據庫連接URL
    private String user; // 數據庫連接user
    private String password; // 數據庫連接password
    private int minConnection; // 數據庫連接池最小連接數
    private int maxConnection; // 數據庫連接池最大連接數
    private long timeoutValue; // 連接的最大空閑時間
    private long waitTime; // 取得連接的最大等待時間
    private int incrementalConnections=5; //連接池自動增加連接的數量

    public String getDriver()
    {
        return driver;
    }

    public void setDriver(String driver)
    {
        this.driver = driver;
    }

    public String getUrl()
    {
        return url;
    }

    public void setUrl(String url)
    {
        this.url = url;
    }

    public String getUser()
    {
        return user;
    }

    public void setUser(String user)
    {
        this.user = user;
    }

    public String getPassword()
    {
        return password;
    }

    public void setPassword(String password)
    {
        this.password = password;
    }

    public int getMinConnection()
    {
        return minConnection;
    }

    public void setMinConnection(int minConnection)
    {
        this.minConnection = minConnection;
    }

    public int getMaxConnection()
    {
        return maxConnection;
    }

    public void setMaxConnection(int maxConnection)
    {
        this.maxConnection = maxConnection;
    }

    public long getTimeoutValue()
    {
        return timeoutValue;
    }

    public void setTimeoutValue(long timeoutValue)
    {
        this.timeoutValue = timeoutValue;
    }

    public long getWaitTime()
    {
        return waitTime;
    }

    public void setWaitTime(long waitTime)
    {
        this.waitTime = waitTime;
    }

    public void setIncrementalConnections(int incrementalConnections)
    {
        this.incrementalConnections = incrementalConnections;
    }

    public int getIncrementalConnections()
    {
        return incrementalConnections;
    }

}

  文件ConnectionPool.java

package exam.Util.DButil;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Enumeration;
import java.util.Vector;

public class ConnectionPool
{

    private ConnectionParam param;
    private String testTable = ""; // 測試連接是否可用的測試表名,默認沒有測試表
    private Vector connections = null; // 存放連接池中數據庫連接的向量 , 初始時為
                                       // null,它中存放的對象為PooledConnection 型

    public void setParam(ConnectionParam param)
    {
        this.param = param;
    }

    public ConnectionParam getParam()
    {
        return param;
    }

    /**
     * 構造函數
     *
     * @param param
     */
    public ConnectionPool(ConnectionParam param)
    {
        this.param = param;
    }

    /**
     *
     * 獲取測試數據庫表的名字
     *
     * @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.param.getDriver())
                .newInstance());
        DriverManager.registerDriver(driver); // 註冊 JDBC 驅動程序
        // 創建保存連接的向量 , 初始時有 0 個元素
        connections = new Vector();

        // 根據 initialConnections 中設置的值,創建連接。
        createConnections(this.param.getMinConnection());
        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.param.getMaxConnection() > 0
                    && this.connections.size() >= this.param.getMaxConnection())
            {
                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(this.param.getUrl(),
                this.param.getUser(), this.param.getPassword());

        // 如果這是第一次創建數據庫連接,即檢查數據庫,獲得此數據庫允許支持的
        // 最大客戶連接數目
        // connections.size()==0 表示目前沒有連接己被創建

        if (connections.size() == 0)
        {

            DatabaseMetaData metaData = conn.getMetaData();
            int driverMaxConnections = metaData.getMaxConnections();

            // 數據庫返回的 driverMaxConnections 若為 0 ,表示此數據庫沒有最大
            // 連接限制,或數據庫的最大連接限制不知道
            // driverMaxConnections 為返回的一個整數,表示此數據庫允許客戶連接的數 目
            // 如果連接池中設置的最大連接數量大於數據庫允許的連接數目 , 則置連接池 的最大
            // 連接數目為數據庫允許的最大數目

            if (driverMaxConnections > 0
                    && this.param.getMaxConnection() > driverMaxConnections)
            {
                this.param.setMaxConnection(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(this.param.getIncrementalConnections());
            // 重新從池中查找是否有可用連接
            conn = findFreeConnection();
            if (conn == null)
            {
                // 如果創建連接後仍獲得不到可用的連接,則返回 null
                return null;
            }
        }
        return conn;

    }

    /**
     *
     * 查找連接池中所有的連接,查找一個可用的數據庫連接,
     *
     * 如果沒有可用的連接,返回 null
     *
     * @return 返回一個可用的數據庫連接
     */

    private Connection findFreeConnection() throws SQLException
    {

        Connection conn = null;
        PooledConnection pConn = null;
        // 獲得連接池向量中所有的對象
        Enumeration enumerate = connections.elements();
        // 遍歷所有的對象,看是否有可用的連接
        while (enumerate.hasMoreElements())
        {
            pConn = (PooledConnection) enumerate.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 enumerate = connections.elements();
        // 遍歷連接池中的所有連接,找到這個要返回的連接對象
        while (enumerate.hasMoreElements())
        {
            pConn = (PooledConnection) enumerate.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 enumerate = connections.elements();
        while (enumerate.hasMoreElements())
        {
            // 獲得一個連接對象
            pConn = (PooledConnection) enumerate.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 enumerate = connections.elements();
        while (enumerate.hasMoreElements())
        {
            pConn = (PooledConnection) enumerate.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)
        {
            e.printStackTrace();
        }
    }

    /**
     *
     * 內部使用的用於保存連接池中連接對象的類 此類中有兩個成員,
     * 一個是數據庫的連接,另一個是指示此連接是否 正在使用的標誌。
     */

    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;
        }

    }

}

  

實現數據庫連接池