自定义数据库连接池

简介: 自定义数据库连接池

时间很快就到周末了,学习计划也已经进行了五天了,既然是周末的话,那当然要多学习一点知识,毕竟拥有这么充裕的时间。

今天的学习内容是数据库连接池。
那什么是数据库连接池,它有什么作用是我们首先会想到的问题。
数据库连接池负责分配、管理和释放数据库连接,它允许应用程序重复使用一个现有的数据库连接,而不是再重新建立一个。

应用程序直接获取连接的缺点:
用户每次请求都需要向数据库获得链接,而数据库创建连接通常需要消耗相对较大的资源,创建时间也较长。假设网站一天10万访问量,数据库服务器就需要创建10万次连接,极大的浪费数据库的资源,并且极易造成数据库服务器内存溢出、拓机。

缺点显而易见,应用程序在创建连接和销毁连接的时候是极其消耗资源的,而使用数据库连接池则能够优化程序性能。

连接池原理:
在服务器端一次性地创建多个连接,将多个连接保存在一个连接池对象中,当请求需要操作数据库时,不会为请求创建新的连接,而是直接从连接池中获得一个连接。当操作数据库结束,并不需要真正的去关闭连接,而是将连接放回到连接池中。

了解了数据库连接池的优点后,我们关心的是该如何去实现数据库连接池呢?
在Java中提供了javax.sql.Datasource接口用于实现数据库连接池。
老话说得好,光说不练假把式,只练不打无用功,现在我们就来写一个程序感受一下。

在MyEclipse中新建一个web项目,取名demo。
新建一个类,取名MyDataSource,然后实现DataSource接口,要实现的方法非常多,但是不用紧张,我们只关注两个方法。

public Connection getConnection() throws SQLException {
    return null;
}

public Connection getConnection(String username, String password)
        throws SQLException {
    return null;
}

方便起见,我们只看无参的getConnection()方法。自定义的连接池需要有如下功能

  • 一次性地创建多个连接
  • 实现getConnection方法,从连接池获得一个连接
  • 当用户使用连接后,提供方法将连接放回到连接池中

代码如下:

/**
 * 自定义连接池
 * 
 *  一次性地创建多个连接
 *  
 *  实现getConnection方法,从连接池获得一个连接
 *  
 *  当用户使用连接后,提供方法将连接放回到连接池中
 *  
 * @author Administrator
 * 
 */
public class MyDataSource implements DataSource {
    
    private LinkedList<Connection> dataSources = new LinkedList<Connection>();
    
    public MyDataSource(){
        //一次性创建10个连接
        for(int i = 0;i < 10;i++){
            try {
                Connection connection = JDBCUtils.getConnection();
                //将连接加入到连接池中
                dataSources.add(connection);
            } catch (Exception e) {
                e.printStackTrace();
            }
            
        }
    }

    public PrintWriter getLogWriter() throws SQLException {
        return null;
    }

    public void setLogWriter(PrintWriter out) throws SQLException {

    }

    public void setLoginTimeout(int seconds) throws SQLException {

    }

    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return null;
    }

    public <T> T unwrap(Class<T> iface) throws SQLException {
        return null;
    }

    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return false;
    }

    public Connection getConnection() throws SQLException {
        //取出连接池中的一个连接
        Connection connection = dataSources.removeFirst();//删除第一个连接并返回
        return connection;
    }

    public Connection getConnection(String username, String password)
            throws SQLException {
        return null;
    }
}

代码中的JDBCUtils是我编写的一个工具类,在之前的博客中也都有提及,为了方便大家,我就再贴一次。

/**
 * JDBC 工具类,抽取公共方法
 * 
 * @author seawind
 * 
 */
public class JDBCUtils {
    private static final String DRIVERCLASS;
    private static final String URL;
    private static final String USER;
    private static final String PWD;

    static {
        ResourceBundle bundle = ResourceBundle.getBundle("dbconfig");
        DRIVERCLASS = bundle.getString("DRIVERCLASS");
        URL = bundle.getString("URL");
        USER = bundle.getString("USER");
        PWD = bundle.getString("PWD");
    }

    // 建立连接
    public static Connection getConnection() throws Exception {
        loadDriver();
        return DriverManager.getConnection(URL, USER, PWD);
    }

    // 装载驱动
    private static void loadDriver() throws ClassNotFoundException {
        Class.forName(DRIVERCLASS);
    }

    // 释放资源
    public static void release(ResultSet rs, Statement stmt, Connection conn) {
        if (rs != null) {
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
            rs = null;
        }

        release(stmt, conn);
    }

