MySQL学习(六)——自定义连接池

1、连接池概念

用池来管理Connection，这样可以重复使用Connection。有了池，我们就不用自己来创建Connection，而是通过池来获取Connection对象。当使用完Connection后，调用Connection的close()方法也不会真的关闭Connection，而是把Connection“归还”给池，池就可以再利用这个Connection对象了。

2、自定义连接池

1）基于MySQL学习(五)——使用JDBC完成用户表CRUD的操作，把db.properties和JDBCUtils_V3.java文件复制到当前包下

2）MyConnection.java文件

装饰者设计模式，专门用于增强方法（此处如果不对close()做增强方法，用户调用conn.close()将连接真正释放，连接池将无连接可用，所以我们希望用户调用了close()方法，连接仍归还给连接池）

//1.实现同一个接口
public class MyConnection implements Connection{
    //3.定义变量
    private Connection conn;
    
    private LinkedList<Connection> pool;
    
    //2.编写一个构造方法(参数使用了面向对象的多态特性)
    public    MyConnection(Connection conn,LinkedList<Connection> pool){
        this.conn=conn;
        this.pool=pool;
    }
    //4.书写需要增强的方法
    @Override
    public void close() throws SQLException {
        pool.add(conn);
    }
    /*
     * 此方法必须覆盖！否则会出现空指针异常
     */
    @Override
    public PreparedStatement prepareStatement(String sql) throws SQLException {
        return conn.prepareStatement(sql);
    }
    ……
}

3）MyDataSource1.java文件

创建连接池实现数据源，并实现接口javax.sql.DataSource

public class MyDataSource1 implements DataSource{
    //1.创建一个容器用于存储Connection对象
    private static LinkedList<Connection> pool=new LinkedList<Connection>();
    //2.创建5个连接放到容器中去
    static{
        for(int i=0;i<5;i++){
            Connection conn=JDBCUtils_V3.getConnection();
            //放入池子中的connection对象已经经过改造了
            MyConnection myconn=new MyConnection(conn,pool);
            pool.add(myconn);
        }
    }
    /**
     * 重写获取连接的方法
     */
    @Override
    public Connection getConnection() throws SQLException {
        Connection conn=null;
        //3.使用前先判断
        if(pool.size()==0){
            //3.1池子里没有，我们再创建一些
            conn=JDBCUtils_V3.getConnection();
            pool.add(conn);
        }
        //4.从池子里面获取一个连接对象Connection
        conn=pool.remove(0); 
        return conn;
    }
    ……
}

4）TestMyDataSource.java文件

public class TestMyDataSource {
    /*
     * 添加用户
     * 使用改造过后的Connection
     */ 
    @Test
    public void testAddUser1(){
        Connection conn=null;
        PreparedStatement pstmt=null;
        //1.创建自定义连接池对象
        DataSource dataSource=new MyDataSource1();
        
        try {
            //2.从池子中获取连接
            conn=dataSource.getConnection();
            String sql="insert tbl_user values(null,?,?)";
            //必须在自定义的conntection类中重写prepareStatement(sql)语句
            pstmt=conn.prepareStatement(sql);
            pstmt.setString(1, "chenga");
            pstmt.setString(2, "123");
            int rows=pstmt.executeUpdate();
            if(rows>0){
                System.out.println("添加成功");
            }else{
                System.out.println("添加不成功");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }finally{
            //调用的release()方法中的close()其实是MyConnection的close()方法，该方法内部将当前连接归还到连接池
            JDBCUtils_V3.release(conn,pstmt,null);         
        }
    }
}

执行Junit Test，效果如下：