天天看点

mybatis 多数据源_Spring Boot + Mybatis多数据源和动态数据源配置

应用场景

项目需要同时连接两个不同的数据库A, B,并且它们都为主从架构,一台写库,多台读库,

多数据源。

mybatis 多数据源_Spring Boot + Mybatis多数据源和动态数据源配置

首先要将spring boot自带的DataSourceAutoConfiguration禁掉,因为它会读取application.properties文件的spring.datasource.*属性并自动配置单数据源。在@SpringBootApplication注解中添加exclude属性即可:

@SpringBootApplication(exclude = {

DataSourceAutoConfiguration.class

})

public class TitanWebApplication {

public static void main(String[] args) {

SpringApplication.run(TitanWebApplication.class, args);

}

}

然后在application.properties中配置多数据源连接信息:

# titan库

spring.datasource.titan-master.url=jdbc:mysql://X.X.X.X:port/titan?characterEncoding=UTF-8

spring.datasource.titan-master.username=

spring.datasource.titan-master.password=

spring.datasource.titan-master.driver-class-name=com.mysql.jdbc.Driver

# 连接池配置

# 省略

# 其它库

spring.datasource.db2.url=jdbc:mysql://X.X.X.X:port/titan2?characterEncoding=UTF-8

spring.datasource.db2.username=

spring.datasource.db2.password=

spring.datasource.db2.driver-class-name=com.mysql.jdbc.Driver

由于我们禁掉了自动数据源配置,因些下一步就需要手动将这些数据源创建出来:

@Configuration

public class DataSourceConfig {

@Bean(name = "titanMasterDS")

@ConfigurationProperties(prefix = "spring.datasource.titan-master") // application.properteis中对应属性的前缀

public DataSource dataSource1() {

return DataSourceBuilder.create().build();

}

@Bean(name = "ds2")

@ConfigurationProperties(prefix = "spring.datasource.db2") // application.properteis中对应属性的前缀

public DataSource dataSource2() {

return DataSourceBuilder.create().build();

}

}

接下来需要配置两个mybatis的SqlSessionFactory分别使用不同的数据源:

@Configuration

@MapperScan(basePackages = {"titan.mapper"}, sqlSessionFactoryRef = "sqlSessionFactory1")

public class MybatisDbAConfig {

@Autowired

@Qualifier("titanMasterDS")

private DataSource ds1;

@Bean

public SqlSessionFactory sqlSessionFactory1() throws Exception {

SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();

factoryBean.setDataSource(ds1); // 使用titan数据源, 连接titan库

return factoryBean.getObject();

}

@Bean

public SqlSessionTemplate sqlSessionTemplate1() throws Exception {

SqlSessionTemplate template = new SqlSessionTemplate(sqlSessionFactory1()); // 使用上面配置的Factory

return template;

}

}

经过上面的配置后,titan.mapper下的Mapper接口,都会使用titan数据源。同理可配第二个SqlSessionFactory:

@Configuration

@MapperScan(basePackages = {"other.mapper"}, sqlSessionFactoryRef = "sqlSessionFactory2")

public class MybatisDbBConfig {

@Autowired

@Qualifier("ds2")

private DataSource ds2;

@Bean

public SqlSessionFactory sqlSessionFactory2() throws Exception {

SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();

factoryBean.setDataSource(ds2);

return factoryBean.getObject();

}

@Bean

public SqlSessionTemplate sqlSessionTemplate2() throws Exception {

SqlSessionTemplate template = new SqlSessionTemplate(sqlSessionFactory2());

return template;

}

}

完成这些配置后,假设有2个Mapper titan.mapper.UserMapper和other.mapper.RoleMapper,使用前者时会自动连接titan库,后者连接ds2库。

动态数据源

使用动态数据源的初衷,是能在应用层做到读写分离,即在程序代码中控制不同的查询方法去连接不同的库。除了这种方法以外,数据库中间件也是个不错的选择,它的优点是数据库集群对应用来说只暴露为单库,不需要切换数据源的代码逻辑。

我们通过自定义注解 + AOP的方式实现数据源动态切换。

首先定义一个ContextHolder, 用于保存当前线程使用的数据源名:

public class DataSourceContextHolder {

public static final Logger log = LoggerFactory.getLogger(DataSourceContextHolder.class);

public static final String DEFAULT_DS = "titan-master";

private static final ThreadLocal contextHolder = new ThreadLocal<>();

// 设置数据源名

public static void setDB(String dbType) {

log.debug("切换到{}数据源

继续阅读