分析Atomikos数据连接池源码，弄清testQuery

桃扇骨 2023-02-23 08:15 27阅读 0赞

### [分析Atomikos数据连接池源码，弄清testQuery][Atomikos_testQuery] ###

分类： [java][]2010-03-18 17:19 1163人阅读 [评论][Link 1](0) 收藏 [举报][Link 2]

作者:fbysss  
[msn:jameslastchina@hotmail.com][msn_jameslastchina_hotmail.com]   
blog:blog.csdn.net/fbysss  
声明：本文由fbysss原创，转载请注明出处  
关键字：Atomikos数据连接池

**前言**

Atomikos数据连接池，国内有一些应用，但testQuery这个属性，在网上均是简单配置，并没有做特殊说明。通过对Atomikos源码的分析，发现这里很有学问。

**分析**

我们使用的数据源是AtomikosDataSourceBean，在其doInit方法中， 会调用AtomikosXAConnectionFactory的createPooledConnection方法，该方法会返回一个AtomikosXAPooledConnection连接。

而AtomikosXAPooledConnection是AbstractXPooledConnection的一个子类，AbstractXPooledConnection中，在调用createConnectionProxy时，会调用testUnderlyingConnection方法，用于进行测试。

在AtomikosDataSourceBean的getConnection时，调用connectionPool的borrowConnection方法**，**从而调用AbstractXPooledConnection中的createConnectionProxy，从而调用testUnderlyingConnection方法。

可以看testUnderlyingConnection中的关键代码，一旦设置了testQuery，每次getConnection的时候，就会连接查询一次（通过jprofiler也可以检测出来）：

**\[java\]** [view plain][][copy][view plain]

1.  String testQuery = getTestQuery();  
2.      **if** (testQuery != **null**) \{  
3.          Configuration.logDebug ( **this** + ": testing connection with query \[" + testQuery + "\]" );  
4.          Statement stmt = **null**;  
5.          **try** \{  
6.              stmt = connection.createStatement();  
7.              ResultSet rs = stmt.executeQuery(testQuery);  
8.              rs.close();  
9.              stmt.close();  
10.         \} **catch** ( SQLException e) \{  
11.             **throw** **new** CreateConnectionException ( "Error executing testQuery" ,  e );  
12.         \}  
13.         Configuration.logDebug ( **this** + ": connection tested OK" );  
14.     \}  
15.     **else** \{  
16.         Configuration.logDebug ( **this** + ": no test query, skipping test" );  
17.     \}

如果失败，抛出CreateConnectionException异常。

**注意**ConnectionPool的borrowConnection方法，其中有一段：

**\[java\]** [view plain][][copy][view plain]

1.  Iterator it = connections.iterator();  
2.    
3.  **while** ( it.hasNext() && ret == **null** ) \{  
4.    
5.     xpc = (XPooledConnection) it.next();  
6.    
7.    
8.    
9.     **if** (xpc.isAvailable()) \{  
10.   
11.        **try** \{  
12.   
13.           ret = xpc.createConnectionProxy ( hmsg );  
14.   
15.           Configuration.logDebug( **this** + ": got connection from pool, new size: " + availableSize() + "/" + totalSize());  
16.   
17.        \} **catch** ( CreateConnectionException ex ) \{  
18.   
19.           String msg = **this** +  ": error creating proxy of connection " + xpc;  
20.   
21.           Configuration.logWarning( msg , ex);  
22.   
23.           it.remove();  
24.   
25.           xpc.destroy();  
26.   
27.        \}  
28.   
29.    \}  
30.   
31. \}

可以看到，其做的事情，就是遍历连接池中的连接，一个一个的测试。注意ret = xpc.**createConnectionProxy** ( hmsg )，一旦该方法抛出CreateConnectionException，就执行it.remove();即将该连接从连接池中删除。

如果设置了testQuery属性，每次获取连接时testQuery，能够保证应用服务器启动之后，与数据库连接暂时中断之后，能够在下一次请求时，自动重新建立连接。

连接池是如何自动建立连接的呢？其实就是简单的把无效的连接一个一个删掉，直到全部删光了，池里面没有有效（poolAvailableSize==0,是根据连接的isTerminated状态来判断的，而不是是否被重置过）的连接了，这样根据连接池的机制，就会调用growPool方法去请求新的连接。

但是这样有性能消耗，而且还不小。对于网站展示部分，不需要实时去检测，可以考虑采用**定时检测**的方法：

n  首先，保证testQuery为空，不配置。

n  创建一个DbPoolMonitorService类，实现ApplicationContextAware接口，这样应用启动时，会自动注入ApplicationContext对象。这样，可以在service中，调用getBean方法，获取AtomikosDataSourceBean的实例，

AtomikosDataSourceBean ads = (AtomikosDataSourceBean)ctx.getBean(“jtaDataSource”);

然后，ads中能够获得minPoolSize/maxPoolSize/availablePoolSize/totalPoolSize等属性，可以做一个界面来监测数据连接池的使用和配置情况，甚至可以动态修改这些属性。

n  做一个定时器，调用一个DbPoolMonitorService定期去手工test。使用SELECT 1 语句即可。

n  注意：try 部分ads.setTestQuery (“SELECT 1”);然后调用BaseDao的实例去执行一句话，依然可以使”SELECT 1” 。finally部分ads.setTestQuery(null)，以保证不对其他部分造成后续影响。

n  可根据用户访问量，来决定定时周期，一般10分钟左右即可。这样，避免了每次获取连接去做一次检测操作，又能够将故障限制在一定时间范围内。是一个较好的折衷做法。

**补充：**我们再看 ret = xpc.**createConnectionProxy** ( hmsg )这句话，如果这时候DB和WebAppServer的网络链路已经正常，DB正常运行，返回将不是空，会退出循环。这样，如果连接池中的连接是多个，则只会生成**1个**新的连接，如果要保证连接池机制的效果，需要在写监控程序的时候，去取得所有有效连接，循环test。

[Atomikos_testQuery]: http://blog.csdn.net/fbysss/article/details/5393626
[java]: http://blog.csdn.net/fbysss/article/category/38099
[Link 1]: http://blog.csdn.net/fbysss/article/details/5393626#comments
[Link 2]: http://blog.csdn.net/fbysss/article/details/5393626#report
[msn_jameslastchina_hotmail.com]: mailto:msn:jameslastchina@hotmail.com
[view plain]: http://blog.csdn.net/fbysss/article/details/5393626#