Java实战之用hutool-db实现多数据源配置

当我们的应用需要连接多个数据库时，常规的做法是通过JDBC来手动连接和操作不同的数据源。这样的代码通常会显得冗长和重复，对于代码维护和可读性都造成了一定的负担。事实上，Java中有一些工具库可以帮助我们更容易地实现多数据源配置和自动化操作，比如常用的HikariCP、Druid等数据库连接池，还有hutool-db库。

Hutool-db简介

Hutool-db是Hutool生态中的一个开源库，提供了简单易用的JDBC封装和SQLite、MySQL等数据库操作的工具类。其中，hutool-db的数据源支持多种方式，包括：

1. JDBC驱动管理器（默认JDBC连接池方式）：通过驱动管理器管理不同数据源的连接池，可以在配置文件中定义多个数据源；
2. 数据库连接池（可选）：可以使用Druid等第三方连接池管理不同数据源的连接池。

通过多重连接池管理多个数据源的方式，hutool-db可以让我们更容易地实现多数据源的管理和访问。

多数据源配置

在使用hutool-db实现多数据源配置时，我们需要先在项目中引入依赖。具体的引入方式请参考hutool-db官方文档。

引入依赖后，我们需要在配置文件中定义多个数据源。以YAML配置文件为例，假设我们有一个主库和一个从库，定义方式如下：

db:
  default:
    # 主库配置
    url: jdbc:mysql://localhost:3306/mall
    username: root
    password: root
  slave:
    # 从库配置
    url: jdbc:mysql://localhost:3307/mall
    username: root
    password: root

在代码中，我们通过DbUtil.getDb(name)方法获取指定名称的数据库连接。hutool-db会根据配置文件中的数据源配置，自动创建连接。实现代码如下：

// 获取主库连接
Db db = DbUtil.getDb("default");
// 获取从库连接
Db db2 = DbUtil.getDb("slave");

我们可以通过Db.find()方法执行SQL查询，通过Db.use()方法切换当前数据源。示例如下：

// 使用主库
Db.use("default");
// 查询user表的所有记录
List<Entity> userList = Db.find("select * from user");
// 使用从库
Db.use("slave");
// 查询user表的所有记录
List<Entity> userList2 = Db.find("select * from user");

示例一：自定义数据源切换逻辑

有时我们需要自定义数据源的切换逻辑，比如根据请求中的参数判断使用哪个数据源。此时，我们可以通过重写DbConfig类中的determineDbType()方法来实现。

先定义一个枚举类，用来表示数据源类型：

public enum DataSourceType {
    MASTER,
    SLAVE
}

接下来，实现DbConfig的子类，重写determineDbType()方法：

public class MyDbConfig extends DbConfig {
    private DataSourceType dataSourceType = DataSourceType.MASTER;

    // 设置数据源类型
    public void setDataSourceType(DataSourceType dataSourceType) {
        this.dataSourceType = dataSourceType;
    }

    @Override
    public DbType determineDbType(DatabaseMetaData meta) throws SQLException {
        switch (dataSourceType) {
            case MASTER:
                return DbType.MYSQL;
            case SLAVE:
                return DbType.MYSQL_SLAVE;
            default:
                throw new AssertionError(dataSourceType.name());
        }
    }
}

然后在代码中，我们可以通过如下方式切换数据源：

// 使用主库
MyDbConfig dbConfig = new MyDbConfig();
dbConfig.setDataSourceType(DataSourceType.MASTER);
Db db = Db.use(dbConfig);
// 查询user表的所有记录
List<Entity> userList = db.find("select * from user");

// 使用从库
dbConfig.setDataSourceType(DataSourceType.SLAVE);
Db db2 = Db.use(dbConfig);
// 查询user表的所有记录
List<Entity> userList2 = db2.find("select * from user");

示例二：使用Druid连接池

除了默认的JDBC连接池方式外，hutool-db还允许我们使用第三方连接池，比如Druid。使用Druid连接池的步骤如下：

1. 在项目中引入Druid的依赖；
2. 在代码中，先创建Druid数据源；
3. 创建DbConfig对象并设置数据源类型为自定义类型；
4. 将Druid数据源设置到DbConfig中；
5. 调用Db.use(DbConfig)方法获取数据库连接。

具体的实现代码如下：

// 引入Druid依赖
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.2.6</version>
        </dependency>

// 创建Druid数据源
DataSource ds = DruidDataSourceFactory.createDataSource(dataSourceConfig);
// 创建DbConfig对象
MyDbConfig dbConfig = new MyDbConfig();
dbConfig.setDataSourceType(DataSourceType.MY_DATA_SOURCE_TYPE);
// 设置Druid数据源到DbConfig中
DbConfig.DbSetting dbSetting = new DbConfig.DbSetting();
dbSetting.setDataSource(ds);
dbSetting.setDbType(MySql.INSTANCE);
dbConfig.setCurrentDbSetting(dbSetting);
// 获取数据库连接
Db db = Db.use(dbConfig);

通过以上方式，我们可以自定义连接池类型，实现更加灵活的数据源配置。

总的来说，hutool-db是一个很方便的多数据源管理解决方案，支持多种连接池类型和数据源配置的方式。我们可以根据需要选择适合的方式进行配置。