玩转MySQL ----- 自己写的数据库连接池引入线程池

这里继续更JDBC的工具类，应该是V5版本了：

V5版本引入了线程池，为了后面项目service层方便进行事务操作：

创建线程池：
 private static ThreadLocal<Connection> t=new ThreadLocal<Connection>();

package cn.hncu.pub;

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.util.ArrayList;
import java.util.List;
import java.util.Properties;


//数据库连接池---v5---用户直接调用con.close()就可以实现还回池中---采用动态代理模式增强conn
public class Conn5Utils {
    //创建一个池
	private static List<Connection> pool=new ArrayList<Connection>();
	private static int NUM=3; 
	
	//创建线程池
	private static ThreadLocal<Connection> t=new ThreadLocal<Connection>();
	
	
	static{
		
		try {
			Properties p=new Properties();
			p.load(Conn5Utils.class.getClassLoader().getResourceAsStream("jdbc.properties"));
            
			String driver=p.getProperty("driver");
			String url=p.getProperty("url");
			String user=p.getProperty("username");
			String pwd=p.getProperty("password");
			
			Class.forName(driver);
			
			for(int i=0;i<NUM;i++){
				final Connection conn=DriverManager.getConnection(url, user, pwd);
				Object proxyObj=Proxy.newProxyInstance(
						Conn5Utils.class.getClassLoader(),
						//conn.getClass().getInterfaces(), //坑:不能替代下一行，因为mysql中使用了自己的类加载器
						new Class[]{Connection.class}, 
						new InvocationHandler() {
							
							@Override
							public Object invoke(Object proxy, Method method, Object[] args)
									throws Throwable {
								//把close方法拦截并更改成把“代理后的对象”放回池中的动作
								if("close".equals(method)){
									pool.add((Connection) proxy);
									System.out.println("往池中还回来一个连接。。。");
									return null; //跳过下面的放行，这样原有的close()功能就没了
								}
								System.out.println();
								
								
								return method.invoke(conn, args);//放行
							}
						});
				
	                   pool.add((Connection) proxyObj); //将代理后的对象添加到池子中
			}
		} catch (Exception e) {
		     throw new RuntimeException(e.getMessage(),e);
		}
		
		
	}
	
	public static synchronized Connection getConn(){
		 
		//先从t池中拿，若没有才到pool中去拿新的
		Connection conn = t.get();
		
		if(conn==null){
			if(pool.size()<=0){
				System.out.println("池中已经没有了,请稍候再连");
				try {
					Thread.sleep(1000);
				} catch (InterruptedException e) {
					throw new RuntimeException(e.getMessage(),e);
				}
				return getConn();
			}
			System.out.println("从池中拿走一个...");
			conn=pool.remove(0);
			t.set(conn);//把这个新的存入t池
		}
		return conn;
	}
}

 

测试类：

package cn.hncu.tx;

import java.sql.Connection;

import cn.hncu.pub.Conn5Utils;

public class Conn5UtilsTest {

	public static void main(String[] args) {
		Connection conn = Conn5Utils.getConn();
		Connection conn2 = Conn5Utils.getConn();
		Connection conn3 = new Hello().aa();

		System.out.println(conn);
		System.out.println(conn2);
		System.out.println(conn3);

		new Thread() {
			/*
			 * (non-Javadoc)
			 * 
			 * @see java.lang.Thread#run()
			 */
			@Override
			public void run() {
				System.out.println(">>" + Conn5Utils.getConn());
				Connection con = new Hello().aa();
				System.out.println("con>>" + con);
			}

		}.start();

	}

}

class Hello {
	public Connection aa() {
		return Conn5Utils.getConn();
	}
}

结果：