关闭
软件测试网 软件测试培训 软件测试论坛 测试解决方案 文章资料精选 软件测试博客 软件测试招聘

Java：自定义数据库连接池

发表于：2013-3-26 09:36

字体：大 中 小 | 上一篇 | 下一篇 | 我要投稿

 作者：ztfriend    来源：51Testing软件测试网采编

#
软件开发
#
java

　　连接池是非常好的想法，应用很普遍。自己写一个数据库连接池，并不像想象中那样困难。一般系统对连接池的功能不会有太多要求，使用自己的连接池未必是个坏主意。下面以Oracle为例，但是对Teradata和Greenplum也是可行的。另外我还实现了连接有效性检查（checkConn）和恢复连接(resetConn)的方法。本例编程采用的是JRE1.4.2环境（别忘了准备访问数据库的jar包）。有任何问题请随时留言，欢迎探讨。

　　在Oracle内创建测试数据：

  1. drop table my_table;  
  2. create table my_table(  
  3.        field_id varchar2(3),  
  4.        field_content varchar2(60),  
  5.        record_create_date date default sysdate  
  6. );  
  8. insert into my_table(field_id,field_content) values('001','this is first record');  
  9. insert into my_table(field_id,field_content) values('002','this is second record');  
  10. insert into my_table(field_id,field_content) values('003','this is third record');  
  11. commit;

