不用Spring如何正確使用Druid連接池數據源

阿里巴巴的開源數據庫連接池工具Druid數據源，在實際使用的時候，通常會和Spring或SpringBoot整合一起使用。

但有些時候如果不用使用Spring或SpringBoot，而直接使用Druid的原生API來操作數據庫的話，該如何正確使用呢？

下面是一個使用Druid連接池簡單操作數據庫的工具類，重點關注Druid數據源的創建和復用：

/**
 * 數據庫操作工具類
 *
 * @author shiyanjun
 * @since 2020/01/05
 */
@Slf4j
public class DbUtil {
    private static final String DB_URL = "jdbc:mysql://localhost:3306/authapi"; // 數據庫連接URL
    private static final String DB_USERNAME = "root"; // 數據庫用戶名
    private static final String DB_PASSWORD = "123456"; // 數據庫密碼

    // Druid數據源，全局唯一（只創建一次）
    private static DruidDataSource druidDataSource;

    /**
     * 執行SQL更新
     *
     * @param updateSql
     * @throws SQLException
     */
    public static void insert(String updateSql) throws SQLException {
        Connection connection = null;
        Statement statement = null;
        ResultSet resultSet = null;
        try {
            connection = getDruidConnection();
            statement = connection.createStatement();
            int count = statement.executeUpdate(updateSql);
            log.info(">>>>>>>>>>>>> 插入數據 {}", count);
        } finally {
            // 切記!!! 一定要釋放資源
            closeResource(connection, statement, resultSet);
        }
    }

    /**
     * 執行SQL查詢
     *
     * @param querySql
     * @return
     * @throws Exception
     */
    public static List<Map<String, Object>> executeQuery(String querySql) throws Exception {
        List<Map<String, Object>> resultList = new ArrayList<>();
        Connection connection = null;
        Statement statement = null;
        ResultSet resultSet = null;
        try {
            connection = getDruidConnection();
            statement = connection.createStatement();
            resultSet = statement.executeQuery(querySql);
            ResultSetMetaData metaData = resultSet.getMetaData();
            while (resultSet.next()) {
                int columnCount = metaData.getColumnCount();
                Map<String, Object> resultMap = new LinkedHashMap<>();
                for (int i = 1; i <= columnCount; i++) {
                    String columnName = metaData.getColumnName(i);// 字段名稱
                    Object columnValue = resultSet.getObject(columnName);// 字段值
                    resultMap.put(columnName, columnValue);
                }
                resultList.add(resultMap);
            }
            log.info(">>>>>>>>>>>> 查詢數據:{}", resultList);
        } finally {
            // 切記!!! 一定要釋放資源
            closeResource(connection, statement, resultSet);
        }
        return resultList;
    }

    /**
     * 獲取Druid數據源
     *
     * @return
     * @throws SQLException
     */
    private static DruidDataSource getDruidDataSource() throws SQLException {
        // 保證Druid數據源在多線程下只創建一次
        if (druidDataSource == null) {
            synchronized (DbUtil.class) {
                if (druidDataSource == null) {
                    druidDataSource = createDruidDataSource();
                    return druidDataSource;
                }
            }
        }
        log.info(">>>>>>>>>>> 復用Druid數據源:url={}, username={}, password={}",
                druidDataSource.getUrl(), druidDataSource.getUsername(), druidDataSource.getPassword());
        return druidDataSource;
    }

    /**
     * 創建Druid數據源
     *
     * @return
     * @throws SQLException
     */
    private static DruidDataSource createDruidDataSource() throws SQLException {
        DruidDataSource druidDataSource = new DruidDataSource();
        druidDataSource.setUrl(DB_URL);
        druidDataSource.setUsername(DB_USERNAME);
        druidDataSource.setPassword(DB_PASSWORD);

        /*----下面的具體配置參數自己根據項目情況進行調整----*/
        druidDataSource.setMaxActive(20);
        druidDataSource.setInitialSize(1);
        druidDataSource.setMinIdle(1);
        druidDataSource.setMaxWait(60000);

        druidDataSource.setValidationQuery("select 1 from dual");

        druidDataSource.setTimeBetweenEvictionRunsMillis(60000);
        druidDataSource.setMinEvictableIdleTimeMillis(300000);

        druidDataSource.setTestWhileIdle(true);
        druidDataSource.setTestOnBorrow(false);
        druidDataSource.setTestOnReturn(false);

        druidDataSource.setPoolPreparedStatements(true);
        druidDataSource.setMaxPoolPreparedStatementPerConnectionSize(20);

        druidDataSource.init();
        log.info(">>>>>>>>>>> 創建Druid數據源:url={}, username={}, password={}",
                druidDataSource.getUrl(), druidDataSource.getUsername(), druidDataSource.getPassword());
        return druidDataSource;
    }

    /**
     * 獲取Druid連接
     *
     * @return
     * @throws SQLException
     */
    private static DruidPooledConnection getDruidConnection() throws SQLException {
        DruidDataSource druidDataSource = getDruidDataSource();
        DruidPooledConnection connection = druidDataSource.getConnection();
        return connection;
    }

    /**
     * 釋放資源
     *
     * @param connection
     * @param statement
     * @param resultSet
     * @throws SQLException
     */
    private static void closeResource(Connection connection,
                                      Statement statement, ResultSet resultSet) throws SQLException {
        // 注意資源釋放順序
        if (resultSet != null) {
            resultSet.close();
        }
        if (statement != null) {
            statement.close();
        }
        if (connection != null) {
            connection.close();
        }
    }
}

 

一個簡單的測試類，模擬多線程向數據庫表中插入1w條數據：

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * CREATE TABLE `auth_code` (
 * `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '自增主鍵',
 * `auth_code` varchar(255) CHARACTER SET utf8mb4 NOT NULL COMMENT '授權碼',
 * `status` varchar(255) CHARACTER SET utf8mb4 NOT NULL DEFAULT '1' COMMENT '狀態',
 * `remark` varchar(255) CHARACTER SET utf8mb4 DEFAULT NULL COMMENT '備注',
 * `create_date` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '創建日期',
 * `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '修改日期',
 * PRIMARY KEY (`id`)
 * ) ENGINE=InnoDB AUTO_INCREMENT=18 DEFAULT CHARSET=latin1;
 */
public class AppTest {
    private static SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
    private static ExecutorService executorService = Executors.newCachedThreadPool();// 線程池

    public static void main(String[] args) throws Exception {
        String INSERT_SQL = "INSERT INTO `authapi`.`auth_code` " +
                "(`auth_code`, `status`, `remark`, `create_date`, `update_time`) VALUES ('%s',%s,'%s','%s','%s')";
        // 模擬多線程向數據庫插入1萬條數據
        for (int i = 0; i < 10000; i++) {
            Thread.sleep(5);
            String code = UUID.randomUUID().toString().replaceAll("-", "");
            String dateStr = dateFormat.format(new Date());
            String formatSql = String.format(INSERT_SQL, code, i, code, dateStr, dateStr);
            executorService.execute(() -> {
                try {
                    DbUtil.insert(formatSql);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
        }
    }
}