在entity、service、dao和controller四个层次的同级建立一个config文件夹,之后分别建立两个数据库的配置类,如图所示
其中一个数据库配置类:
import com.alibaba.druid.pool.DruidDataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import javax.sql.DataSource;
@Configuration
@MapperScan(basePackages = "com.demo.tool.mapper.XXX1", sqlSessionFactoryRef = "XXX1SqlSessionFactory")
public class XXX1DBConfig {
@Value("${spring.datasource.XXX1.driver-class-name}")
private String driverClassName;
@Value("${spring.datasource.XXX1.url}")
private String url;
@Value("${spring.datasource.XXX1.username}")
private String username;
@Value("${spring.datasource.XXX1.password}")
private String password;
@Bean(name = "XXX1")
@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 = "XXX1SqlSessionFactory")
@Primary
public SqlSessionFactory sqlSessionFactory(@Qualifier("XXX1") DataSource dataSource) throws Exception {
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(dataSource);
bean.setMapperLocations(
new PathMatchingResourcePatternResolver().getResources("classpath*:/mapper/XXX1/*Mapper.xml"));
return bean.getObject();
}
@Bean(name = "XXX1TransactionManager")
@Primary
public DataSourceTransactionManager transactionManager(@Qualifier("XXX1") DataSource dataSource){
return new DataSourceTransactionManager(dataSource);
}
@Bean(name = "XXX1SqlSessionTemplate")
@Primary
public SqlSessionTemplate testSqlSessionTemplate(
@Qualifier("XXX1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
return new SqlSessionTemplate(sqlSessionFactory);
}
}
最后在Service层使用@DS(“XXX1”)方法来切换数据库,具体方法在网上查查就有了。
另外,如果切换失败的话,在方法上加上
@Transactional(transactionManager = "XXX1TransactionManager", propagation = Propagation.REQUIRES_NEW)
就可以强制切换事务。