    public static void release(Statement stmt, Connection conn) {
        if (stmt != null) {
            try {
                stmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
            stmt = null;
        }
        if (conn != null) {
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
            conn = null;
        }
    }
}

现在新建一个测试类,取名MyDataSourceTest

public class MyDataSourceTest {
    public static void main(String[] args) throws SQLException {
        //创建连接池
        MyDataSource dataSource = new MyDataSource();
        //从连接池中获得一个连接
        Connection connection = dataSource.getConnection();
        //操作数据库
        String sql = "select * from account";
        PreparedStatement stmt = connection.prepareStatement(sql);
        ResultSet rs = stmt.executeQuery();
        while (rs.next()) {
            System.out.println(rs.getString("name"));
        }
        JDBCUtils.release(stmt, connection);
    }
}

运行测试类
在这里插入图片描述
控制台成功输出用户姓名。
但是这段程序是有问题的,因为JDBCUtils工具类中的release()方法会将连接关闭,而我们的想法是将连接归还到连接池而不是关闭它。实现方法有很多种,你可以在自定义连接池MyDataSource中添加一个方法用于归还连接。
重新修改MyDataSource类,代码如下

/**
 * 自定义连接池
 * 
 *  一次性地创建多个连接
 *  
 *  实现getConnection方法,从连接池获得一个连接
 *  
 *  当用户使用连接后,提供方法将连接放回到连接池中
 *  
 * @author Administrator
 * 
 */
public class MyDataSource implements DataSource {
    
    private LinkedList<Connection> dataSources = new LinkedList<Connection>();
    
    public MyDataSource(){
        //一次性创建10个连接
        for(int i = 0;i < 10;i++){
            try {
                Connection connection = JDBCUtils.getConnection();
                //将连接加入到连接池中
                dataSources.add(connection);
            } catch (Exception e) {
                e.printStackTrace();
            }
            
        }
    }

    public PrintWriter getLogWriter() throws SQLException {
        return null;
    }

    public void setLogWriter(PrintWriter out) throws SQLException {

    }

    public void setLoginTimeout(int seconds) throws SQLException {

    }

    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return null;
    }

    public <T> T unwrap(Class<T> iface) throws SQLException {
        return null;
    }

    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return false;
    }
    
    /**
     * 添加一个方法,用于归还连接
     * @param conn
     */
    public void releaseConnection(Connection conn){
        dataSources.add(conn);
        System.out.println("将连放回到连接池中,数量" + dataSources.size());
    }

    public Connection getConnection() throws SQLException {
        //取出连接池中的一个连接
        Connection connection = dataSources.removeFirst();//删除第一个连接并返回
        System.out.println("取出一个连接,剩余" + dataSources.size() + "个连接");
        return connection;
    }

    public Connection getConnection(String username, String password)
            throws SQLException {
        return null;
    }
}

测试类中最后就应该调用你刚刚添加的方法,运行测试类
在这里插入图片描述
结果很明显,目的达到了。
这样就实现了一个简易的数据库连接池,但该程序其实有一个很不好的地方,因为如果想要将连接放回连接池而不是关闭它,就得调用在MyDataSource类中自己添加的方法,而很多人或许已经习惯了JDBC的编程,会很习惯地去调用close()方法,而一旦调用了close()方法,连接就被关闭了,不会放回连接池了。所以你就需要通知用户去主动使用你添加的API,这种方法其实是不可取的。那有什么办法能够让调用者随着自己的编程习惯去调用close()方法的同时,还能够将连接放回连接池而不是关闭它呢?
我们可以去修改close()方法原来的逻辑,怎么修改呢?
在Java中有三种方法可以增强原有的方法

  • 类继承 、方法覆盖

必须控制对象创建,才能使用该方式

  • 装饰者模式方法加强

必须和目标对象实现相同接口或继续相同父类,特殊构造器(传入被包装对象)

  • 动态代理

有关方法增强的问题,可以参考我的这篇博客。
理解Java方法增强
熟悉方法增强的小伙伴们可以忽略哈。(手动滑稽)

我们使用动态代理来对Connection接口的run()方法进行增强。
修改MyDataSource类中的代码

/**
 * 自定义连接池
 * 
 * 一次性地创建多个连接
 * 
 * 实现getConnection方法,从连接池获得一个连接
 * 
 * 当用户使用连接后,提供方法将连接放回到连接池中
 * 
 * @author Administrator
 * 
 */
public class MyDataSource implements DataSource {

    private LinkedList<Connection> dataSources = new LinkedList<Connection>();

    public MyDataSource() {
        // 一次性创建10个连接
        for (int i = 0; i < 10; i++) {
            try {
                Connection connection = JDBCUtils.getConnection();
                // 将连接加入到连接池中
                dataSources.add(connection);
            } catch (Exception e) {
                e.printStackTrace();
            }

        }
    }

    public PrintWriter getLogWriter() throws SQLException {
        return null;
    }

    public void setLogWriter(PrintWriter out) throws SQLException {

    }

