最简单的数据库连接池

gjfer

[代码][Java]代码

1. package com.jxva.dao;
   
2. 
   
3. import java.io.PrintWriter;
   
4. import java.lang.reflect.InvocationHandler;
   
5. import java.lang.reflect.Method;
   
6. import java.lang.reflect.Proxy;
   
7. import java.sql.Connection;
   
8. import java.sql.DriverManager;
   
9. import java.sql.SQLException;
   
10. import java.sql.SQLFeatureNotSupportedException;
    
11. import java.util.Date;
    
12. import java.util.concurrent.ConcurrentHashMap;
    
13. import java.util.concurrent.ConcurrentMap;
    
14. import java.util.concurrent.Semaphore;
    
15. 
    
16. import javax.sql.DataSource;
    
17. 
    
18. import com.jxva.log.Logger;
    
19. import com.jxva.util.RandomUtil;
    
20. 
    
21. public class SimpleDataSource implements DataSource{
    
22. 
    
23.     private String driverClass;
    
24.     private String url;
    
25.     private String username;
    
26.     private String password;
    
27.     private int minSize;
    
28.     private int maxSize;
    
29. 
    
30.     private static final Logger log = Logger.getLogger(SimpleDataSource.class);
    
31.     private static final ConcurrentMap<Connection, Date> pool = new ConcurrentHashMap<Connection, Date>();
    
32.     private Semaphore semaphore;
    
33.      
    
34.     public SimpleDataSource() {
    
35.         minSize = 1;
    
36.         maxSize = 5;
    
37.         log.debug("Database Connection Pool initializing...");
    
38.     }
    
39. 
    
40.      
    
41.     public void close(Connection conn) throws SQLException {
    
42.         try{
    
43.             synchronized (pool) {               
    
44.                 if (pool.size() < maxSize) {
    
45.                     pool.put(conn,new Date());
    
46.                     return;
    
47.                 }
    
48.             }
    
49.             conn.close();
    
50.         }finally{
    
51.             semaphore.release();
    
52.         }
    
53.     }
    
54. 
    
55.     public void destroy() throws SQLException {
    
56.         for (Connection conn : pool.keySet()) {
    
57.             conn.close();
    
58.         }
    
59.         pool.clear();
    
60.     }
    
61. 
    
62.     @Override
    
63.     public Connection getConnection() throws SQLException {
    
64.         try {
    
65.             semaphore.acquire();
    
66.         } catch (InterruptedException e1) {
    
67.             //ignore
    
68.         }
    
69.         synchronized (pool) {
    
70.             if (!pool.isEmpty()) {
    
71.                 //Connection conn = pool.keySet().iterator().next();// fetch last connection for ever
    
72.                  Connection conn=(Connection)pool.keySet().toArray()[RandomUtil.getRandomNum(pool.size())]; //random fetch one connection
    
73.                 pool.remove(conn);
    
74.                 try {
    
75.                     conn.setAutoCommit(true);
    
76.                 } catch (SQLException e) {
    
77.                     pool.clear();
    
78.                     conn = getConnection();
    
79.                 }
    
80.                 return getProxyConnection(conn);
    
81.             }
    
82.         }
    
83.         //System.out.println(url);
    
84.         //System.out.println(username+":"+password);
    
85.         //return DriverManager.getConnection("jdbc:mysql://10.204.79.234:3306/library?zeroDateTimeBehavior=convertToNull&useUnicode=true&characterEncoding=utf-8",username,password);
    
86.          
    
87.         //url = "jdbc:mysql://10.204.79.234:3306/library?zeroDateTimeBehavior=convertToNull&useUnicode=true&characterEncoding=utf-8";
    
88.         return getProxyConnection(DriverManager.getConnection(url,username,password));
    
89.     }
    
90. 
    
91.     private Connection getProxyConnection(Connection connection) {
    
92.         final Connection conn = connection;
    
93.         InvocationHandler handler = new InvocationHandler() {
    
94.             public Object invoke(Object proxy, Method method, Object[] params) throws Exception {
    
95.                 Object ret = null;
    
96.                 if ("close".equals(method.getName())) {
    
97.                     close(conn);
    
98.                 } else {
    
99.                     ret = method.invoke(conn, params);
    
100.                 }
     
101.                 return ret;
     
102.             }
     
103.         };
     
104.         return (Connection) Proxy.newProxyInstance(Connection.class.getClassLoader(), new Class[] { Connection.class }, handler);
     
105.     }
     
106. 
     
107.     public String getUrl() {
     
108.         return url;
     
109.     }
     
110. 
     
111.     public void setUrl(String url) {
     
112.         this.url = url;
     
113.     }
     
114. 
     
115.     public String getUsername() {
     
116.         return username;
     
117.     }
     
118. 
     
119.     public void setUsername(String username) {
     
120.         this.username = username;
     
121.     }
     
122. 
     
123.     public String getPassword() {
     
124.         return password;
     
125.     }
     
126. 
     
127.     public void setPassword(String password) {
     
128.         this.password = password;
     
129.     }
     
130. 
     
131.     public int getMinSize() {
     
132.         return minSize;
     
133.     }
     
134. 
     
135.     public void setMinSize(int minSize) {
     
136.         this.minSize = minSize;
     
137.     }
     
138. 
     
139.     public int getMaxSize() {
     
140.         return maxSize;
     
141.     }
     
142. 
     
143.     public void setMaxSize(int maxSize) {
     
144.         semaphore=new Semaphore(maxSize,false);
     
145.         this.maxSize = maxSize;
     
146.     }
     
147. 
     
148.     public String getDriverClass() {
     
149.         return driverClass;
     
150.     }
     
151. 
     
152.     public void setDriverClass(String driverClass) {
     
153.         try {
     
154.             Class.forName(driverClass);
     
155.         } catch (ClassNotFoundException e) {
     
156.             log.error(driverClass+" not exists.");
     
157.             throw new DataSourceException(driverClass+" not exists.",e);
     
158.         }
     
159.         this.driverClass = driverClass;
     
160.     }
     
161. 
     
162.     @Override
     
163.     public PrintWriter getLogWriter() throws SQLException {
     
164.         // TODO Auto-generated method stub
     
165.         return null;
     
166.     }
     
167. 
     
168.     @Override
     
169.     public int getLoginTimeout() throws SQLException {
     
170.         // TODO Auto-generated method stub
     
171.         return 0;
     
172.     }
     
173. 
     
174.     @Override
     
175.     public java.util.logging.Logger getParentLogger()
     
176.             throws SQLFeatureNotSupportedException {
     
177.         // TODO Auto-generated method stub
     
178.         return null;
     
179.     }
     
180. 
     
181.     @Override
     
182.     public void setLogWriter(PrintWriter arg0) throws SQLException {
     
183.         // TODO Auto-generated method stub
     
184.          
     
185.     }
     
186. 
     
187.     @Override
     
188.     public void setLoginTimeout(int arg0) throws SQLException {
     
189.         // TODO Auto-generated method stub
     
190.          
     
191.     }
     
192. 
     
193.     @Override
     
194.     public boolean isWrapperFor(Class<?> arg0) throws SQLException {
     
195.         // TODO Auto-generated method stub
     
196.         return false;
     
197.     }
     
198. 
     
199.     @Override
     
200.     public <T> T unwrap(Class<T> arg0) throws SQLException {
     
201.         // TODO Auto-generated method stub
     
202.         return null;
     
203.     }
     
204. 
     
205.     @Override
     
206.     public Connection getConnection(String arg0, String arg1)
     
207.             throws SQLException {
     
208.         // TODO Auto-generated method stub
     
209.         return null;
     
210.     }
     
211. }

复制代码