亲手教你SpringBoot中的多数据源集成问题

public void afterPropertiesSet() {

         if ( this .targetDataSources == null ) {

             throw new IllegalArgumentException( "Property 'targetDataSources' is required" );

         }

         this .resolvedDataSources = new HashMap<Object, DataSource>( this .targetDataSources.size());

         for (Map.Entry<Object, Object> entry : this .targetDataSources.entrySet()) {

             Object lookupKey = resolveSpecifiedLookupKey(entry.getKey());

             DataSource dataSource = resolveSpecifiedDataSource(entry.getValue());

             this .resolvedDataSources.put(lookupKey, dataSource);

         if ( this .defaultTargetDataSource != null ) {

             this .resolvedDefaultDataSource = resolveSpecifiedDataSource( this .defaultTargetDataSource);

     }

@Override

public Connection getConnection() throws SQLException {

         return determineTargetDataSource().getConnection();

}

protected DataSource determineTargetDataSource() {

         Assert.notNull( this .resolvedDataSources, "DataSource router not initialized" );

         Object lookupKey = determineCurrentLookupKey();

         DataSource dataSource = this .resolvedDataSources.get(lookupKey);

         if (dataSource == null && ( this .lenientFallback || lookupKey == null )) {

             dataSource = this .resolvedDefaultDataSource;

         if (dataSource == null ) {

             throw new IllegalStateException( "Cannot determine target DataSource for lookup key [" + lookupKey + "]" );

         return dataSource;

// 创建DynamicDataSource

         GenericBeanDefinition beanDefinition = new GenericBeanDefinition();

         beanDefinition.setBeanClass(DyncRouteDataSource. class );

         beanDefinition.setSynthetic( true );

         MutablePropertyValues mpv = beanDefinition.getPropertyValues();

         //默认数据源

         mpv.addPropertyValue( "defaultTargetDataSource" , defaultDataSource);

         //其他数据源

         mpv.addPropertyValue( "targetDataSources" , targetDataSources);

         //注册到spring bean容器中

         registry.registerBeanDefinition( "dataSource" , beanDefinition);

@SpringBootApplication (exclude = { DataSourceAutoConfiguration. class , DataSourceTransactionManagerAutoConfiguration. class , MybatisAutoConfiguration. class })

@Import (DyncDataSourceRegister. class )

@ServletComponentScan

@ComponentScan (basePackages = { "com.happy.springboot.api" , "com.happy.springboot.service" })

public class Application {

     public static void main(String[] args) {

         SpringApplication.run(Application. class , args);

     }

}

import java.lang.annotation.Documented;

import java.lang.annotation.ElementType;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;

@Target ({ ElementType.TYPE, ElementType.METHOD })

@Retention (RetentionPolicy.RUNTIME)

@Documented

public @interface TargetDataSource {

     String value();

     // 是否分库

     boolean isSharding() default false ;

     // 获取分库策略

     String strategy() default "" ;

}

// 动态数据源上下文

public class DyncDataSourceContextHolder {

     private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

     public static List<String> dataSourceNames = new ArrayList<String>();

     public static void setDataSource(String dataSourceName) {

         contextHolder.set(dataSourceName);

     }

     public static String getDataSource() {

         return contextHolder.get();

     public static void clearDataSource() {

         contextHolder.remove();

     public static boolean containsDataSource(String dataSourceName) {

         return dataSourceNames.contains(dataSourceName);

/**

  *

  * @author jasoHsu

  * 动态数据源在切换，将数据源设置到ThreadLocal对象中

  */

@Component

@Order (- 10 )

@Aspect

public class DyncDataSourceInterceptor {

     @Before ( "@annotation(targetDataSource)" )

     public void changeDataSource(JoinPoint point, TargetDataSource targetDataSource) throws Exception {

         String dbIndx = null ;

         String targetDataSourceName = targetDataSource.value() + (dbIndx == null ? "" : dbIndx);

         if (DyncDataSourceContextHolder.containsDataSource(targetDataSourceName)) {

             DyncDataSourceContextHolder.setDataSource(targetDataSourceName);

         }

     @After ( "@annotation(targetDataSource)" )

     public void resetDataSource(JoinPoint point, TargetDataSource targetDataSource) {

         DyncDataSourceContextHolder.clearDataSource();

//

public class DyncRouteDataSource extends AbstractRoutingDataSource {

@Override

     protected Object determineCurrentLookupKey() {

         return DyncDataSourceContextHolder.getDataSource();

     public DataSource findTargetDataSource() {

         return this .determineTargetDataSource();

@Resource

     DyncRouteDataSource dynamicRouteDataSource;

     @Bean (name = "sqlSessionFactory" )

     public SqlSessionFactory getinvestListingSqlSessionFactory() throws Exception {

         String configLocation = "classpath:/conf/mybatis/configuration.xml" ;

         String mapperLocation = "classpath*:/com/happy/springboot/service/dao/*/*Mapper.xml" ;

         SqlSessionFactory sqlSessionFactory = createDefaultSqlSessionFactory(dynamicRouteDataSource, configLocation,

                 mapperLocation);

         return sqlSessionFactory;

     }

     @Bean (name = "txManager" )

     public PlatformTransactionManager txManager() {

         return new DataSourceTransactionManager(dynamicRouteDataSource);

spring:

   dataSource:

     happyboot:

       driverClassName: com.mysql.jdbc.Driver

       url: jdbc:mysql: //localhost:3306/happy_springboot?characterEncoding=utf8&useSSL=true

       username: root

       password: admin

     extdsnames: happyboot01,happyboot02

     happyboot01:

       driverClassName: com.mysql.jdbc.Driver

       url: jdbc:mysql: //localhost:3306/happyboot01?characterEncoding=utf8&useSSL=true

       username: root

       password: admin

     happyboot02:

       driverClassName: com.mysql.jdbc.Driver

       url: jdbc:mysql: //localhost:3306/happyboot02?characterEncoding=utf8&useSSL=true

       username: root

       password: admin

@Service

public class UserExtServiceImpl implements IUserExtService {

     @Autowired

     private UserInfoMapper userInfoMapper;

     @TargetDataSource (value= "happyboot01" )

     @Override

     public UserInfo getUserBy(Long id) {

         return userInfoMapper.selectByPrimaryKey(id);

     }

}