    public void setLoginTimeout(int seconds) throws SQLException {

    }

    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return null;
    }

    public <T> T unwrap(Class<T> iface) throws SQLException {
        return null;
    }

    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return false;
    }

    /**
     * 添加一个方法,用于归还连接
     * 
     * @param conn
     */
    public void releaseConnection(Connection conn) {
        dataSources.add(conn);
        System.out.println("将连放回到连接池中,数量" + dataSources.size());
    }

    public Connection getConnection() throws SQLException {
        // 取出连接池中的一个连接
        final Connection connection = dataSources.removeFirst();// 删除第一个连接并返回
        System.out.println("取出一个连接,剩余" + dataSources.size() + "个连接");
        // 将目标connection对象进行增强
        Connection connProxy = (Connection) Proxy.newProxyInstance(connection
                .getClass().getClassLoader(), connection.getClass()
                .getInterfaces(), new InvocationHandler() {

            //执行代理对象的任何方法都将执行invoke方法
            public Object invoke(Object proxy, Method method, Object[] args)
                    throws Throwable {
                //只需要增强close方法
                if(method.getName().equals("close")){
                    //需要增强的方法
                    //不将连接真正关闭,将连接放回连接池
                    releaseConnection(connection);
                    return null;
                }else{
                    //不需要增强的方法,保持方法原有的功能即可
                    return method.invoke(connection, args);
                }
            }
        });
        return connProxy;
    }

    public Connection getConnection(String username, String password)
            throws SQLException {
        return null;
    }
}

回到测试类,此时调用Connection接口中的close()方法,然后执行
在这里插入图片描述
方法成功增强了。调用者在调用close()方法后并不会把连接关闭了,而是放回连接池中。由此我们的想法便实现了。

目录
相关文章
|
6月前
|
关系型数据库 MySQL 数据库
第十四章 演示MYSQL自定义values.yaml绑定PV和PVC和数据库用户密码
第十四章 演示MYSQL自定义values.yaml绑定PV和PVC和数据库用户密码
78 0
|
XML 开发框架 前端开发
J2EE之自定义框架知识(下篇 综合运用增删改查)
J2EE之自定义框架知识(下篇 综合运用增删改查)
63 0
|
2月前
|
数据库 Python
django中数据库外键可以自定义名称吗
django中数据库外键可以自定义名称吗
|
3月前
|
SQL Shell API
python Django教程 之 模型(数据库)、自定义Field、数据表更改、QuerySet API
python Django教程 之 模型(数据库)、自定义Field、数据表更改、QuerySet API
|
4月前
|
SQL DataWorks 关系型数据库
DataWorks产品使用合集之数据集成时源头提供数据库自定义函数调用返回数据,数据源端是否可以写自定义SQL实现
DataWorks作为一站式的数据开发与治理平台,提供了从数据采集、清洗、开发、调度、服务化、质量监控到安全管理的全套解决方案,帮助企业构建高效、规范、安全的大数据处理体系。以下是对DataWorks产品使用合集的概述,涵盖数据处理的各个环节。
|
3月前
|
存储 运维 Java
SpringBoot使用log4j2将日志记录到文件及自定义数据库
通过上述步骤,你可以在Spring Boot应用中利用Log4j2将日志输出到文件和数据库中。这不仅促进了良好的日志管理实践,也为应用的监控和故障排查提供了强大的工具。强调一点,配置文件和代码的具体实现可能需要根据应用的实际需求和运行环境进行调优和修改,始终记住测试配置以确保一切运行正常。
605 0
|
6月前
|
Cloud Native 关系型数据库 分布式数据库
云原生数据库PolarDB自定义权限策略参考
本文介绍了自定义权限策略,它是RAM访问控制体系中除系统策略外,用户可创建、更新和删除的权限策略。用户需维护策略版本更新,并将策略绑定到RAM身份以授予权限。策略支持版本控制和删除,但删除前需确保无引用。文章提供了相关操作文档链接,包括创建、修改、删除策略及管理引用记录和版本。此外,列举了PolarDB在MySQL、PostgreSQL和XScale版的自定义权限策略场景和示例,并提示用户在使用时需了解授权信息。
59 0
|
6月前
|
存储 关系型数据库 MySQL
【mybatis-plus】Springboot+AOP+自定义注解实现多数据源操作(数据源信息存在数据库)
【mybatis-plus】Springboot+AOP+自定义注解实现多数据源操作(数据源信息存在数据库)
|
Java
jpa实现增删改查,分页,自定义查询,jpql查询
jpa实现增删改查,分页,自定义查询,jpql查询
88 0
|
6月前
|
SQL 存储 定位技术
数据库基础(七):用户自定义数据类型与标量值、内嵌表值、多语句表值函数等用户定义函数的创建、使用、删除
数据库基础(七):用户自定义数据类型与标量值、内嵌表值、多语句表值函数等用户定义函数的创建、使用、删除
125 1