java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > SpringBoot动态数据源

SpringBoot中动态数据源是实现与用途

作者：vker

这篇文章主要是来和大家讨论一下SpringBoot中动态数据源是实现与用途，文中的示例代码简洁易懂，具有一定的学习价值，感兴趣的可以了解一下

一、应用场景

主从复制与读写分离通常是一起使用的。

二、实现方式

1、技术栈

2、Maven 依赖

下边依赖放入同一个pom.xml文件就可以

<properties>
    <java.version>11</java.version>
    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
    <spring-boot.version>2.6.13</spring-boot.version>
</properties>
<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-jdbc</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-aop</artifactId>
    </dependency>
    <dependency>
        <groupId>com.mysql</groupId>
        <artifactId>mysql-connector-j</artifactId>
        <scope>runtime</scope>
    </dependency>
    <dependency>
        <groupId>com.alibaba</groupId>
        <artifactId>druid-spring-boot-starter</artifactId>
        <version>1.2.16</version>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
</dependencies>
<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-dependencies</artifactId>
            <version>${spring-boot.version}</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

3、数据库结构

数据库结构非常简单，主要实现思想，复杂的都是同一个操作方式。

（1）数据库：db_dynamic1

DROP TABLE IF EXISTS `tb_user`;
CREATE TABLE `tb_user` (
  `id` bigint NOT NULL AUTO_INCREMENT,
  `username` varchar(50) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=4 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;
BEGIN;
INSERT INTO `tb_user` (`id`, `username`) VALUES (1, '张三');
INSERT INTO `tb_user` (`id`, `username`) VALUES (2, '李四');
INSERT INTO `tb_user` (`id`, `username`) VALUES (3, '王五');
COMMIT;
SET FOREIGN_KEY_CHECKS = 1;

（2）数据库：db_dynamic2

DROP TABLE IF EXISTS `tb_user`;
CREATE TABLE `tb_user` (
  `id` int NOT NULL AUTO_INCREMENT,
  `username` varchar(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=4 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;
BEGIN;
INSERT INTO `tb_user` (`id`, `username`) VALUES (1, 'Vker');
INSERT INTO `tb_user` (`id`, `username`) VALUES (2, 'Jack');
INSERT INTO `tb_user` (`id`, `username`) VALUES (3, 'Lucy');
COMMIT;
SET FOREIGN_KEY_CHECKS = 1;

4、application.yml 配置文件

将下面配置文件的url, username, password换成你自己的

spring:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    druid:
      master:
        driver-class-name: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://127.0.0.1:3306/db_dynamic1?serverTimezone=UTC&useUnicode=true&characterEncoding=UTF8&useSSL=false
        username: root
        password: root_root
      slave:
        driver-class-name: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://127.0.0.1:3306/db_dynamic2?serverTimezone=UTC&useUnicode=true&characterEncoding=UTF8&useSSL=false
        username: root
        password: root_root
      initial-size: 1
      min-idle: 1
      max-active: 20
      test-on-borrow: true

5、定义一个 DataSources 注解

这个注解主要用于多数据源的选择，默认是 master

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSources {
    String value() default "master";
}

6、配置一下动态数据源

这个类主要用于配置多数据源，不要忘记@Primary注解，否则会报错。

@Component
@Primary
public class DynamicDataSources extends AbstractRoutingDataSource {
    public static ThreadLocal<String> name = new ThreadLocal<>();
    @Resource
    DataSource master;
    @Resource
    DataSource slave;
    @Override
    protected Object determineCurrentLookupKey() {
        return name.get();
    }
    @Override
    public void afterPropertiesSet() {
        Map<Object, Object> ds = new HashMap<>();
        ds.put("master", master);
        ds.put("slave", slave);
        super.setTargetDataSources(ds);
        super.setDefaultTargetDataSource(master);
        super.afterPropertiesSet();
    }
}

7、写一个基于 DataSources 注解的动态代理

这个动态代理用于检查类或者方法上的@DataSources注解，取到注解中的value值，将值传入动态数据源配置类DynamicDataSources的ThreadLocal中，用于选择数据源。

@Component
@Aspect
public class DynamicDataSourcesAspect {
    Logger logger = LoggerFactory.getLogger(DynamicDataSourcesAspect.class);
    @Before("@annotation(dataSources)")
    public void before(DataSources dataSources) {
        String value = dataSources.value();
        DynamicDataSources.name.set(value);
        logger.info("进入AOP代理: {}", value);
    }
}

8、再来写一个读数据库配置信息的配置类

这个配置类主要将application.yml中的数据库配置信息的配置信息进行加载，然后使用数据源事务管理器DataSourceTransactionManager进行注册。

@Configuration
public class DynamicDataSourcesConfig {
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.druid.master")
    public DataSource master() {
        return DruidDataSourceBuilder.create().build();
    }
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.druid.slave")
    public DataSource slave() {
        return DruidDataSourceBuilder.create().build();
    }
    @Bean
    public DataSourceTransactionManager dataSourceTransactionManager1(@Qualifier("master") DataSource master) {
        DataSourceTransactionManager dataSourceTransactionManager =
                new DataSourceTransactionManager();
        dataSourceTransactionManager.setDataSource(master);
        return dataSourceTransactionManager;
    }
    @Bean
    public DataSourceTransactionManager dataSourceTransactionManager2(@Qualifier("slave") DataSource slave) {
        DataSourceTransactionManager dataSourceTransactionManager =
                new DataSourceTransactionManager();
        dataSourceTransactionManager.setDataSource(slave);
        return dataSourceTransactionManager;
    }
}

重点的都做完了，接下来就写一下基本的业务层代码吧，注释就不写了，我相信小伙伴们都能看懂。

业务层代码

1、实体类

public class User {
    private Long id;
    private String username;
   // 省略了 getter setter
}

2、Dao 数据层

public interface UserMapper {
    List<User> selectAllUser();
}
@Repository
public class UserMapperImpl implements UserMapper {
    @Resource
    private JdbcTemplate jdbcTemplate;
    @Override
    public List<User> selectAllUser() {
        String sql = "select * from tb_user";
        return jdbcTemplate.query(sql,
                new BeanPropertyRowMapper<>(User.class));
    }
}

3、Service 服务层

public interface UserService {
    List<User> list();
}
@Service
public class UserServiceImpl implements UserService {
    @Resource
    private UserMapper userMapper;
    @Override
    public List<User> list() {
        return userMapper.selectAllUser();
    }
}

4、Controller 控制层

这里说一下，你可以将注解 @DataSources 注解放到你想使用不同数据源的方法上，如下，默认 master 可以不写。

@RestController
public class UserController {
    @Resource
    private UserService userService;
    @GetMapping("list1")
    public List<User> list1() {
        return userService.list();
    }
    @GetMapping("list2")
    @DataSources("slave")
    public List<User> lis2t() {
        return userService.list();
    }
}

5、最后来看一下启动类

需要把 DataSourceAutoConfiguration 排除掉哦

@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
public class DynamicDatasourceDemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(DynamicDatasourceDemoApplication.class, args);
    }
}

6、测试结果

list1接口

list2接口

好了，以上就是动态数据源的实现方式了，欢迎小伙伴们留言，下期我们看看，如何实现主从复制和读写分离。

以上就是SpringBoot中动态数据源是实现与用途的详细内容，更多关于SpringBoot动态数据源的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:
阅读全文