package com.thhy.general.config.mysql;
|
|
import com.thhy.general.common.enums.DataSources;
|
import com.zaxxer.hikari.HikariDataSource;
|
import org.apache.ibatis.logging.slf4j.Slf4jImpl;
|
import org.apache.ibatis.logging.stdout.StdOutImpl;
|
import org.apache.ibatis.plugin.Interceptor;
|
import org.apache.ibatis.session.SqlSessionFactory;
|
import org.apache.ibatis.transaction.TransactionFactory;
|
import org.mybatis.spring.SqlSessionFactoryBean;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Qualifier;
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
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 org.springframework.transaction.TransactionManager;
|
import org.springframework.transaction.annotation.EnableTransactionManagement;
|
|
import javax.sql.DataSource;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
|
@Configuration
|
@ConditionalOnProperty(prefix = "spring.datasource.hikari",value = "minIdle",havingValue = "10")
|
@EnableTransactionManagement
|
public class MysqlConfig {
|
|
@Autowired
|
private MysqlPrimary mysqlPrimary;
|
|
@Autowired
|
private MysqlSecondary mysqlSecondary;
|
|
@Autowired
|
private HikariConfig hikariConfig;
|
|
@Autowired
|
private SwtichInterceptor swtichInterceptor;
|
|
@Bean
|
@Primary
|
public DataSource dataSourcePrimary() {
|
HikariDataSource hikariDataSource = new HikariDataSource();
|
hikariDataSource.setMaxLifetime(hikariConfig.getMaxLifeTime());
|
hikariDataSource.setMaximumPoolSize(hikariConfig.getMaxPoolSize());
|
hikariDataSource.setMinimumIdle(hikariConfig.getMinIdle());
|
hikariDataSource.setIdleTimeout(hikariConfig.getIdleTimeOut());
|
DataSource dataSource = DataSourceBuilder.create().url(mysqlPrimary.getUrl())
|
.driverClassName(mysqlPrimary.getDriverClassName())
|
.username(mysqlPrimary.getUsername())
|
.password(mysqlPrimary.getPassword()).build();
|
hikariDataSource.setDataSource(dataSource);
|
return hikariDataSource;
|
}
|
|
@Bean
|
public DataSource dataSourceSecondary() {
|
HikariDataSource hikariDataSource = new HikariDataSource();
|
hikariDataSource.setMaxLifetime(hikariConfig.getMaxLifeTime());
|
hikariDataSource.setMaximumPoolSize(hikariConfig.getMaxPoolSize());
|
hikariDataSource.setMinimumIdle(hikariConfig.getMinIdle());
|
hikariDataSource.setIdleTimeout(hikariConfig.getIdleTimeOut());
|
DataSource dataSource = DataSourceBuilder.create().url(mysqlSecondary.getUrl())
|
.driverClassName(mysqlSecondary.getDriverClassName())
|
.username(mysqlSecondary.getUsername())
|
.password(mysqlSecondary.getPassword()).build();
|
hikariDataSource.setDataSource(dataSource);
|
return hikariDataSource;
|
}
|
|
|
|
@Bean
|
public DynamicDataSource proxyDataSource(@Qualifier("dataSourcePrimary") DataSource dataSourcePrimary,
|
@Qualifier("dataSourceSecondary") DataSource dataSourceSecondary) {
|
Map<Object, Object> mappedDataSource = new HashMap<>();
|
mappedDataSource.put(DataSources.MASTER.name(), dataSourcePrimary);
|
mappedDataSource.put(DataSources.SLAVE.name(), dataSourceSecondary);
|
DynamicDataSource proxy = new DynamicDataSource();
|
proxy.setDefaultTargetDataSource(dataSourcePrimary);
|
proxy.setTargetDataSources(mappedDataSource);
|
proxy.afterPropertiesSet();
|
return proxy;
|
}
|
|
/*@Bean
|
public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception{
|
SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
|
sqlSessionFactoryBean.setDataSource(dataSource);
|
sqlSessionFactoryBean.setTypeAliasesPackage("com.thhy.*.modules.*.*.mapper");
|
sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:mapping/*.xml"));
|
SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBean.getObject();
|
|
return sqlSessionFactory;
|
}*/
|
|
@Bean
|
public SqlSessionFactory SqlSessionFactory(DynamicDataSource dynamicDataSource)
|
throws Exception {
|
final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
|
sessionFactory.setPlugins(new Interceptor[]{swtichInterceptor});
|
sessionFactory.setDataSource(dynamicDataSource);
|
//sessionFactory.setTransactionFactory();
|
org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
|
configuration.setLogImpl(StdOutImpl.class);
|
configuration.setMapUnderscoreToCamelCase(true);
|
sessionFactory.setConfiguration(configuration);
|
|
sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver()
|
.getResources("classpath*:mapping/*.xml"));
|
return sessionFactory.getObject();
|
}
|
|
@Bean
|
public TransactionManager TransactionManager(DynamicDataSource dynamicDataSource) {
|
return new DataSourceTransactionManager(dynamicDataSource);
|
}
|
|
/*@Bean(name = "priSqlSessionFactory")
|
@Primary
|
public SqlSessionFactory masterSqlSessionFactory(@Qualifier("dataSourcePrimary") DataSource dataSourcePrimary)
|
throws Exception {
|
final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
|
sessionFactory.setPlugins(new Interceptor[]{swtichInterceptor});
|
sessionFactory.setDataSource(dataSourcePrimary);
|
sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver()
|
.getResources("classpath*:mapping/*.xml"));
|
return sessionFactory.getObject();
|
}
|
|
|
@Bean(name = "secSqlSessionFactory")
|
public SqlSessionFactory clusterSqlSessionFactory(@Qualifier("dataSourceSecondary") DataSource dataSourceSecondary)
|
throws Exception {
|
final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
|
sessionFactory.setPlugins(new Interceptor[]{swtichInterceptor});
|
sessionFactory.setDataSource(dataSourceSecondary);
|
|
sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver()
|
.getResources("classpath*:mapping/*.xml"));
|
return sessionFactory.getObject();
|
}*/
|
|
|
}
|
