玩转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();
  }
}

结果：