72 lines
3.3 KiB
Java
72 lines
3.3 KiB
Java
package com.zhgd.config;
|
|
|
|
import com.baomidou.mybatisplus.annotation.DbType;
|
|
import com.baomidou.mybatisplus.core.MybatisConfiguration;
|
|
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
|
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
|
|
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
|
|
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
|
|
import com.zhgd.xmgl.tenant.TenantHandler;
|
|
import org.apache.ibatis.session.SqlSessionFactory;
|
|
import org.mybatis.spring.SqlSessionTemplate;
|
|
import org.mybatis.spring.annotation.MapperScan;
|
|
import org.springframework.beans.factory.annotation.Qualifier;
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
|
import org.springframework.boot.jdbc.DataSourceBuilder;
|
|
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.wflow.mapper", sqlSessionTemplateRef = "sqlSessionTemplate2")
|
|
public class DataSourceTwoConfig {
|
|
|
|
@Primary
|
|
@Bean(name = "dataSource2")
|
|
@ConfigurationProperties(prefix = "spring.datasource.db2")
|
|
public DataSource dataSource() {
|
|
return DataSourceBuilder.create().build();
|
|
}
|
|
|
|
@Primary
|
|
@Bean(name = "sqlSessionFactory2")
|
|
public SqlSessionFactory db1SqlSessionFactory(@Qualifier("dataSource2") DataSource dataSource) throws Exception {
|
|
MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
|
|
bean.setDataSource(dataSource);
|
|
//开启驼峰
|
|
MybatisConfiguration configuration = new MybatisConfiguration();
|
|
configuration.setMapUnderscoreToCamelCase(true);
|
|
configuration.setLogImpl(StdOutImplCustom.class);
|
|
bean.setConfiguration(configuration);
|
|
bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:com/wflow/**/*.xml"));
|
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
|
TenantLineInnerInterceptor tenantLineInnerInterceptor = new TenantLineInnerInterceptor();
|
|
tenantLineInnerInterceptor.setTenantLineHandler(tenantHandler());
|
|
interceptor.addInnerInterceptor(tenantLineInnerInterceptor);
|
|
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
|
|
bean.setPlugins(interceptor);
|
|
return bean.getObject();
|
|
}
|
|
|
|
@Bean(name = "transactionManager2")
|
|
public DataSourceTransactionManager transactionManager(@Qualifier("dataSource2") DataSource dataSource) {
|
|
return new DataSourceTransactionManager(dataSource);
|
|
}
|
|
|
|
@Bean(name = "sqlSessionTemplate2")
|
|
public SqlSessionTemplate sqlSessionTemplate(@Qualifier("sqlSessionFactory2") SqlSessionFactory sqlSessionFactory) throws Exception {
|
|
return new SqlSessionTemplate(sqlSessionFactory);
|
|
}
|
|
|
|
@Bean
|
|
@ConditionalOnMissingBean
|
|
public TenantHandler tenantHandler() {
|
|
return new TenantHandler();
|
|
}
|
|
}
|