手写 mysql 连接池_带你手写一个数据库连接池

importorg.springframework.scheduling.annotation.Scheduled;importjava.sql.Connection;importjava.sql.SQLException;importjava.util.concurrent.LinkedBlockingQueue;importjava.util.concurrent.TimeUnit;importjava.util.concurrent.atomic.AtomicInteger;/*** 数据库连接池实现*/

public class DbPoolImpl implementsDbPool {/*** 空闲连接池*/

private LinkedBlockingQueueidleConnectPool;/*** 活跃连接池*/

private LinkedBlockingQueuebusyConnectPool;/*** 当前正在被使用的连接数*/

private AtomicInteger activeSize = new AtomicInteger(0);/*** 最大连接数*/

private final intmaxSize;public DbPoolImpl(intmaxSize) {this.maxSize =maxSize;

init();//init pool

}/*** 连接池初始化*/@Overridepublic voidinit() {

idleConnectPool= new LinkedBlockingQueue<>();

busyConnectPool= new LinkedBlockingQueue<>();

}/*** 获取一个连接

*

*@return

*/@OverridepublicConnection getConnection() {//从idle池中取出一个连接

Connection connection =idleConnectPool.poll();if (connection != null) {//如果有连接，则放入busy池中

busyConnectPool.offer(connection);

System.out.println("获取到连接");returnconnection;

}//synchronized (DbPoolImpl.class) {//锁--效率低下//idle池中没有连接//如果idle池中连接未满maxSize，就新建一个连接

if (activeSize.get()

if (activeSize.incrementAndGet() <=maxSize) {

connection= DbUtil.createConnection();//创建新连接

busyConnectPool.offer(connection);returnconnection;

}

}//}//如果空闲池中连接数达到maxSize， 则阻塞等待归还连接

try{

System.out.println("排队等待连接");

connection= idleConnectPool.poll(10000, TimeUnit.MILLISECONDS);//阻塞获取连接，如果10秒内有其他连接释放，

if (connection == null) {

System.out.println("等待超时");throw new RuntimeException("等待连接超时");

}

System.out.println("等待到了一个连接");

}catch(InterruptedException e) {

e.printStackTrace();

}returnconnection;

}/*** 释放一个连接

*

*@paramconnection*/@Overridepublic voidreleaseConnection(Connection connection) {if (connection != null) {

busyConnectPool.remove(connection);

idleConnectPool.offer(connection);

}

}/*** 销毁连接池*/@Overridepublic voiddestroy() {

}/*** 定时对连接进行健康检查

* 注意：只能对idle连接池中的连接进行健康检查，

* 不可以对busyConnectPool连接池中的连接进行健康检查，因为它正在被客户端使用;*///@Scheduled(fixedRate= 60 * 1000)public voidcheck() {for (int i = 0; i < activeSize.get(); i++) {

Connection connection=idleConnectPool.poll();try{boolean valid = connection.isValid(2000);if (!valid) {//如果连接不可用，则创建一个新的连接

connection =DbUtil.createConnection();

}

idleConnectPool.offer(connection);//放进一个可用的连接

} catch(SQLException e) {

e.printStackTrace();

}

}

}

}