SpringBoot利用自定义注解实现多数据源

秒速五厘米 2024-03-17 20:30 98阅读 0赞

## 自定义多数据源 ##

SpringBoot利用自定义注解实现多数据源，前置知识：注解、Aop、SpringBoot整合Mybaits

### 1、搭建工程 ###

创建一个SpringBoot工程，并引入依赖

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependencies&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;groupId&gt;org.springframework.boot&lt;/groupId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;artifactId&gt;spring-boot-starter-web&lt;/artifactId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;!-- 解析多数据源注解 --&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;groupId&gt;org.springframework.boot&lt;/groupId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;artifactId&gt;spring-boot-starter-aop&lt;/artifactId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;groupId&gt;org.mybatis.spring.boot&lt;/groupId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;artifactId&gt;mybatis-spring-boot-starter&lt;/artifactId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;version&gt;2.3.0&lt;/version&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;groupId&gt;com.mysql&lt;/groupId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;artifactId&gt;mysql-connector-j&lt;/artifactId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;scope&gt;runtime&lt;/scope&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;!-- https://mvnrepository.com/artifact/com.alibaba/druid-spring-boot-starter --&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;groupId&gt;com.alibaba&lt;/groupId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;artifactId&gt;druid-spring-boot-starter&lt;/artifactId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;version&gt;1.2.18&lt;/version&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;groupId&gt;org.springframework.boot&lt;/groupId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;artifactId&gt;spring-boot-starter-test&lt;/artifactId&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;scope&gt;test&lt;/scope&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependency&gt;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>&lt;/dependencies&gt;</code></td> 
  </tr> 
 </tbody> 
</table>

