一種完成數據庫連接池的方法(3)
發表時間:2023-08-18 來源:明輝站整理相關軟件相關文章人氣:2
[摘要]DataSourceImpl是一個實現了接口javax.sql.DataSource的類,該類維護著一個連接池的對象。由于該類是一個受保護的類,因此它暴露給使用者的方法只有接口DataSource中...
DataSourceImpl是一個實現了接口javax.sql.DataSource的類,該類維護著一個連接池的對象。由于該類是一個受保護的類,因此它暴露給使用者的方法只有接口DataSource中定義的方法,其他的所有方法對使用者來說都是不可視的。我們先來關心用戶可訪問的一個方法getConnection
/**
* @see javax.sql.DataSource#getConnection(String,String)
*/
public Connection getConnection(String user, String password) throws SQLException
{
//首先從連接池中找出空閑的對象
Connection conn = getFreeConnection(0);
if(conn == null){
//判斷是否超過最大連接數,如果超過最大連接數
//則等待一定時間查看是否有空閑連接,否則拋出異常告訴用戶無可用連接
if(getConnectionCount() >= connParam.getMaxConnection())
conn = getFreeConnection(connParam.getWaitTime());
else{//沒有超過連接數,重新獲取一個數據庫的連接
connParam.setUser(user);
connParam.setPassword(password);
Connection conn2 = DriverManager.getConnection(connParam.getUrl(),
user, password);
//代理將要返回的連接對象
_Connection _conn = new _Connection(conn2,true);
synchronized(conns){
conns.add(_conn);
}
conn = _conn.getConnection();
}
}
return conn;
}
/**
* 從連接池中取一個空閑的連接
* @param nTimeout如果該參數值為0則沒有連接時只是返回一個null
* 否則的話等待nTimeout毫秒看是否還有空閑連接,如果沒有拋出異常
* @return Connection
* @throws SQLException
*/
protected synchronized Connection getFreeConnection(long nTimeout)
throws SQLException
{
Connection conn = null;
Iterator iter = conns.iterator();
while(iter.hasNext()){
_Connection _conn = (_Connection)iter.next();
if(!_conn.isInUse()){
conn = _conn.getConnection();
_conn.setInUse(true);
break;
}
}
if(conn == null && nTimeout > 0){
//等待nTimeout毫秒以便看是否有空閑連接
try{
Thread.sleep(nTimeout);
}catch(Exception e){}
conn = getFreeConnection(0);
if(conn == null)
throw new SQLException("沒有可用的數據庫連接");
}
return conn;
}
DataSourceImpl類中實現getConnection方法的跟正常的數據庫連接池的邏輯是一致的,首先判斷是否有空閑的連接,如果沒有的話判斷連接數是否已經超過最大連接數等等的一些邏輯。但是有一點不同的是通過DriverManager得到的數據庫連接并不是及時返回的,而是通過一個叫_Connection的類中介一下,然后調用_Connection.getConnection返回的。如果我們沒有通過一個中介也就是JAVA中的Proxy來接管要返回的接口對象,那么我們就沒有辦法截住Connection.close方法。
終于到了核心所在,我們先來看看_Connection是如何實現的,然后再介紹是客戶端調用Connection.close方法時走的是怎樣一個流程,為什么并沒有真正的關閉連接。
/**
* 數據連接的自封裝,屏蔽了close方法
* @author Liudong
*/
class _Connection implements InvocationHandler
{
private final static String CLOSE_METHOD_NAME = "close";
private Connection conn = null;
//數據庫的忙狀態
private boolean inUse = false;
//用戶最后一次訪問該連接方法的時間
private long lastAccessTime = System.currentTimeMillis();
_Connection(Connection conn, boolean inUse){
this.conn = conn;
this.inUse = inUse;
}
/**
* Returns the conn.
* @return Connection
*/
public Connection getConnection() {
//返回數據庫連接conn的接管類,以便截住close方法
Connection conn2 = (Connection)Proxy.newProxyInstance(
conn.getClass().getClassLoader(),
conn.getClass().getInterfaces(),this);
return conn2;
}
/**
* 該方法真正的關閉了數據庫的連接
* @throws SQLException
*/
void close() throws SQLException{
//由于類屬性conn是沒有被接管的連接,因此一旦調用close方法后就直接關閉連接
conn.close();
}
/**
* Returns the inUse.
* @return boolean
*/
public boolean isInUse() {
return inUse;
}
/**
* @see java.lang.reflect.InvocationHandler#invoke(java.lang.Object, java.lang.reflect.Method, java.lang.Object)
*/
public Object invoke(Object proxy, Method m, Object[] args)
throws Throwable
{
Object obj = null;
//判斷是否調用了close的方法,如果調用close方法則把連接置為無用狀態
if(CLOSE_METHOD_NAME.equals(m.getName()))
setInUse(false);
else
obj = m.invoke(conn, args);
//設置最后一次訪問時間,以便及時清除超時的連接
lastAccessTime = System.currentTimeMillis();
return obj;
}
/**
* Returns the lastAccessTime.
* @return long
*/
public long getLastAccessTime() {
return lastAccessTime;
}
/**
* Sets the inUse.
* @param inUse The inUse to set
*/
public void setInUse(boolean inUse) {
this.inUse = inUse;
}
}
一旦使用者調用所得到連接的close方法,由于用戶的連接對象是經過接管后的對象,因此JAVA虛擬機會首先調用_Connection.invoke方法,在該方法中首先判斷是否為close方法,如果不是則將代碼轉給真正的沒有被接管的連接對象conn。否則的話只是簡單的將該連接的狀態設置為可用。到此您可能就明白了整個接管的過程,但是同時也有一個疑問:這樣的話是不是這些已建立的連接就始終沒有辦法真正關閉?答案是可以的。我們來看看ConnectionFactory.unbind方法,該方法首先找到名字對應的連接池對象,然后關閉該連接池中的所有連接并刪除掉連接池。在DataSourceImpl類中定義了一個close方法用來關閉所有的連接,詳細代碼如下:
/**
* 關閉該連接池中的所有數據庫連接
* @return int 返回被關閉連接的個數
* @throws SQLException
*/
public int close() throws SQLException
{
int cc = 0;
SQLException excp = null;
Iterator iter = conns.iterator();
while(iter.hasNext()){
try{
((_Connection)iter.next()).close();
cc ++;
}catch(Exception e){
if(e instanceof SQLException)
excp = (SQLException)e;
}
}
if(excp != null)
throw excp;
return cc;
}
該方法一一調用連接池中每個對象的close方法,這個close方法對應的是_Connection中對close的實現,在_Connection定義中關閉數據庫連接的時候是直接調用沒有經過接管的對象的關閉方法,因此該close方法真正的釋放了數據庫資源。
以上文字只是描述了接口方法的接管,具體一個實用的連接池模塊還需要對空閑連接的監控并及時釋放連接,詳細的代碼請參照附件。