正文

轻松掌握MyBatis动态数据源，多环境数据库切换不求人

/0 浏览量
0414

在Java开发中，尤其是在使用Spring框架进行项目开发时，数据库的配置和管理是一个重要的环节。MyBatis作为一款优秀的持久层框架，支持动态数据源配置，使得在不同环境中切换数据库变得轻松简单。本文将带你一步步了解如何在MyBatis中配置动态数据源，实现多环境数据库切换。

什么是动态数据源？

动态数据源（Dynamic DataSource）指的是在运行时动态切换数据源的技术。在Java项目中，通常会有开发、测试和生产等多个环境，每个环境都需要配置不同的数据库。动态数据源技术可以在不重启应用的情况下，根据不同的运行环境自动切换到对应的数据源。

MyBatis动态数据源实现原理

MyBatis动态数据源的核心原理是通过在运行时动态地修改SqlSessionFactory的配置来实现数据源的切换。具体来说，就是通过创建多个DataSource实例，并在运行时根据当前环境动态地替换SqlSessionFactory中的DataSource。

配置MyBatis动态数据源

以下是一个简单的MyBatis动态数据源配置示例：

1. 添加依赖

首先，在你的项目中添加MyBatis和动态数据源的依赖。以下是Maven依赖示例：

<dependencies>
    <!-- MyBatis依赖 -->
    <dependency>
        <groupId>org.mybatis</groupId>
        <artifactId>mybatis</artifactId>
        <version>3.5.7</version>
    </dependency>
    <!-- 动态数据源依赖 -->
    <dependency>
        <groupId>com.baomidou</groupId>
        <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
        <version>3.3.6</version>
    </dependency>
</dependencies>

2. 配置文件

在application.properties或application.yml中配置不同环境的数据库信息：

# 开发环境
spring.datasource.dynamic.primary=dev
spring.datasource.dynamic.datasource.dev.url=jdbc:mysql://localhost:3306/dev_db?useUnicode=true&characterEncoding=utf-8
spring.datasource.dynamic.datasource.dev.username=root
spring.datasource.dynamic.datasource.dev.password=root

# 测试环境
spring.datasource.dynamic.primary=test
spring.datasource.dynamic.datasource.test.url=jdbc:mysql://localhost:3306/test_db?useUnicode=true&characterEncoding=utf-8
spring.datasource.dynamic.datasource.test.username=root
spring.datasource.dynamic.datasource.test.password=root

# 生产环境
spring.datasource.dynamic.primary=prod
spring.datasource.dynamic.datasource.prod.url=jdbc:mysql://localhost:3306/prod_db?useUnicode=true&characterEncoding=utf-8
spring.datasource.dynamic.datasource.prod.username=root
spring.datasource.dynamic.datasource.prod.password=root

3. 创建动态数据源配置类

创建一个配置类，用于初始化动态数据源：

import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

@Configuration
public class DataSourceConfig {

    @Bean
    public DataSource dataSource() {
        DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
        Map<Object, Object> dataSourceMap = new HashMap<>();

        // 添加不同环境的数据库配置
        dataSourceMap.put("dev", devDataSource());
        dataSourceMap.put("test", testDataSource());
        dataSourceMap.put("prod", prodDataSource());

        dynamicRoutingDataSource.setTargetDataSources(dataSourceMap);
        dynamicRoutingDataSource.setDefaultTargetDataSource(devDataSource());

        return dynamicRoutingDataSource;
    }

    // 创建不同环境的数据库连接
    private DataSource devDataSource() {
        // ...
    }

    private DataSource testDataSource() {
        // ...
    }

    private DataSource prodDataSource() {
        // ...
    }
}

4. 使用动态数据源

在MyBatis的Mapper接口或XML映射文件中，使用@DS注解或<choose>标签来指定使用哪个数据源：

@Mapper
public interface UserMapper {
    @Select("SELECT * FROM user WHERE id = #{id}")
    @DS("dev")
    User getUserById(@Param("id") Integer id);
}

或者使用<choose>标签：

<select id="getUserById" resultType="User">
    <choose>
        <when test="dataSource == 'dev'">
            SELECT * FROM user WHERE id = #{id}
        </when>
        <when test="dataSource == 'test'">
            SELECT * FROM test_user WHERE id = #{id}
        </when>
        <otherwise>
            SELECT * FROM user WHERE id = #{id}
        </otherwise>
    </choose>
</select>

总结

通过以上步骤，你可以在MyBatis中轻松配置动态数据源，实现多环境数据库切换。在实际项目中，可以根据需要扩展和优化动态数据源配置，以满足不同的业务需求。希望本文能帮助你更好地掌握MyBatis动态数据源技术。

-- 展开阅读全文 --