连接池是非常好的想法,应用很普遍。自己写一个数据库连接池,并不像想象中那样困难。一般系统对连接池的功能不会有太多要求,使用自己的连接池未必是个坏主意。下面以Oracle为例,但是对Teradata和Greenplum也是可行的。另外我还实现了连接有效性检查(checkConn)和恢复连接(resetConn)的方法。本例编程采用的是JRE1.4.2环境(别忘了准备访问数据库的jar包)。有任何问题请随时留言,欢迎探讨。
在Oracle内创建测试数据:
- drop table my_table;
- create table my_table(
- field_id varchar2(3),
- field_content varchar2(60),
- record_create_date date default sysdate
- );
-
- insert into my_table(field_id,field_content) values('001','this is first record');
- insert into my_table(field_id,field_content) values('002','this is second record');
- insert into my_table(field_id,field_content) values('003','this is third record');
- commit;
DBPool.java:
- package dataWebService;
-
- import java.sql.DriverManager;
- import java.util.Date;
- import java.sql.Connection;
- import java.sql.SQLException;
- import java.sql.Statement;
-
- public class DBPool{
- private String cls;
- private String url;
- private String usr;
- private String pss;
- private int connCount = 3;
- private Connection[] connections;
- private String[] connStatus;
- private Date[] lastQueryTime;
-
- public DBPool(DBPoolConfiguration poolConfiguration){
- this.connCount=poolConfiguration.getConnCount();
- this.cls=poolConfiguration.getCls();
- this.url=poolConfiguration.getUrl();
- this.usr=poolConfiguration.getUsr();
- this.pss=poolConfiguration.getPss();
- this.connections=new Connection[this.connCount];
- this.connStatus=new String[this.connCount];
- for(int i=0;i<this.connCount;i++){
- this.connStatus[i]="X";
- }
- this.lastQueryTime = new Date[this.connCount];
- }
-
- public DBPool(String cls,String url,String usr,String pss){
- this.cls=cls;
- this.url=url;
- this.usr=usr;
- this.pss=pss;
- this.connections=new Connection[this.connCount];
- this.connStatus=new String[this.connCount];
- for(int i=0;i<this.connCount;i++){
- this.connStatus[i]="X";
- }
- this.lastQueryTime = new Date[this.connCount];
- }
-
- public void initPool(){
- if(connCount<1){
- System.out.println("请正确设置连接池窗口个数");
- }else{
- try{
- Class.forName(this.cls);
- }catch(ClassNotFoundException e){
- System.out.println(e.getMessage());
- }catch(Exception e){
- System.out.println(e.getMessage());
- }
-
- for(int i=0;i<this.connCount;i++){
- try{
- this.connections[i]=DriverManager.getConnection(this.url, this.usr, this.pss);
- this.connStatus[i]="Y";
- }catch(SQLException e){
- System.out.println(e.getMessage());
- }catch(Exception e){
- System.out.println(e.getMessage());
- }
- }
- System.out.println("initPool is ready...");
- }
- }
-
- public void freePool(){
- for(int i=0;i<this.connCount;i++){
- try{
- this.connections[i].commit();
- this.connections[i].close();
- this.connStatus[i]="X";
- this.lastQueryTime[i]=null;
- }catch(Exception e){
- try{
- this.connections[i].close();
- this.connStatus[i]="X";
- this.lastQueryTime[i]=null;
- }catch(Exception e1){
- System.out.println(e1.getMessage());
- }
- }
- }
- System.out.println("freePool is over ...");
- }
-
- public DBPoolConnection getPoolConn() throws DBPoolIsFullException{
- DBPoolConnection poolConnection = new DBPoolConnection();
- poolConnection.connNbr=getConnNbr();
- if(poolConnection.connNbr==-1){
- throw new DBPoolIsFullException("连接池已满");
- }else{
- poolConnection.conn=getConn(poolConnection.connNbr);
- }
- return poolConnection;
- }
-
- public void freePoolConn(DBPoolConnection poolConnection){
- if(poolConnection==null){
- System.out.println("poolConnection==null,不需要释放");
- }else{
- freeConn(poolConnection.connNbr);
- }
- }
-
- public void printPoolStatus(){
- for(int i=0;i<this.connStatus.length;i++){
- System.out.println("");
- System.out.print(this.connStatus[i].toString());
- if(this.lastQueryTime[i]==null){
- System.out.print("-[null] ");
- }else{
- System.out.print("-["+this.lastQueryTime[i].toString()+"] ");
- }
- }
- System.out.println("");
- }
-
- public String getCls(){
- return this.cls;
- }
-
- public String getUrl(){
- return this.url;
- }
-
- public String getUsr(){
- return this.usr;
- }
-
- int getConnNbr(){
- int iConn=-1;
- for(int i=0;i<this.connCount;i++){
- if(this.connStatus[i].equals("Y")){
- this.lastQueryTime[i]=new Date();
- this.connStatus[i]="N";
- iConn=i;
- break;
- }
- }
- return iConn;
- }
-
- Connection getConn(int i){
- return this.connections[i];
- }
-
- void closeConnForTest(DBPoolConnection poolConnection){
- try{
- this.connections[poolConnection.connNbr].close();
- }catch(SQLException e){
- System.out.println(e.getMessage());
- }
- }
-
- boolean checkConn(DBPoolConnection poolConnection){
- Statement stmt=null;
- String checkMessage="";
- boolean checkResult=true;
-
-
- try{
- String sql = "select * from dual";
- stmt = this.connections[poolConnection.connNbr].createStatement();
- stmt.executeQuery(sql);
- stmt.close();
- checkMessage = "checkConn:checkMessage:execute sql success";
- System.out.println(checkMessage);
- }catch(Exception e){
- checkMessage = e.getMessage();
- System.out.println(e.getMessage());
- if(checkMessage==null){
- checkMessage="e.getMessage() is null";
- System.out.println(checkMessage);
- }
-
- if (checkMessage.indexOf("ORA-00942")>=0){
- checkResult=true;
- }else if(checkMessage.indexOf("does not exist")>=0){
- checkResult=true;
- }else if(checkMessage.indexOf("Syntax error")>=0){
- checkResult=true;
- }else{
- checkResult=false;
- }
- }
- return checkResult;
- }
-
- boolean resetConn(DBPoolConnection poolConnection){
- boolean result=false;
-
- if(poolConnection==null){
- System.out.println("poolConnection==null,不知道您想重设哪个连接");
- }else if(poolConnection.connNbr==-1){
- System.out.println("poolConnection.connNbr==-1,不知道您想重设哪个连接");
- }else{
- if(checkConn(poolConnection)==true){
- System.out.println("连接有效,不需要重设");
- }else{
-
- try{
- Class.forName(this.cls);
- }catch(ClassNotFoundException e){
- System.out.println(e.getMessage());
- }catch(Exception e){
- System.out.println(e.getMessage());
- }
- try{
- this.connections[poolConnection.connNbr]=DriverManager.getConnection(this.url, this.usr, this.pss);
- this.connStatus[poolConnection.connNbr]="Y";
- System.out.println(poolConnection.connNbr+"连接已重建");
- result = true;
- }catch(SQLException e){
- System.out.println(e.getMessage());
- }catch(Exception e){
- System.out.println(e.getMessage());
- }
- }
- }
- return result;
- }
-
- void freeConn(int i){
- try{
- if(i==-1){
- System.out.println("i=-1,不需要释放");
- }else{
- this.connections[i].commit();
- }
- }catch(SQLException e){
- System.out.println(e.getMessage());
- }catch(Exception e){
- System.out.println(e.getMessage());
- }
- this.connStatus[i]="Y";
- }
- }
DBPoolConfiguration.java
- package dataWebService;
-
- public class DBPoolConfiguration {
- private String cls;
- private String url;
- private String usr;
- private String pss;
- private int connCount;
-
- public String getCls() {
- return cls;
- }
- public void setCls(String cls) {
- this.cls = cls;
- }
- public String getUrl() {
- return url;
- }
- public void setUrl(String url) {
- this.url = url;
- }
- public String getUsr() {
- return usr;
- }
- public void setUsr(String usr) {
- this.usr = usr;
- }
- public String getPss() {
- return pss;
- }
- public void setPss(String pss) {
- this.pss = pss;
- }
- public int getConnCount() {
- return connCount;
- }
- public void setConnCount(int connCount) {
- this.connCount = connCount;
- }
- }
DBPoolConnection.java:
- package dataWebService;
-
- import java.sql.Connection;
-
- public class DBPoolConnection{
- public int connNbr=-1;
- public Connection conn=null;
- DBPoolConnection(){
- this.connNbr=-1;
- this.conn = null;
- }
- }
DBPoolIsFullException.java
- package dataWebService;
-
- public class DBPoolIsFullException extends Exception{
- static final long serialVersionUID=1L;
- DBPoolIsFullException(String message){
- super(message);
- }
- }
Test.java
- package myAction;
-
- import dataWebService.DBPool;
- import dataWebService.DBPoolConnection;
- import dataWebService.DBPoolConfiguration;
- import java.sql.ResultSet;
- import java.sql.ResultSetMetaData;
- import java.sql.Statement;
-
- public class Test {
- static String rpad(String str,int len){
- String s = str;
- if(s==null){
- s="";
- }
- while(s.getBytes().length<len){
- s += " ";
- }
- return s;
- }
-
- public static void main(String[] args) {
-
- String cls ="",url="",usr="",pss="",sql="";
- Statement stmt=null;
- ResultSet rs=null;
- String gapStr="|";
- int connCount=2;
-
-
- cls = "oracle.jdbc.driver.OracleDriver";
- url = "jdbc:oracle:thin:@localhost:1521:myoradb";
- usr = "abc";
- pss = "123";
-
- sql = "select t.field_id,t.field_content,to_char(t.record_create_date,'YYYYMMDD') day from my_table t";
-
- DBPoolConfiguration poolConfiguration=new DBPoolConfiguration();
- poolConfiguration.setCls(cls);
- poolConfiguration.setUrl(url);
- poolConfiguration.setUsr(usr);
- poolConfiguration.setPss(pss);
- poolConfiguration.setConnCount(connCount);
-
- DBPool myPool = new DBPool(poolConfiguration);
- myPool.initPool();
-
-
- System.out.println("");
- System.out.print("after init Pool");
- myPool.printPoolStatus();
- System.out.println("");
-
- DBPoolConnection c1=null;
-
- try{
- c1 = myPool.getPoolConn();
-
- System.out.println("");
- System.out.print("after getPoolConn");
- myPool.printPoolStatus();
- System.out.println("");
-
- stmt = c1.conn.createStatement();
- rs = stmt.executeQuery(sql);
-
- ResultSetMetaData md = rs.getMetaData();
-
- String recordHead = "";
- int colCount = md.getColumnCount();
- for(int i=1;i<=colCount;i++){
- if(recordHead.equals("")){
- recordHead += rpad(md.getColumnName(i),md.getColumnDisplaySize(i));
- }else{
- recordHead+= gapStr + rpad(md.getColumnName(i),md.getColumnDisplaySize(i));
- }
- }
- System.out.println(recordHead);
-
- while(rs.next()){
- String tmp = "";
- for(int i=1;i<=colCount;i++){
- int colSize = md.getColumnDisplaySize(i)>md.getColumnName(i).length()?md.getColumnDisplaySize(i):md.getColumnName(i).length();
- if(tmp.equals("")){
- tmp += rpad(rs.getString(i),colSize);
- }else{
- tmp += gapStr + rpad(rs.getString(i),colSize);
- }
- }
- System.out.println(tmp);
- }
-
- stmt.close();
- myPool.freePoolConn(c1);
-
- System.out.println("");
- System.out.print("after freePoolConn");
- myPool.printPoolStatus();
- System.out.println("");
- }catch(Exception e){
- System.out.println(e.getMessage());
- }
-
- myPool.freePool();
-
- }
- }
下面是Test.java的执行结果:
- initPool is ready...
-
- after init Pool
- Y-[null]
- Y-[null]
-
-
- after getPoolConn
- N-[Wed Mar 20 14:46:31 GMT 2013]
- Y-[null]
-
- FIELD_ID|FIELD_CONTENT |DAY
- 001 |this is first record |20130320
- 002 |this is second record |20130320
- 003 |this is third record |20130320
-
- after freePoolConn
- Y-[Wed Mar 20 14:46:31 GMT 2013]
- Y-[null]
-
- freePool is over ...
感谢您的耐心读到此处,我相信接下去的文字会更有价值。
保持连接池简单性的几个设计思想(不一定正确):
1)在系统中连接池不应作为一个独立的模块,最好是作为某模块的底层实现。这样可以将超时管理、请求队列、确保资源释放、数据分页(方言不可避免了)等功能剥离出去。
2)固定连接数比动态连接数容易实现,简单的代码更易于维护。
本例有待完善之处(供参考):
1)查询空闲连接要遍历数组,这样当池中的连接数很大时可能会有问题(具体没测过)
2)可以添加连接池阻塞的功能,有时候后台数据库需要重启,或者想重建连接池以提供更多的并发连接数?
本文转自 hexiaini235 51CTO博客,原文链接:http://blog.51cto.com/idata/1159243,如需转载请自行联系原作者