前言
在項(xiàng)目開發(fā)中，經(jīng)常會(huì)涉及到一個(gè)應(yīng)用程序調(diào)用多個(gè)數(shù)據(jù)的情況。今天介紹一個(gè)SpringBoot+mybatis的多數(shù)據(jù)源的解決方案。
數(shù)據(jù)庫(kù)準(zhǔn)備
創(chuàng)建兩個(gè)數(shù)據(jù)庫(kù)，兩個(gè)數(shù)據(jù)庫(kù)都有Im_person表，兩個(gè)表中無(wú)數(shù)據(jù)。
代碼結(jié)構(gòu)
說(shuō)明：我這里只是為了體現(xiàn)效果，就省略了service步驟。各位大牛開發(fā)，不喜勿噴，理解萬(wàn)歲，嘻嘻！！
- application.yml中配置兩個(gè)數(shù)據(jù)源，配置如下：
master:
datasource:
type: com.alibaba.druid.pool.DruidDataSource
driver-class-name: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://127.0.0.1:3306/db1?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&serverTimezone=GMT%2B8
username: root
password: 123456
slave:
datasource:
type: com.alibaba.druid.pool.DruidDataSource
driver-class-name: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://127.0.0.1:3306/db2?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&serverTimezone=GMT%2B8
username: root
password: 123456
2.SpringBoot多數(shù)據(jù)源配置類。
@Configuration
@MapperScan(basePackages = "com.zhangls.multipledatasource.dao.master", sqlSessionFactoryRef = "masterSqlSessionFactory")
public class MasterDataSourceConfig {
@Value("${master.datasource.driver-class-name}")
private String driverClassName;
@Value("${master.datasource.url}")
private String url;
@Value("${master.datasource.username}")
private String username;
@Value("${master.datasource.password}")
private String password;
@Bean(name = "masterDataSource")
@Primary
public DataSource dataSource() {
DruidDataSource dataSource = new DruidDataSource();
dataSource.setDriverClassName(this.driverClassName);
dataSource.setUrl(this.url);
dataSource.setUsername(this.username);
dataSource.setPassword(this.password);
return dataSource;
}
@Bean(name = "masterSqlSessionFactory")
@Primary
public SqlSessionFactory sqlSessionFactory(@Qualifier("masterDataSource") DataSource dataSource) throws Exception {
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(dataSource);
ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
List<String> mapperLocations = new ArrayList<>();
mapperLocations.add("classpath*:/mapper/master/*.xml");
List<Resource> resources = new ArrayList();
if (mapperLocations != null) {
for (String mapperLocation : mapperLocations) {
try {
Resource[] mappers = resourceResolver.getResources(mapperLocation);
resources.addAll(Arrays.asList(mappers));
} catch (IOException e) {
// ignore
}
}
}
bean.setMapperLocations(resources.toArray(new Resource[resources.size()]));
return bean.getObject();
}
@Bean(name = "masterTransactionManager")
@Primary
public DataSourceTransactionManager transactionManager(@Qualifier("masterDataSource") DataSource dataSource) {
return new DataSourceTransactionManager(dataSource);
}
@Bean(name = "masterSqlSessionTemplate")
@Primary
public SqlSessionTemplate testSqlSessionTemplate(
@Qualifier("masterSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
return new SqlSessionTemplate(sqlSessionFactory);
}
}
@Configuration
@MapperScan(basePackages = "com.zhangls.multipledatasource.dao.slave", sqlSessionFactoryRef = "slaveSqlSessionFactory")
public class SlaveDataSourceConfig {
@Value("${slave.datasource.driver-class-name}")
private String driverClassName;
@Value("${slave.datasource.url}")
private String url;
@Value("${slave.datasource.username}")
private String username;
@Value("${slave.datasource.password}")
private String password;
@Bean(name = "slaveDataSource")
public DataSource dataSource() {
DruidDataSource dataSource = new DruidDataSource();
dataSource.setDriverClassName(this.driverClassName);
dataSource.setUrl(this.url);
dataSource.setUsername(this.username);
dataSource.setPassword(this.password);
return dataSource;
}
@Bean(name = "slaveSqlSessionFactory")
public SqlSessionFactory sqlSessionFactory(@Qualifier("slaveDataSource") DataSource dataSource) throws Exception {
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(dataSource);
bean.setMapperLocations(
new PathMatchingResourcePatternResolver().getResources("classpath*:/mapper/slave/*.xml"));
return bean.getObject();
}
@Bean(name = "slaveTransactionManager")
public DataSourceTransactionManager transactionManager(@Qualifier("slaveDataSource") DataSource dataSource) {
return new DataSourceTransactionManager(dataSource);
}
@Bean(name = "slaveSqlSessionTemplate")
public SqlSessionTemplate testSqlSessionTemplate(
@Qualifier("masterSqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
return new SqlSessionTemplate(sqlSessionFactory);
}
}
說(shuō)明：兩個(gè)數(shù)據(jù)的代碼基本相同，不同的是要配置好掃描包的路徑，以及mybatis的SQL文件的路徑。所以不同的連接DAO和mapper文件要放在不同的文件夾，方便配置管理。
實(shí)現(xiàn)效果
以上的配置已經(jīng)實(shí)現(xiàn)了多數(shù)據(jù)源的配置，下面實(shí)現(xiàn)一個(gè)功能。通過(guò)一個(gè)接口同時(shí)向兩個(gè)庫(kù)的兩個(gè)表中導(dǎo)入不同的數(shù)據(jù)。
controller代碼如下：
@RestController
public class DataSourceController {
@Resource
private MasterPersonMapper masterPersonMapper;
@Resource
private SlavePersonMapper slavePersonMapper;
@GetMapping("/datasource")
public String datasource() {
ImPerson person1 = new ImPerson();
person1.setPersonId("1");
person1.setGender("男");
person1.setBirthday(new Date());
person1.setLocation("中國(guó)");
masterPersonMapper.insertSelective(person1);
ImPerson person2 = new ImPerson();
person2.setPersonId("2");
person2.setGender("女");
person2.setBirthday(new Date());
person2.setLocation("中國(guó)北京");
slavePersonMapper.insertSelective(person2);
return "導(dǎo)入成功";
}
}
執(zhí)行：
結(jié)果：
可以看到兩個(gè)庫(kù)中的兩個(gè)表都導(dǎo)入了數(shù)據(jù)。
