自定义连接池

布满荆棘的人生 2023-10-17 17:52 144阅读 0赞

package com.hong.dao;
    
    import java.lang.reflect.InvocationHandler;
    import java.lang.reflect.Method;
    import java.lang.reflect.Proxy;
    import java.sql.Connection;
    import java.sql.DriverManager;
    import java.sql.SQLException;
    import java.util.LinkedList;
    import java.util.List;
    
    public class CustomizePool {
    
    	/***
    	 * 定义变量存储初始化数目，最大连接数，当前创建的连接数目
    	 * 初始化一个连接池（定义一个函数获取连接）（获取连接时采取动态代理将close增强）
    	 * 获取连接
    	 * ！！！先判断连接池中是否还有连接
    	 * ！！！有则直接拿没有则继续判断当前已经创建的连接数是否小于最大连接数
    	 * ！！！是则创建连接没有则抛出异常
    	 * 释放连接
    	 * ！！！先判断连接池中的连接数是否小于初始化时的连接数
    	 * ！！！是则放回连接池否则就关闭连接当前已创建连接数减一
    	 * 定义一个main方法实验一把
    	 */
    	private int init_count = 3;
    	private int max_count = 6;
    	private int current_count = 0;
    	private LinkedList<Connection> list = new LinkedList<Connection>();
    	
    	public CustomizePool(){
    		for(int i=0;i<init_count;i++){
    			current_count++;
    			list.addLast(createConn());
    		}
    	}
    
    	private Connection createConn() {
    		try {
    			Class.forName("com.mysql.jdbc.Driver");
    			final Connection conn=DriverManager.getConnection("jdbc:mysql:///test", "root", "825z.jkc76");
    			//使用动态代理包装一下close方法，事件驱动
    			Connection newConn =(Connection)Proxy.newProxyInstance(
    					conn.getClass().getClassLoader(), 
    					new Class[]{Connection.class}, //不能使用getInterfaces，这里使用new Class[]{Connection.class}作为一个接口
    					new InvocationHandler() {//事件驱动，封装连接
    						
    						@Override
    						public Object invoke(Object proxy, Method method, Object[] args)
    								throws Throwable {
    							// TODO Auto-generated method stub
    							String methodName = method.getName();
    							if("close".equals(methodName)){
    								if(list.size()<init_count){
    									list.addLast(conn);
    								}else{
    									current_count--;
    									try {
    										conn.close();
    									} catch (SQLException e) {
    										throw new RuntimeException(e);
    									}
    								}
    							}else{
    								return method.invoke(conn, args);
    							}
    							return null;
    						}
    					}
    					);
    			return newConn;
    		} catch (ClassNotFoundException e) {
    			throw new RuntimeException(e);
    		} catch (SQLException e) {
    			throw new RuntimeException(e);
    		}
    	}
    	private Connection getConn(){
    		if(list.size()>0){
    			return list.removeFirst();
    		}else{
    			if(current_count<max_count){
    				current_count++;
    				return createConn();
    			}else{
    				throw new RuntimeException("已经超过最大连接数？。？");
    			}
    		}
    	}
    	private void releaseConnection(Connection conn){
    		if(list.size()<init_count){
    			list.addLast(conn);
    		}else{
    			current_count--;
    			try {
    				conn.close();
    			} catch (SQLException e) {
    				throw new RuntimeException(e);
    			}
    		}
    	}
    	public static void main(String args[]){
    		CustomizePool pool = new CustomizePool();
    		pool.getConn();
    		pool.getConn();
    		pool.getConn();
    		pool.getConn();
    		pool.getConn();
    		Connection conn = pool.getConn();
    		try {
    			conn.close();
    		} catch (SQLException e) {
    			throw new RuntimeException(e);
    		}
    		pool.getConn();
    		System.out.println("当前连接池还有的连接"+pool.list.size());
    		System.out.println("当前连接池最大连接"+pool.max_count);
    		System.out.println("当前已经创建的连接数"+pool.current_count);
    	}
    }