　　DBPool.java：

  1. package dataWebService;  
  3. import java.sql.DriverManager;  
  4. import java.util.Date;  
  5. import java.sql.Connection;  
  6. import java.sql.SQLException;  
  7. import java.sql.Statement;  
  9. public class DBPool{  
  10.     private String cls;  
  11.     private String url;  
  12.     private String usr;  
  13.     private String pss;   
  14.     private int connCount = 3;//连接数 
  15.     private Connection[] connections;//保存数据库连接 
  16.     private String[] connStatus;// 已连可用Y   已连不可用N   未连接X 
  17.     private Date[] lastQueryTime;//时间戳 
  18.       
  19.     public DBPool(DBPoolConfiguration poolConfiguration){  
  20.         this.connCount=poolConfiguration.getConnCount();  
  21.         this.cls=poolConfiguration.getCls();  
  22.         this.url=poolConfiguration.getUrl();  
  23.         this.usr=poolConfiguration.getUsr();  
  24.         this.pss=poolConfiguration.getPss();  
  25.         this.connections=new Connection[this.connCount];  
  26.         this.connStatus=new String[this.connCount];  
  27.         for(int i=0;i<this.connCount;i++){  
  28.             this.connStatus[i]="X";//初始化全部未连接 
  29.         }  
  30.         this.lastQueryTime = new Date[this.connCount];        
  31.     }  
  32.       
  33.     public DBPool(String cls,String url,String usr,String pss){  
  34.         this.cls=cls;  
  35.         this.url=url;  
  36.         this.usr=usr;  
  37.         this.pss=pss;  
  38.         this.connections=new Connection[this.connCount];  
  39.         this.connStatus=new String[this.connCount];  
  40.         for(int i=0;i<this.connCount;i++){  
  41.             this.connStatus[i]="X";//初始化全部未连接 
  42.         }  
  43.         this.lastQueryTime = new Date[this.connCount];  
  44.     }  
  46.     public void initPool(){  
  47.         if(connCount<1){  
  48.             System.out.println("请正确设置连接池窗口个数");  
  49.         }else{  
  50.             try{  
  51.                 Class.forName(this.cls);//register class 
  52.             }catch(ClassNotFoundException e){  
  53.                 System.out.println(e.getMessage());  
  54.             }catch(Exception e){  
  55.                 System.out.println(e.getMessage());//other exceptions 
  56.             }  
  58.             for(int i=0;i<this.connCount;i++){  
  59.                 try{  
  60.                     this.connections[i]=DriverManager.getConnection(this.url, this.usr, this.pss);  
  61.                     this.connStatus[i]="Y";  
  62.                 }catch(SQLException e){  
  63.                     System.out.println(e.getMessage());  
  64.                 }catch(Exception e){  
  65.                     System.out.println(e.getMessage());//other exceptions 
  66.                 }  
  67.             }     
  68.             System.out.println("initPool is ready...");  
  69.         }//end if 
  70.     }  
  72.     public void freePool(){  
  73.         for(int i=0;i<this.connCount;i++){  
  74.             try{  
  75.                 this.connections[i].commit();  
  76.                 this.connections[i].close();  
  77.                 this.connStatus[i]="X";  
  78.                 this.lastQueryTime[i]=null;  
  79.             }catch(Exception e){  
  80.                 try{  
  81.                     this.connections[i].close();  
  82.                     this.connStatus[i]="X";  
  83.                     this.lastQueryTime[i]=null;                   
  84.                 }catch(Exception e1){  
  85.                     System.out.println(e1.getMessage());//just for catch 
  86.                 }  
  87.             }             
  88.         }  
  89.         System.out.println("freePool is over ...");  
  90.     }  
  92.     public DBPoolConnection getPoolConn() throws DBPoolIsFullException{  
  93.         DBPoolConnection poolConnection = new DBPoolConnection();  
  94.         poolConnection.connNbr=getConnNbr();  
  95.         if(poolConnection.connNbr==-1){  
  96.             throw new DBPoolIsFullException("连接池已满");  
  97.         }else{  
  98.             poolConnection.conn=getConn(poolConnection.connNbr);  
  99.         }  
  100.         return poolConnection;  
  101.     }  
  103.     public void freePoolConn(DBPoolConnection poolConnection){  
  104.         if(poolConnection==null){  
  105.             System.out.println("poolConnection==null，不需要释放");  
  106.         }else{  
  107.             freeConn(poolConnection.connNbr);  
  108.         }  
  109.     }  
  110.       
  111.     public void printPoolStatus(){  
  112.         for(int i=0;i<this.connStatus.length;i++){  
  113.             System.out.println("");  
  114.             System.out.print(this.connStatus[i].toString());  
  115.             if(this.lastQueryTime[i]==null){  
  116.                 System.out.print("-[null] ");  
  117.             }else{  
  118.                 System.out.print("-["+this.lastQueryTime[i].toString()+"] ");  
  119.             }  
  120.         }  
  121.         System.out.println("");  
  122.     }  
  123.       
  124.     public String getCls(){  
  125.         return this.cls;  
  126.     }  
  128.     public String getUrl(){  
  129.         return this.url;  
  130.     }  
  132.     public String getUsr(){  
  133.         return this.usr;  
  134.     }  
  136.     int getConnNbr(){  
  137.         int iConn=-1;  
  138.         for(int i=0;i<this.connCount;i++){  
  139.             if(this.connStatus[i].equals("Y")){  
  140.                 this.lastQueryTime[i]=new Date();  
  141.                 this.connStatus[i]="N";  
  142.                 iConn=i;  
  143.                 break;  
  144.             }  
  145.         }  
  146.         return iConn;  
  147.     }  
  148.       
  149.     Connection getConn(int i){  
  150.         return this.connections[i];  
  151.     }  
  152.           
  153.     void closeConnForTest(DBPoolConnection poolConnection){  
  154.         try{  
  155.             this.connections[poolConnection.connNbr].close();  
  156.         }catch(SQLException e){  
  157.             System.out.println(e.getMessage());   
  158.         }  
  159.     }  
  160.           
  161.     boolean checkConn(DBPoolConnection poolConnection){  
  162.         Statement stmt=null;  
  163.         String checkMessage="";  
  164.         boolean checkResult=true;  
  165.           
  166.         //检查连接是否有效 
  167.         try{  
  168.             String sql = "select * from dual";  
  169.             stmt = this.connections[poolConnection.connNbr].createStatement();  
  170.             stmt.executeQuery(sql);//execute sql 
  171.             stmt.close();  
  172.             checkMessage = "checkConn:checkMessage:execute sql success";  
  173.             System.out.println(checkMessage);  
  174.         }catch(Exception e){  
  175.             checkMessage = e.getMessage();  
  176.             System.out.println(e.getMessage());//other exceptions 
  177.             if(checkMessage==null){  
  178.                 checkMessage="e.getMessage() is null";  
  179.                 System.out.println(checkMessage);  
  180.             }  
  181.             //采取激进重连的策略，尽量避免业务中断 
  182.             if (checkMessage.indexOf("ORA-00942")>=0){  
  183.                 checkResult=true;//不需要重连 
  184.             }else if(checkMessage.indexOf("does not exist")>=0){  
  185.                 checkResult=true;//不需要重连 
  186.             }else if(checkMessage.indexOf("Syntax error")>=0){  
  187.                 checkResult=true;//不需要重连 
  188.             }else{            
  189.                 checkResult=false;//需要重连 
  190.             }  
  191.         }  
  192.         return checkResult;  
  193.     }  
  194.       
  195.     boolean resetConn(DBPoolConnection poolConnection){  
  196.         boolean result=false;//默认不需要重建连接  
  197.           
  198.         if(poolConnection==null){  
  199.             System.out.println("poolConnection==null，不知道您想重设哪个连接");  
  200.         }else if(poolConnection.connNbr==-1){  
  201.             System.out.println("poolConnection.connNbr==-1，不知道您想重设哪个连接");  
  202.         }else{  
  203.             if(checkConn(poolConnection)==true){  
  204.                 System.out.println("连接有效，不需要重设");  
  205.             }else{  
  206.                 //重设连接 
  207.                 try{  
  208.                     Class.forName(this.cls);//register class 
  209.                 }catch(ClassNotFoundException e){  
  210.                     System.out.println(e.getMessage());  
  211.                 }catch(Exception e){  
  212.                     System.out.println(e.getMessage());//other exceptions 
  213.                 }  
  214.                 try{  
  215.                     this.connections[poolConnection.connNbr]=DriverManager.getConnection(this.url, this.usr, this.pss);  
  216.                     this.connStatus[poolConnection.connNbr]="Y";  
  217.                     System.out.println(poolConnection.connNbr+"连接已重建");  
  218.                     result = true;//告知调用者连接已重建 
  219.                 }catch(SQLException e){  
  220.                     System.out.println(e.getMessage());  
  221.                 }catch(Exception e){  
  222.                     System.out.println(e.getMessage());//other exceptions 
  223.                 }     
  224.             }  
  225.         }  
  226.         return result;  
  227.     }  
  228.           
  229.     void freeConn(int i){  
  230.         try{  
  231.             if(i==-1){  
  232.                 System.out.println("i=-1，不需要释放");  
  233.             }else{  
  234.                 this.connections[i].commit();  
  235.             }  
  236.         }catch(SQLException e){  
  237.             System.out.println(e.getMessage());  
  238.         }catch(Exception e){  
  239.             System.out.println(e.getMessage());//other exceptions 
  240.         }  
  241.         this.connStatus[i]="Y";  
  242.     }  
  243. }

31/3123>
《2023软件测试行业现状调查报告》独家发布~

热门推荐

搜索风云榜

测试技术了解

2023测试行业调查报告

挣点稿费

AI与软件测试

关注51Testing

相关阅读

热门标签

博文推荐

热点聚焦

最近讲堂

友情链接

51Testing软件测试网 蓝色理想 软件测试培训 太平洋软件 云计算 第三媒体数码 数据恢复 新客网 Java开源大全 数据分析

联系我们

快捷面板 站点地图 联系我们 广告服务 关于我们 站长统计 发展历程

法律顾问：上海兰迪律师事务所 项棋律师
版权所有 上海博为峰软件技术股份有限公司 Copyright©51testing.com 2003-2024
投诉及意见反馈：webmaster@51testing.com; 业务联系：service@51testing.com 021-64471599-8017

沪ICP备05003035号

沪公网安备 31010102002173号