SpringBoot多数据源配置的全过程记录

dynamic:

   datasources:

     master:

       driver- class -name: com.mysql.cj.jdbc.driver

       url: jdbc:mysql: //localhost:3306/result?useunicode=true&characterencoding=utf8xxxxxxxx

       username: root

       password: 123456

     china:

       driver- class -name: com.mysql.cj.jdbc.driver

       url: jdbc:mysql: //localhost:3306/china?useunicode=true&characterencoding=utf8xxxxxxxx

       username: root

       password: 123456

     japan:

       driver- class -name: com.mysql.cj.jdbc.driver

       url: jdbc:mysql: //localhost:3306/japan?useunicode=true&characterencoding=utf8xxxxxxxx

       username: root

       password: 123456

     korea:

       driver- class -name: com.mysql.cj.jdbc.driver

       url: jdbc:mysql: //localhost:3306/korea?useunicode=true&characterencoding=utf8xxxxxxxx

       username: root

       password: 123456

package com.statistics.dynamicds.core.config;

import com.statistics.dynamicds.core.country;

import lombok.data;

import org.springframework.boot.context.properties.configurationproperties;

import org.springframework.context.annotation.configuration;

import java.util.map;

import static com.statistics.dynamicds.core.config.multidatasourceproperties.prefix;

@data

@configuration

@configurationproperties (prefix = prefix)

public class multidatasourceproperties {

   public static final string prefix = "dynamic" ;

   private map<country, datasourceproperties> datasources;

   @data

   public static class datasourceproperties {

     private string driverclassname;

     private string url;

     private string username;

     private string password;

   }

}

package com.statistics.dynamicds.core;

public enum country {

   master( "master" , 0 ),

   china( "china" , 86 ),

   japan( "japan" , 81 ),

   korea( "korea" , 82 ),

   // 其他国家省略

   private final string name;

   private final int id;

   country(string name, int id) {

     this .name = name;

     this .id = id;

   }

   public int getid() {

     return id;

   }

   public string getname() {

     return name;

   }

}

<dependency>

     <groupid>org.springframework.boot</groupid>

     <artifactid>spring-boot-starter-data-jpa</artifactid>

</dependency>

<dependency>

      <groupid>org.projectlombok</groupid>

      <artifactid>lombok</artifactid>

      <version> 1.18 . 22 </version>

      <scope>provided</scope>

</dependency>

package com.statistics.dynamicds.autoconfig;

import com.statistics.dynamicds.core.dynamicdatasourcerouter;

import com.statistics.dynamicds.core.country;

import com.statistics.dynamicds.core.config.multidatasourceproperties;

import com.zaxxer.hikari.hikaridatasource;

import org.springframework.beans.factory.support.abstractbeandefinition;

import org.springframework.beans.factory.support.beandefinitionbuilder;

import org.springframework.beans.factory.support.beandefinitionregistry;

import org.springframework.boot.context.properties.bind.binder;

import org.springframework.boot.jdbc.datasourcebuilder;

import org.springframework.context.environmentaware;

import org.springframework.context.annotation.importbeandefinitionregistrar;

import org.springframework.core.env.environment;

import org.springframework.core.type.annotationmetadata;

import javax.annotation.nonnull;

import java.util.map;

import java.util.stream.collectors;

import static com.statistics.dynamicds.core.config.multidatasourceproperties.prefix;

public class multidatasourceimportbeandefinitionregistrar implements importbeandefinitionregistrar, environmentaware {

   public static final string datasource_beanname = "dynamicdatasourcerouter" ;

   private environment environment;

   @override

   public void registerbeandefinitions( @nonnull annotationmetadata importingclassmetadata, beandefinitionregistry registry) {

     multidatasourceproperties multidatasourceproperties = binder.get(environment)

             .bind(prefix, multidatasourceproperties. class )

             .orelsethrow(() -> new runtimeexception( "no found dynamicds config" ));

     final hikaridatasource[] defaulttargetdatasource = { null };

     map<country, hikaridatasource> targetdatasources = multidatasourceproperties.getdatasources().entryset().stream()

             .collect(collectors.tomap(

                     map.entry::getkey,

                     entry -> {

               multidatasourceproperties.datasourceproperties datasourceproperties = entry.getvalue();

               hikaridatasource datasource = datasourcebuilder.create()

                       .type(hikaridatasource. class )

                       .driverclassname(datasourceproperties.getdriverclassname())

                       .url(datasourceproperties.geturl())

                       .username(datasourceproperties.getusername())

                       .password(datasourceproperties.getpassword())

                       .build();

               datasource.setpoolname( "hikaripool-" + entry.getkey());

               if (country.master == entry.getkey()) {

                 defaulttargetdatasource[ 0 ] = datasource;

               }

               return datasource;

             }));

     abstractbeandefinition beandefinition = beandefinitionbuilder.genericbeandefinition(dynamicdatasourcerouter. class )

             .addconstructorargvalue(defaulttargetdatasource[ 0 ])

             .addconstructorargvalue(targetdatasources)

             .getbeandefinition();

     registry.registerbeandefinition(datasource_beanname, beandefinition);

   }

   @override

   public void setenvironment( @nonnull environment environment) {

     this .environment = environment;

   }

}

package com.statistics.dynamicds.core;

import org.springframework.jdbc.datasource.lookup.abstractroutingdatasource;

import java.util.map;

public class dynamicdatasourcerouter extends abstractroutingdatasource {

   public dynamicdatasourcerouter(object defaulttargetdatasource, map<object, object> targetdatasources) {

     this .setdefaulttargetdatasource(defaulttargetdatasource);

     this .settargetdatasources(targetdatasources);

   }

   @override

   protected object determinecurrentlookupkey() {

     return datasourcecontextholder.getlookupkey();

   }

}

package com.statistics.dynamicds.core;

public class datasourcecontextholder {

   private static final threadlocal<country> holder = threadlocal.withinitial(() -> country.master);

   public static void setlookupkey(country lookupkey) {

     holder.set(lookupkey);

   }

   public static country getlookupkey() {

     return holder.get();

   }

   public static void clear() {

     holder.remove();

   }

}

package com.statistics.dynamicds.core;

import org.aspectj.lang.proceedingjoinpoint;

import org.aspectj.lang.annotation.around;

import org.aspectj.lang.annotation.aspect;

import org.aspectj.lang.annotation.pointcut;

import org.springframework.stereotype测试数据ponent;

@aspect

@component

public class dynamicdatasourceaspect {

   @pointcut ( "execution(* com.statistics.dao..*.*(..))" )

   void aspect() {

   }

   @around ( "aspect()" )

   public object around(proceedingjoinpoint joinpoint) throws throwable {

     for (object arg : joinpoint.getargs()) {

       if (arg instanceof country) {

         datasourcecontextholder.setlookupkey((country) arg);

         break ;

       }

     }

     try {

       return joinpoint.proceed();

     } finally {

       datasourcecontextholder.clear();

     }

   }

}