欢迎访问宙启技术站
智能推送

springboot配置多数据源框架的示例分析

发布时间:2023-05-15 18:15:01

SpringBoot是一款快速开发的框架,在数据访问时也得到了很好的应用。不过,在同时使用多个数据源时,我们需要进行特殊的配置,在这里,我们将介绍使用SpringBoot配置多数据源的方法。

首先,我们需要在SpringBoot中使用多个数据源,可以通过将多个数据源的配置文件分别存储在application.properties中,配置不同的前缀名称来实现:

spring.datasource.primary.url=jdbc:mysql://localhost:3306/test
spring.datasource.primary.username=root
spring.datasource.primary.password=root

spring.datasource.secondary.url=jdbc:mysql://localhost:3306/test2
spring.datasource.secondary.username=root
spring.datasource.secondary.password=root

这里配置了两个数据源,一个是primary数据源,一个是secondary数据源。同时,我们需要创建两个数据源的连接池,分别为primary和secondary。

在SpringBoot提供的2.0.0之前的版本中,使用Druid连接池比较方便,我们可以通过使用Druid的自动配置来实现:

@Configuration
public class DruidDataSourceConfig {

    @Bean(name = "primaryDataSource")
    @Qualifier("primaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DruidDataSourceBuilder.create().build();
    }

    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DruidDataSourceBuilder.create().build();
    }
}

这里我们使用了@ConfigurationProperties自动将配置文件中的spring.datasource.primary和spring.datasource.secondary转化成对应的数据源类对象。

在SpringBoot 2.0.0之后的版本中,我们可以使用HikariCP连接池。我们可以通过以下方式来进行设置:

spring.datasource.primary.hikari.jdbc-url=jdbc:mysql://localhost:3306/test
spring.datasource.primary.hikari.username=root
spring.datasource.primary.hikari.password=root

spring.datasource.secondary.hikari.jdbc-url=jdbc:mysql://localhost:3306/test2
spring.datasource.secondary.hikari.username=root
spring.datasource.secondary.hikari.password=root

@Configuration
public class HikariDataSourceConfig {

    @Bean(name = "primaryDataSource")
    @Qualifier("primaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.primary.hikari")
    public HikariDataSource primaryDataSource() {
        return (HikariDataSource) DataSourceBuilder.create().type(HikariDataSource.class).build();
    }

    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.secondary.hikari")
    public HikariDataSource secondaryDataSource() {
        return (HikariDataSource) DataSourceBuilder.create().type(HikariDataSource.class).build();
    }
}

这里我们设置了HikariCP连接池的相关属性,也使用@ConfigurationProperties来转化为类对象。

至此,我们已经完成了两个数据源的配置。下面我们需要使用多数据源的功能,在这里我们使用Mybatis来进行相关操作。我们可以使用Mybatis提供的SqlSessionFactory,来为每个数据源创建单独的SqlSessionFactory实例。

@Configuration
@MapperScan(basePackages = "com.example.mapper")
public class MybatisConfig {

    @Bean(name = "primarySqlSessionFactory")
    public SqlSessionFactory primarySqlSessionFactory(@Qualifier("primaryDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        return bean.getObject();
    }

    @Bean(name = "secondarySqlSessionFactory")
    public SqlSessionFactory secondarySqlSessionFactory(@Qualifier("secondaryDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        return bean.getObject();
    }

    @Bean(name = "primarySqlSessionTemplate")
    public SqlSessionTemplate primarySqlSessionTemplate(@Qualifier("primarySqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

    @Bean(name = "secondarySqlSessionTemplate")
    public SqlSessionTemplate secondarySqlSessionTemplate(@Qualifier("secondarySqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

这里我们定义了两个SqlSessionFactory,一个对应primary数据源,一个对应secondary数据源,在应用中可以使用@Qualifier来指定使用哪个数据源。同时,我们也定义了对应的SqlSessionTemplate。

最后,我们可以使用以上的配置来进行数据操作:

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    @Qualifier("primarySqlSessionTemplate")
    private SqlSessionTemplate primaryTemplate;

    @Autowired
    @Qualifier("secondarySqlSessionTemplate")
    private SqlSessionTemplate secondaryTemplate;

    @Override
    public User getPrimaryUserById(Long userId) {
        return primaryTemplate.selectOne("com.example.mapper.UserMapper.selectByPrimaryKey", userId);
    }

    @Override
    public User getSecondaryUserById(Long userId) {
        return secondaryTemplate.selectOne("com.example.mapper.UserMapper.selectByPrimaryKey", userId);
    }
}

这里使用了@Autowired注解进行注入,使用@Qualifier注解指定了对应的SqlSessionTemplate。通过以上的配置,我们已经成功实现了SpringBoot多数据源的配置与使用。