### 2、定义多数据源注解 ###

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* 1、定义多数据源注解</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @author ss_419</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* TODO 这个注解将来可以加在service类上或者方法上，通过value属性来指定类或者方法应该使用那个数据源</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Retention(RetentionPolicy.RUNTIME)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Target({ElementType.TYPE, ElementType.METHOD})</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public @interface DataSource { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>String value() default DataSourceType.DEFAULT_DS_NAME;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

### 3、创建一个多数据上下文对象 ###

这个类用来存储当前线程所使用的数据源名称

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* TODO 这个类用来存储当前线程所使用的数据源名称</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @author ss_419</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @version 1.0</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @date 2023/5/21 09:21</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class DynamicDataSourceContextHolder { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private static final ThreadLocal&lt;String&gt; CONTEXT_HOLDER = new ThreadLocal&lt;&gt;();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public static void setDataSourceType(String dataSourceType) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>CONTEXT_HOLDER.set(dataSourceType);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public static String getDataSourceType() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return CONTEXT_HOLDER.get();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public static void clearDataSourceType() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>CONTEXT_HOLDER.remove();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

### 4、配置aop ###

*  @annotation(org.pp.dd.annotation.DataSource) 如果有@DataSource注解就给拦截下来
 *  @within(org.pp.dd.annotation.DataSource) 表示类上有@DataSource注解就将类中的方法给拦截下来

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* TODO</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @author ss_419</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @version 1.0</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @date 2023/5/21 09:42</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Component</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Aspect</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Order(11)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class DataSourceAspect { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* 定义切点</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @annotation(org.pp.dd.annotation.DataSource) 如果有@DataSource注解就给拦截下来</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @within(org.pp.dd.annotation.DataSource) 表示类上有@DataSource注解就将类中的方法给拦截下来</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Pointcut("@annotation(org.pp.dd.annotation.DataSource) || @within(org.pp.dd.annotation.DataSource)")</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void pc() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* 环绕通知</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @param pjp</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @return</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Around("pc()")</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Object around(ProceedingJoinPoint pjp) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 获取方法上的有效注解</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>DataSource dataSource = getDataSource(pjp);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>if (dataSource != null) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 获取注解中数据源的名称</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>String value = dataSource.value();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>DynamicDataSourceContextHolder.setDataSourceType(value);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>try { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return pjp.proceed();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>} catch (Throwable e) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>throw new RuntimeException(e);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>} finally { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>DynamicDataSourceContextHolder.clearDataSourceType();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private DataSource getDataSource(ProceedingJoinPoint pjp) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>MethodSignature signature = (MethodSignature) pjp.getSignature();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 获取方法上的注解</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>DataSource annotation = AnnotationUtils.findAnnotation(signature.getMethod(), DataSource.class);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>if (annotation != null) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 说明方法上有注解</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return annotation;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return (DataSource) AnnotationUtils.findAnnotation(signature.getDeclaringType(), DataSource.class);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

### 5、读取参数DruidProperties ###

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* TODO 读取数据源</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @author ss_419</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @version 1.0</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @date 2023/5/21 10:20</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@ConfigurationProperties(prefix = "spring.datasource")</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class DruidProperties { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private String type;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private String driverClassName;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private Map&lt;String, Map&lt;String ,String&gt;&gt; ds;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private Integer initialSize;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private Integer minIdle;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private Integer maxActive;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private Integer maxWait;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* 在这个方法中设置公共属性</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @param dataSource</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @return</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public DataSource dataSource(DruidDataSource dataSource){ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>dataSource.setInitialSize(initialSize);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>dataSource.setMinIdle(minIdle);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>dataSource.setMaxActive(maxActive);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>dataSource.setMaxWait(maxWait);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return dataSource;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public String getType() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return type;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setType(String type) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.type = type;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public String getDriverClassName() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return driverClassName;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setDriverClassName(String driverClassName) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.driverClassName = driverClassName;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Map&lt;String, Map&lt;String, String&gt;&gt; getDs() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return ds;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setDs(Map&lt;String, Map&lt;String, String&gt;&gt; ds) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.ds = ds;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Integer getInitialSize() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return initialSize;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setInitialSize(Integer initialSize) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.initialSize = initialSize;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Integer getMinIdle() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return minIdle;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setMinIdle(Integer minIdle) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.minIdle = minIdle;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Integer getMaxActive() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return maxActive;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setMaxActive(Integer maxActive) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.maxActive = maxActive;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Integer getMaxWait() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return maxWait;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public void setMaxWait(Integer maxWait) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.maxWait = maxWait;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

### 6、加载数据源LoadDataSource ###

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* TODO 加载数据源</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @author ss_419</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @version 1.0</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @date 2023/5/21 10:30</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Component</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@EnableConfigurationProperties(DruidProperties.class)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class LoadDataSource { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Autowired</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>DruidProperties druidProperties;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Map&lt;String, DataSource&gt; loadAllDataSource() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Map&lt;String, DataSource&gt; map = new HashMap&lt;&gt;();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Map&lt;String, Map&lt;String, String&gt;&gt; ds = druidProperties.getDs();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>try { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Set&lt;String&gt; keySet = ds.keySet();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>for (String key : keySet) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>map.put(key, druidProperties.dataSource((DruidDataSource) DruidDataSourceFactory.createDataSource(ds.get(key))));</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>} catch (Exception e) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>throw new RuntimeException(e);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return map;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

### 7、定义数据源管理器 ###

当系统需要调用数据源的时候，数据源以key-value存起来，当需要数据源时调用determineCurrentLookupKey()方法来获取数据源。  
由于本人实力原因，解答不了大家这里的疑惑。大致功能 通过修改本地线程的值，来实现数据源的切换。

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* TODO 设置数据源</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* 当系统需要调用数据源的时候，数据源以key-value存起来，当需要数据源时调用determineCurrentLookupKey()方法</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @author ss_419</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @version 1.0</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @date 2023/5/21 10:47</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Component</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class DynamicDataSource extends AbstractRoutingDataSource { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public DynamicDataSource(LoadDataSource loadDataSource) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>//1、设置所有的数据源</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Map&lt;String, DataSource&gt; allDs = loadDataSource.loadAllDataSource();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>super.setTargetDataSources(new HashMap&lt;&gt;(allDs));</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>//2、设置默认数据源</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>super.setDefaultTargetDataSource(allDs.get(DataSourceType.DEFAULT_DS_NAME));</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>super.afterPropertiesSet();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>/**</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* 这个方法用来返回数据源名称，当系统需要获取数据源的时候会自动调用该方法获取数据源名称</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>* @return</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>*/</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Override</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>protected Object determineCurrentLookupKey() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return DynamicDataSourceContextHolder.getDataSourceType();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

定一个用于存储数据库类型的接口，这个接口类似于枚举类：

### 8、测试 ###

创建User实体：

创建UserService：

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>@Service</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 在类上加注解的效果，会使该类的所有方法都切入到新的数据源中</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>//@DataSource</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class UserService { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Autowired</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>UserMapper userMapper;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 在方法上加注解的效果，只会让指定的方法切入到另一个数据源中</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>//@DataSource("slave")</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public List&lt;User&gt; findUsers(){ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return userMapper.findAllUsers();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

创建UserMapper：

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>@Mapper</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public interface UserMapper { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Select("SELECT * FROM user")</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>List&lt;User&gt; findAllUsers();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

测试类：

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>@SpringBootTest</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>class DynamicDatasourcesApplicationTests { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Autowired</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>UserService userService;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Test</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>void contextLoads() { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>List&lt;User&gt; users = userService.findUsers();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>users.stream()</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>.forEach(user -&gt; System.out.println(user));</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

默认选择主库的数据源：

![image][]

执行结果如下：

![image][image 1]

在Service上加上注解，指定数据源为从库：

![image][image 2]

执行结果如下：

![image][image 3]

[image]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/6b403c9b6470490c9de17bb40d463140.png
[image 1]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/bafd712ffe604bc384e8cb054d8978a3.png
[image 2]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/724fa78cf8644bc18486d26ffc65834d.png
[image 3]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/12f4b1799e584627ae49c2932eaf2f34.png