Spring Boot + MyBatis + Druid環境下配置多資料來源
阿新 • • 發佈：2018-12-09
專案中採用Spring Boot + MyBatis + Druid的架構，在原資料來源的基礎上需要新增一個新的資料來源。
除錯期間，發現添加了SqlSessionFactoryBean後，原資料來源有一部分欄位無法取值，後發現是application.yml中的配置失效，所以需要在SqlSessionFactoryBean中新增所需配置。
1. 在application.yml下配置兩套資料來源資訊
預設資料來源：
spring: datasource: type: com.alibaba.druid.pool.DruidDataSource driverClassName: com.mysql.jdbc.Driver url: jdbc:mysql://192.168.1.10:3306/database1?useUnicode=true&characterEncoding=utf8 username: root password: **** initialSize: 1 minIdle: 3 maxActive: 20 # 配置獲取連線等待超時的時間 maxWait: 60000 # 配置間隔多久才進行一次檢測，檢測需要關閉的空閒連線，單位是毫秒 timeBetweenEvictionRunsMillis: 60000 # 配置一個連線在池中最小生存的時間，單位是毫秒 minEvictableIdleTimeMillis: 30000 validationQuery: select 'x' testWhileIdle: true testOnBorrow: false testOnReturn: false
新增資料來源
hana: datasource: type: com.alibaba.druid.pool.DruidDataSource driverClassName: com.mysql.jdbc.Driver url: jdbc:mysql://192.168.6.12:3306/hana?useUnicode=true&characterEncoding=utf8 username: root password: 1234 initialSize: 1 minIdle: 3 maxActive: 20 # 配置獲取連線等待超時的時間 maxWait: 60000 # 配置間隔多久才進行一次檢測，檢測需要關閉的空閒連線，單位是毫秒 timeBetweenEvictionRunsMillis: 60000 # 配置一個連線在池中最小生存的時間，單位是毫秒 minEvictableIdleTimeMillis: 30000 validationQuery: select 'x' testWhileIdle: true testOnBorrow: false testOnReturn: false
2. 針對兩個資料來源分別寫配置類DruidDBConfig.java
要點：
a. basePackages 指定該資料來源對應的MyBatis的Mapper類所在的包，表示指定包名下的資料庫操作使用該資料來源；
@MapperScan(basePackages = {"com.simba.*.dao"}, sqlSessionFactoryRef = "masterSqlSessionFactory ")
b. 定義一個masterSqlSessionFactory bean，並在上述MapperScan中指定
如下程式碼指定了MyBatis的XML Mapping檔案的路徑和MapUnderscoreToCamelCase屬性
Resource[] resource = new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/**/*Mapper.xml");
sqlSessionFactoryBean.setMapperLocations(resource);
org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
configuration.setMapUnderscoreToCamelCase(true);它替代了原來application.yml中的配置，注意定了sqlSessionFactory bean 後，如下配置將不再生效，因此相關配置需在sqlSessionFactory 中完成。
#[棄用]因為多資料來源需要，MyBatis配置棄用，具體配置見DruidDBConfig
#mybatis:
# configuration:
# map-underscore-to-camel-case: true
# mapper-locations: mybatis/**/*Mapper.xml
# typeAliasesPackage: com.simba.**.domain預設資料來源的配置Bean
@Configuration
@MapperScan(basePackages = {"com.simba.*.dao"}, sqlSessionFactoryRef = "masterSqlSessionFactory")//指定該資料來源相應的MyBatis的Mapper類所在的包
public class DruidDBConfig {
private Logger logger = LoggerFactory.getLogger(DruidDBConfig.class);
@Value("${spring.datasource.url}")
private String dbUrl;
@Value("${spring.datasource.username}")
private String username;
@Value("${spring.datasource.password}")
private String password;
@Value("${spring.datasource.driverClassName}")
private String driverClassName;
@Value("${spring.datasource.initialSize}")
private int initialSize;
@Value("${spring.datasource.minIdle}")
private int minIdle;
@Value("${spring.datasource.maxActive}")
private int maxActive;
@Value("${spring.datasource.maxWait}")
private int maxWait;
@Value("${spring.datasource.timeBetweenEvictionRunsMillis}")
private int timeBetweenEvictionRunsMillis;
@Value("${spring.datasource.minEvictableIdleTimeMillis}")
private int minEvictableIdleTimeMillis;
@Value("${spring.datasource.validationQuery}")
private String validationQuery;
@Value("${spring.datasource.testWhileIdle}")
private boolean testWhileIdle;
@Value("${spring.datasource.testOnBorrow}")
private boolean testOnBorrow;
@Value("${spring.datasource.testOnReturn}")
private boolean testOnReturn;
@Bean(initMethod = "init", destroyMethod = "close") //宣告其為Bean例項
@Primary //在同樣的DataSource中，首先使用被標註的DataSource
public DataSource dataSource() {
DruidDataSource datasource = new DruidDataSource();
datasource.setUrl(this.dbUrl);
datasource.setUsername(username);
datasource.setPassword(password);
datasource.setDriverClassName(driverClassName);
//configuration
datasource.setInitialSize(initialSize);
datasource.setMinIdle(minIdle);
datasource.setMaxActive(maxActive);
datasource.setMaxWait(maxWait);
datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
datasource.setValidationQuery(validationQuery);
datasource.setTestWhileIdle(testWhileIdle);
datasource.setTestOnBorrow(testOnBorrow);
datasource.setTestOnReturn(testOnReturn);
return datasource;
}
//建立Session
@Bean(name="masterSqlSessionFactory")
@Primary
public SqlSessionFactory masterSqlSessionFactory(@Qualifier("dataSource") DataSource dataSource) throws Exception{
final SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
sqlSessionFactoryBean.setDataSource(dataSource);
Resource[] resource = new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/**/*Mapper.xml");//MyBatis XML檔案所在路徑
sqlSessionFactoryBean.setMapperLocations(resource);
org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
configuration.setMapUnderscoreToCamelCase(true);
sqlSessionFactoryBean.setConfiguration(configuration);
return sqlSessionFactoryBean.getObject();
}
}新增資料來源的配置bean
@Configuration
@MapperScan(basePackages = "com.hana.dao", sqlSessionFactoryRef = "hanaSqlSessionFactory")
public class HanaDruidDBConfig {
private Logger logger = LoggerFactory.getLogger(HanaDruidDBConfig.class);
@Value("${hana.datasource.url}")
private String dbUrl;
@Value("${hana.datasource.username}")
private String username;
@Value("${hana.datasource.password}")
private String password;
@Value("${hana.datasource.driverClassName}")
private String driverClassName;
@Value("${hana.datasource.initialSize}")
private int initialSize;
@Value("${hana.datasource.minIdle}")
private int minIdle;
@Value("${hana.datasource.maxActive}")
private int maxActive;
@Value("${hana.datasource.maxWait}")
private int maxWait;
@Value("${hana.datasource.timeBetweenEvictionRunsMillis}")
private int timeBetweenEvictionRunsMillis;
@Value("${hana.datasource.minEvictableIdleTimeMillis}")
private int minEvictableIdleTimeMillis;
@Value("${hana.datasource.validationQuery}")
private String validationQuery;
@Value("${hana.datasource.testWhileIdle}")
private boolean testWhileIdle;
@Value("${hana.datasource.testOnBorrow}")
private boolean testOnBorrow;
@Value("${hana.datasource.testOnReturn}")
private boolean testOnReturn;
@Bean(initMethod = "init", destroyMethod = "close") //宣告其為Bean例項
public DataSource hanaDataSource() {
DruidDataSource datasource = new DruidDataSource();
datasource.setUrl(this.dbUrl);
datasource.setUsername(username);
datasource.setPassword(password);
datasource.setDriverClassName(driverClassName);
//configuration
datasource.setInitialSize(initialSize);
datasource.setMinIdle(minIdle);
datasource.setMaxActive(maxActive);
datasource.setMaxWait(maxWait);
datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
datasource.setValidationQuery(validationQuery);
datasource.setTestWhileIdle(testWhileIdle);
datasource.setTestOnBorrow(testOnBorrow);
datasource.setTestOnReturn(testOnReturn);
return datasource;
}
//建立Session
@Bean(name="hanaSqlSessionFactory")
public SqlSessionFactory hanaSqlSessionFactory(@Qualifier("hanaDataSource") DataSource hanaDataSource) throws Exception{
final SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
sqlSessionFactoryBean.setDataSource(hanaDataSource);
/*Resource[] resource = new PathMatchingResourcePatternResolver().getResources(ClusterConfig.MAPPER_LOCATION);
sqlSessionFactoryBean.setMapperLocations(resource);*/
return sqlSessionFactoryBean.getObject();
}
}這樣就完成了配置，根據MapperScan中所指定的包名，自動選擇相應資料來源進行資料操作。