81 lines
3.4 KiB
Java
81 lines
3.4 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.spring.MybatisSqlSessionFactoryBean;
|
|
import com.zhgd.mybatis.DataScopeHandler;
|
|
import com.zhgd.mybatis.DataScopeInterceptor;
|
|
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.Lazy;
|
|
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.zhgd.*.*.*.*.mapper", "com.zhgd.*.*.*.mapper"}, sqlSessionTemplateRef = "db1SqlSessionTemplate")
|
|
public class DataSourceOneConfig {
|
|
|
|
@Bean
|
|
@ConfigurationProperties(prefix = "spring.datasource.db1")
|
|
public DataSource db1DataSource() {
|
|
return DataSourceBuilder.create().build();
|
|
}
|
|
|
|
/**
|
|
* 这个是懒加载的注解，根据自己项目需要看是否添加
|
|
* @param dataSource
|
|
* @return
|
|
* @throws Exception
|
|
*/
|
|
@Bean
|
|
@Lazy
|
|
public SqlSessionFactory db1SqlSessionFactory(@Qualifier("db1DataSource") 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/zhgd/xmgl/**/*.xml"));
|
|
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
|
DataScopeInterceptor dataScopeInterceptor = new DataScopeInterceptor();
|
|
dataScopeInterceptor.setDataScopeHandler(dataScopeHandler());
|
|
interceptor.addInnerInterceptor(dataScopeInterceptor);
|
|
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
|
|
bean.setPlugins(interceptor);
|
|
return bean.getObject();
|
|
}
|
|
|
|
@Bean
|
|
@ConditionalOnMissingBean
|
|
public DataScopeHandler dataScopeHandler() {
|
|
return new DataScopeHandler();
|
|
}
|
|
|
|
@Bean
|
|
@Primary
|
|
public DataSourceTransactionManager db1TransactionManager(@Qualifier("db1DataSource") DataSource dataSource) {
|
|
return new DataSourceTransactionManager(dataSource);
|
|
}
|
|
|
|
@Bean
|
|
@Primary
|
|
@Lazy
|
|
public SqlSessionTemplate db1SqlSessionTemplate(@Qualifier("db1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
|
|
return new SqlSessionTemplate(sqlSessionFactory);
|
|
}
|
|
}
|