#database1
spring.datasource.wh.type=com.zaxxer.hikari.HikariDataSource
spring.datasource.wh.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.wh.jdbcUrl=jdbc:mysql://192.168.1.130:3306/wh?zeroDateTimeBehavior=CONVERT_TO_NULL&serverTimezone=GMT%2B8&useSSL=false&useUnicode=true&characterEncoding=UTF-8
spring.datasource.wh.username=root
spring.datasource.wh.password=root
spring.datasource.wh.minimum-idle=1
spring.datasource.wh.maximum-pool-size=15
spring.datasource.wh.auto-commit=true
spring.datasource.wh.idle-timeout=60000
spring.datasource.wh.pool-name=wuhan
spring.datasource.wh.max-lifetime=1800000
spring.datasource.wh.connection-timeout=30000
spring.datasource.wh.connection-test-query=SELECT 1
#database2
spring.datasource.es.type=com.zaxxer.hikari.HikariDataSource
spring.datasource.es.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.es.jdbcUrl=jdbc:mysql://192.168.1.130:3306/es?zeroDateTimeBehavior=CONVERT_TO_NULL&serverTimezone=GMT%2B8&useSSL=false&useUnicode=true&characterEncoding=UTF-8
spring.datasource.es.username=root
spring.datasource.es.password=root
spring.datasource.es.minimum-idle=1
spring.datasource.es.maximum-pool-size=15
spring.datasource.es.auto-commit=true
spring.datasource.es.idle-timeout=60000
spring.datasource.es.pool-name=enshi
spring.datasource.es.max-lifetime=1800000
spring.datasource.es.connection-timeout=30000
spring.datasource.es.connection-test-query=SELECT 1
import com.zaxxer.hikari.HikariDataSource;
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.boot.context.properties.ConfigurationProperties;
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.PlatformTransactionManager;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;
@MapperScan(basePackages = "xxx.xx.mapper")
@Configuration
public class DynamicDataSourceConfiguration {
@Bean
@Primary
@ConfigurationProperties(prefix = "spring.datasource.wh")
public DataSource wuhan(){
return new HikariDataSource();
}
@Bean
@ConfigurationProperties(prefix = "spring.datasource.es")
public DataSource enshi(){
return new HikariDataSource();
}
@Bean
public DataSource dynamicDataSource(){
DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();
Map<Object,Object> dataSourceMap = new HashMap<>(2);
dataSource.setDefaultTargetDataSource(wuhan());
dataSourceMap.put(CityEnum.wuhan.getCity(),wuhan());
dataSourceMap.put(CityEnum.enshi.getCity(),enshi());
dataSource.setTargetDataSources(dataSourceMap);
return dataSource;
}
@Bean
public SqlSessionFactory sqlSessionFactory() throws Exception{
SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
sqlSessionFactoryBean.setDataSource(dynamicDataSource());
sqlSessionFactoryBean.setMapperLocations( new PathMatchingResourcePatternResolver().getResources("classpath:mapping/*.xml"));
return sqlSessionFactoryBean.getObject();
}
@Bean
public SqlSessionTemplate sqlSessionTemplate() throws Exception {
return new SqlSessionTemplate(sqlSessionFactory());
}
@Bean
public PlatformTransactionManager platformTransactionManager(){
return new DataSourceTransactionManager(dynamicDataSource());
}
}
public class DataSourceHolder {
private static final ThreadLocal<String> contextHolders = new ThreadLocal<>();
public static void setDataSource(String customerType) {
contextHolders.set(customerType);
}
public static String getDataSource() {
return contextHolders.get();
}
public static void clearDataSource() {
contextHolders.remove();
}
}
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicRoutingDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceHolder.getDataSource();
}
}
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
@Configuration
public class SpringWebConfig implements WebMvcConfigurer {
@SuppressWarnings("unused")
@Autowired
private DynamicDataSourceInterceptor dynamicDataSourceInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(dynamicDataSourceInterceptor);
}
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@Component
public class DynamicDataSourceInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
String city = request.getHeader("city");
if (StringUtils.isBlank(city)) {
city = CityEnum.wuhan.getCity();
}
DataSourceHolder.setDataSource(city);
System.err.println(city + "处理开始...");
return true;
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
String city = request.getHeader("city");
if (StringUtils.isBlank(city)) {
city = CityEnum.wuhan.getCity();
}
DataSourceHolder.clearDataSource();
System.err.println(city + "处理完成...");
}
}
public enum CityEnum {
wuhan("wuhan"),
enshi("enshi");
private String city;
CityEnum(String city) {
this.city = city;
}
public String getCity() {
return city;
}
}