├── pom.xml
└── src
    ├── main
        ├── java
        │   └── com
        │   │   └── xh
        │   │       ├── MyGenerator.java
        │   │       ├── SpringBootDynamicDataSourceApplication.java
        │   │       ├── controller
        │   │           └── SysUserController.java
        │   │       ├── datasource
        │   │           ├── CurDataSource.java
        │   │           ├── DataSourceAspect.java
        │   │           ├── DataSourceNames.java
        │   │           ├── DynamicDataSource.java
        │   │           └── DynamicDataSourceConfig.java
        │   │       ├── entity
        │   │           └── SysUser.java
        │   │       ├── mapper
        │   │           └── SysUserMapper.java
        │   │       └── service
        │   │           ├── SysUserService.java
        │   │           └── impl
        │   │               └── SysUserServiceImpl.java
        └── resources
        │   ├── application.properties
        │   ├── doc
        │       └── mysql.sql
        │   └── mapper
        │       └── SysUserMapper.xml
    └── test
        └── java
            └── com
                └── xh
                    └── SpringBootDynamicCurDataSourceApplicationTests.java


/pom.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8"?>
 2 | <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 3 |          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
 4 |     <modelVersion>4.0.0</modelVersion>
 5 |     <parent>
 6 |         <groupId>org.springframework.boot</groupId>
 7 |         <artifactId>spring-boot-starter-parent</artifactId>
 8 |         <version>2.1.5.RELEASE</version>
 9 |         <relativePath/> <!-- lookup parent from repository -->
10 |     </parent>
11 |     <groupId>com.xh</groupId>
12 |     <artifactId>spring-boot-dynamic-data-source</artifactId>
13 |     <version>1.0.0-SNAPSHOT</version>
14 |     <name>spring-boot-dynamic-data-source</name>
15 |     <description>Demo project for Spring Boot</description>
16 | 
17 |     <properties>
18 |         <java.version>1.8</java.version>
19 |     </properties>
20 | 
21 |     <dependencies>
22 |         <dependency>
23 |             <groupId>org.springframework.boot</groupId>
24 |             <artifactId>spring-boot-starter-aop</artifactId>
25 |         </dependency>
26 |         <dependency>
27 |             <groupId>org.springframework.boot</groupId>
28 |             <artifactId>spring-boot-starter-web</artifactId>
29 |         </dependency>
30 | 
31 |         <!--mybatis plus-->
32 |         <dependency>
33 |             <groupId>com.baomidou</groupId>
34 |             <artifactId>mybatis-plus-boot-starter</artifactId>
35 |             <version>3.0.1</version>
36 |         </dependency>
37 |         <dependency>
38 |             <groupId>mysql</groupId>
39 |             <artifactId>mysql-connector-java</artifactId>
40 |             <scope>runtime</scope>
41 |         </dependency>
42 |         <dependency>
43 |             <groupId>org.projectlombok</groupId>
44 |             <artifactId>lombok</artifactId>
45 |             <optional>true</optional>
46 |         </dependency>
47 |         <dependency>
48 |             <groupId>org.springframework.boot</groupId>
49 |             <artifactId>spring-boot-starter-test</artifactId>
50 |             <scope>test</scope>
51 |         </dependency>
52 |         <!--druid-->
53 |         <dependency>
54 |             <groupId>com.alibaba</groupId>
55 |             <artifactId>druid-spring-boot-starter</artifactId>
56 |             <version>1.1.10</version>
57 |         </dependency>
58 |         <!-- https://mvnrepository.com/artifact/org.apache.commons/commons-lang3 -->
59 |         <dependency>
60 |             <groupId>org.apache.commons</groupId>
61 |             <artifactId>commons-lang3</artifactId>
62 |             <version>3.9</version>
63 |         </dependency>
64 |         <!-- swagger -->
65 |         <dependency>
66 |             <groupId>io.springfox</groupId>
67 |             <artifactId>springfox-swagger2</artifactId>
68 |             <version>2.9.2</version>
69 |         </dependency>
70 |         <dependency>
71 |             <groupId>com.github.caspar-chen</groupId>
72 |             <artifactId>swagger-ui-layer</artifactId>
73 |             <version>1.1.3</version>
74 |         </dependency>
75 |     </dependencies>
76 | 
77 |     <build>
78 |         <plugins>
79 |             <plugin>
80 |                 <groupId>org.springframework.boot</groupId>
81 |                 <artifactId>spring-boot-maven-plugin</artifactId>
82 |             </plugin>
83 |             <!-- 打包跳过测试 -->
84 |             <plugin>
85 |                 <groupId>org.apache.maven.plugins</groupId>
86 |                 <artifactId>maven-surefire-plugin</artifactId>
87 |                 <version>2.18.1</version>
88 |                 <configuration>
89 |                     <skipTests>true</skipTests>
90 |                 </configuration>
91 |             </plugin>
92 |         </plugins>
93 |     </build>
94 | 
95 | </project>
96 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/MyGenerator.java:
--------------------------------------------------------------------------------
  1 | package com.xh;
  2 | 
  3 | import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
  4 | import com.baomidou.mybatisplus.core.toolkit.StringPool;
  5 | import com.baomidou.mybatisplus.core.toolkit.StringUtils;
  6 | import com.baomidou.mybatisplus.generator.AutoGenerator;
  7 | import com.baomidou.mybatisplus.generator.InjectionConfig;
  8 | import com.baomidou.mybatisplus.generator.config.*;
  9 | import com.baomidou.mybatisplus.generator.config.po.TableInfo;
 10 | import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
 11 | import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
 12 | 
 13 | import java.util.ArrayList;
 14 | import java.util.List;
 15 | import java.util.Scanner;
 16 | 
 17 | /**
 18 |  * 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
 19 |  * 注意安装配置来生成BaseEntity、BaseController、还有包路径相关等东西。
 20 |  * <p/>
 21 |  * current table :
 22 |  * category,comment,post,user,user_action,user_collection,user_message
 23 |  * <p/>
 24 |  * 完成后：
 25 |  * 生成BaseEntity、BaseController、还有包路径相关等东西
 26 |  * <p/>
 27 |  * 代码生成之后，看到resource/mapper这里，把这个null去掉
 28 |  * <p/>
 29 |  * 注意一下：还需要配置@MapperScan("com.xh.blog.mapper")注解。说明mapper的扫描包。
 30 |  * <p/>
 31 |  * 接下来，写一个小测试，测试一下mybatis plus有没集成成功
 32 |  *
 33 |  * @author xiaohe
 34 |  * @version V1.0.0
 35 |  */
 36 | public class MyGenerator {
 37 | 
 38 |     /**
 39 |      * <p>
 40 |      * 读取控制台内容
 41 |      * </p>
 42 |      */
 43 |     private static String scanner(String tip) {
 44 |         Scanner scanner = new Scanner(System.in);
 45 |         StringBuilder help = new StringBuilder();
 46 |         // sys_user
 47 |         help.append("请输入" + tip + "：");
 48 |         System.out.println(help.toString());
 49 |         if (scanner.hasNext()) {
 50 |             String ipt = scanner.next();
 51 |             if (StringUtils.isNotEmpty(ipt)) {
 52 |                 return ipt;
 53 |             }
 54 |         }
 55 |         throw new MybatisPlusException("请输入正确的" + tip + "！");
 56 |     }
 57 | 
 58 |     public static void main(String[] args) {
 59 |         // 代码生成器
 60 |         AutoGenerator mpg = new AutoGenerator();
 61 | 
 62 |         // 全局配置
 63 |         GlobalConfig gc = new GlobalConfig();
 64 |         String projectPath = System.getProperty("user.dir");
 65 |         gc.setOutputDir(projectPath + "/src/main/java");
 66 |         // gc.setOutputDir("D:\\test")
 67 |         gc.setAuthor("xiaohe");
 68 |         gc.setOpen(false);
 69 |         // 实体属性 Swagger2 注解
 70 |         gc.setSwagger2(true);
 71 |         gc.setServiceName("%sService");
 72 |         mpg.setGlobalConfig(gc);
 73 | 
 74 |         // 数据源配置
 75 |         DataSourceConfig dsc = new DataSourceConfig();
 76 |         dsc.setUrl("jdbc:mysql://localhost:3306/renren_fast?useUnicode=true&useSSL=false&characterEncoding=utf8&serverTimezone=UTC");
 77 |         // dsc.setSchemaName("public")
 78 |         dsc.setDriverName("com.mysql.cj.jdbc.Driver");
 79 |         dsc.setUsername("root");
 80 |         dsc.setPassword("accp");
 81 |         mpg.setDataSource(dsc);
 82 | 
 83 |         // 包配置
 84 |         PackageConfig pc = new PackageConfig();
 85 |         pc.setModuleName(null);
 86 |         pc.setParent("com.xh");
 87 |         mpg.setPackageInfo(pc);
 88 | 
 89 |         // 自定义配置
 90 |         InjectionConfig cfg = new InjectionConfig() {
 91 |             @Override
 92 |             public void initMap() {
 93 |                 // to do nothing
 94 |             }
 95 |         };
 96 | 
 97 |         // 如果模板引擎是 freemarker
 98 |         String templatePath = "/templates/mapper.xml.ftl";
 99 |         // 如果模板引擎是 velocity
100 |         // String templatePath = "/templates/mapper.xml.vm";
101 | 
102 |         // 自定义输出配置
103 |         List<FileOutConfig> focList = new ArrayList<>();
104 |         // 自定义配置会被优先输出
105 |         focList.add(new FileOutConfig(templatePath) {
106 |             @Override
107 |             public String outputFile(TableInfo tableInfo) {
108 |                 // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
109 |                 return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
110 |                         + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
111 |             }
112 |         });
113 | 
114 |         cfg.setFileOutConfigList(focList);
115 |         mpg.setCfg(cfg);
116 | 
117 |         // 配置模板
118 |         TemplateConfig templateConfig = new TemplateConfig();
119 | 
120 |         // 配置自定义输出模板
121 |         //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
122 |         // templateConfig.setEntity("templates/entity2.java")
123 |         // templateConfig.setService()
124 |         // templateConfig.setController()
125 | 
126 |         templateConfig.setXml(null);
127 |         mpg.setTemplate(templateConfig);
128 | 
129 |         // 策略配置
130 |         StrategyConfig strategy = new StrategyConfig();
131 |         strategy.setNaming(NamingStrategy.underline_to_camel);
132 |         strategy.setColumnNaming(NamingStrategy.underline_to_camel);
133 | //        strategy.setSuperEntityClass("com.xh.entity.BaseEntity")
134 |         strategy.setEntityLombokModel(true);
135 |         strategy.setRestControllerStyle(true);
136 | //        strategy.setSuperControllerClass("com.xh.controller.BaseController")
137 |         strategy.setInclude(scanner("表名，多个英文逗号分割").split(","));
138 | //        strategy.setSuperEntityColumns("id", "created", "modified", "status")
139 |         strategy.setControllerMappingHyphenStyle(true);
140 |         strategy.setTablePrefix(pc.getModuleName() + "_");
141 |         mpg.setStrategy(strategy);
142 |         mpg.setTemplateEngine(new FreemarkerTemplateEngine());
143 |         mpg.execute();
144 |     }
145 | }


--------------------------------------------------------------------------------
/src/main/java/com/xh/SpringBootDynamicDataSourceApplication.java:
--------------------------------------------------------------------------------
 1 | package com.xh;
 2 | 
 3 | import com.xh.datasource.DynamicDataSourceConfig;
 4 | import org.mybatis.spring.annotation.MapperScan;
 5 | import org.springframework.boot.SpringApplication;
 6 | import org.springframework.boot.autoconfigure.SpringBootApplication;
 7 | import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
 8 | import org.springframework.context.annotation.Import;
 9 | 
10 | /**
11 |  * @author xiaohe
12 |  */
13 | @MapperScan("com.xh.mapper")
14 | @SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
15 | @Import({DynamicDataSourceConfig.class})
16 | public class SpringBootDynamicDataSourceApplication {
17 | 
18 |     public static void main(String[] args) {
19 |         SpringApplication.run(SpringBootDynamicDataSourceApplication.class, args);
20 |     }
21 | 
22 | }
23 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/controller/SysUserController.java:
--------------------------------------------------------------------------------
 1 | package com.xh.controller;
 2 | 
 3 | 
 4 | import org.springframework.web.bind.annotation.RequestMapping;
 5 | 
 6 | import org.springframework.web.bind.annotation.RestController;
 7 | 
 8 | /**
 9 |  * <p>
10 |  * 系统用户 前端控制器
11 |  * </p>
12 |  *
13 |  * @author xiaohe
14 |  * @since 2019-06-04
15 |  */
16 | @RestController
17 | @RequestMapping("/sys-user")
18 | public class SysUserController {
19 | 
20 | }
21 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/datasource/CurDataSource.java:
--------------------------------------------------------------------------------
 1 | package com.xh.datasource;
 2 | 
 3 | import java.lang.annotation.*;
 4 | 
 5 | /**
 6 |  * 多数据源注解
 7 |  * <p/>
 8 |  * 指定要使用的数据源
 9 |  *
10 |  * @author xiaohe
11 |  * @version V1.0.0
12 |  */
13 | @Target(ElementType.METHOD)
14 | @Retention(RetentionPolicy.RUNTIME)
15 | @Documented
16 | public @interface CurDataSource {
17 | 
18 |     String name() default "";
19 | 
20 | }


--------------------------------------------------------------------------------
/src/main/java/com/xh/datasource/DataSourceAspect.java:
--------------------------------------------------------------------------------
 1 | package com.xh.datasource;
 2 | 
 3 | import lombok.extern.slf4j.Slf4j;
 4 | import org.aspectj.lang.ProceedingJoinPoint;
 5 | import org.aspectj.lang.annotation.Around;
 6 | import org.aspectj.lang.annotation.Aspect;
 7 | import org.aspectj.lang.annotation.Pointcut;
 8 | import org.aspectj.lang.reflect.MethodSignature;
 9 | import org.springframework.core.Ordered;
10 | import org.springframework.stereotype.Component;
11 | 
12 | import java.lang.reflect.Method;
13 | 
14 | /**
15 |  * 多数据源，切面处理类
16 |  *
17 |  * @author xiaohe
18 |  * @version V1.0.0
19 |  */
20 | @Slf4j
21 | @Aspect
22 | @Component
23 | public class DataSourceAspect implements Ordered {
24 | 
25 |     @Pointcut("@annotation(com.xh.datasource.CurDataSource)")
26 |     public void dataSourcePointCut() {
27 | 
28 |     }
29 | 
30 |     @Around("dataSourcePointCut()")
31 |     public Object around(ProceedingJoinPoint point) throws Throwable {
32 |         MethodSignature signature = (MethodSignature) point.getSignature();
33 |         Method method = signature.getMethod();
34 | 
35 |         CurDataSource ds = method.getAnnotation(CurDataSource.class);
36 |         if (ds == null) {
37 |             DynamicDataSource.setDataSource(DataSourceNames.FIRST);
38 |             log.debug("set datasource is " + DataSourceNames.FIRST);
39 |         } else {
40 |             DynamicDataSource.setDataSource(ds.name());
41 |             log.debug("set datasource is " + ds.name());
42 |         }
43 | 
44 |         try {
45 |             return point.proceed();
46 |         } finally {
47 |             DynamicDataSource.clearDataSource();
48 |             log.debug("clean datasource");
49 |         }
50 |     }
51 | 
52 |     @Override
53 |     public int getOrder() {
54 |         return 1;
55 |     }
56 | }
57 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/datasource/DataSourceNames.java:
--------------------------------------------------------------------------------
 1 | package com.xh.datasource;
 2 | 
 3 | /**
 4 |  * 增加多数据源，在此配置
 5 |  *
 6 |  * @author xiaohe
 7 |  * @version V1.0.0
 8 |  */
 9 | public interface DataSourceNames {
10 | 
11 |     String FIRST = "first";
12 | 
13 |     String SECOND = "second";
14 | 
15 | }
16 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/datasource/DynamicDataSource.java:
--------------------------------------------------------------------------------
 1 | package com.xh.datasource;
 2 | 
 3 | import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
 4 | 
 5 | import javax.sql.DataSource;
 6 | import java.util.Map;
 7 | 
 8 | /**
 9 |  * 扩展 Spring 的 AbstractRoutingDataSource 抽象类，重写 determineCurrentLookupKey 方法
10 |  * 动态数据源
11 |  * determineCurrentLookupKey() 方法决定使用哪个数据源
12 |  *
13 |  * @author xiaohe
14 |  * @version V1.0.0
15 |  */
16 | public class DynamicDataSource extends AbstractRoutingDataSource {
17 | 
18 |     /**
19 |      * ThreadLocal 用于提供线程局部变量，在多线程环境可以保证各个线程里的变量独立于其它线程里的变量。
20 |      * 也就是说 ThreadLocal 可以为每个线程创建一个【单独的变量副本】，相当于线程的 private static 类型变量。
21 |      */
22 |     private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();
23 | 
24 |     /**
25 |      * 决定使用哪个数据源之前需要把多个数据源的信息以及默认数据源信息配置好
26 |      *
27 |      * @param defaultTargetDataSource 默认数据源
28 |      * @param targetDataSources       目标数据源
29 |      */
30 |     public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
31 |         super.setDefaultTargetDataSource(defaultTargetDataSource);
32 |         super.setTargetDataSources(targetDataSources);
33 |         super.afterPropertiesSet();
34 |     }
35 | 
36 |     @Override
37 |     protected Object determineCurrentLookupKey() {
38 |         return getDataSource();
39 |     }
40 | 
41 |     public static void setDataSource(String dataSource) {
42 |         CONTEXT_HOLDER.set(dataSource);
43 |     }
44 | 
45 |     public static String getDataSource() {
46 |         return CONTEXT_HOLDER.get();
47 |     }
48 | 
49 |     public static void clearDataSource() {
50 |         CONTEXT_HOLDER.remove();
51 |     }
52 | 
53 | }
54 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/datasource/DynamicDataSourceConfig.java:
--------------------------------------------------------------------------------
 1 | package com.xh.datasource;
 2 | 
 3 | import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
 4 | import org.springframework.boot.context.properties.ConfigurationProperties;
 5 | import org.springframework.context.annotation.Bean;
 6 | import org.springframework.context.annotation.Configuration;
 7 | import org.springframework.context.annotation.Primary;
 8 | 
 9 | import javax.sql.DataSource;
10 | import java.util.HashMap;
11 | import java.util.Map;
12 | 
13 | /**
14 |  * 配置多数据源
15 |  * @author xiaohe
16 |  * @version V1.0.0
17 |  */
18 | @Configuration
19 | public class DynamicDataSourceConfig {
20 | 
21 |     @Bean
22 |     @ConfigurationProperties("spring.datasource.druid.first")
23 |     public DataSource firstDataSource(){
24 | 
25 |         return DruidDataSourceBuilder.create().build();
26 |     }
27 | 
28 |     @Bean
29 |     @ConfigurationProperties("spring.datasource.druid.second")
30 |     public DataSource secondDataSource(){
31 | 
32 |         return DruidDataSourceBuilder.create().build();
33 |     }
34 | 
35 |     @Bean
36 |     @Primary
37 |     public DynamicDataSource dataSource(DataSource firstDataSource, DataSource secondDataSource) {
38 |         Map<Object, Object> targetDataSources = new HashMap<>(5);
39 |         targetDataSources.put(DataSourceNames.FIRST, firstDataSource);
40 |         targetDataSources.put(DataSourceNames.SECOND, secondDataSource);
41 |         return new DynamicDataSource(firstDataSource, targetDataSources);
42 |     }
43 | 
44 | }
45 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/entity/SysUser.java:
--------------------------------------------------------------------------------
 1 | package com.xh.entity;
 2 | 
 3 | import com.baomidou.mybatisplus.annotation.IdType;
 4 | import com.baomidou.mybatisplus.annotation.TableId;
 5 | import java.time.LocalDateTime;
 6 | import java.io.Serializable;
 7 | import io.swagger.annotations.ApiModel;
 8 | import io.swagger.annotations.ApiModelProperty;
 9 | import lombok.Data;
10 | import lombok.EqualsAndHashCode;
11 | import lombok.ToString;
12 | import lombok.experimental.Accessors;
13 | 
14 | /**
15 |  * <p>
16 |  * 系统用户
17 |  * </p>
18 |  *
19 |  * @author xiaohe
20 |  * @since 2019-06-04
21 |  */
22 | @Data
23 | @EqualsAndHashCode(callSuper = false)
24 | @Accessors(chain = true)
25 | @ApiModel(value="SysUser对象", description="系统用户")
26 | @ToString
27 | public class SysUser implements Serializable {
28 | 
29 |     private static final long serialVersionUID = 1L;
30 | 
31 |     @TableId(value = "user_id", type = IdType.AUTO)
32 |     private Long userId;
33 | 
34 |     @ApiModelProperty(value = "用户名")
35 |     private String username;
36 | 
37 |     @ApiModelProperty(value = "密码")
38 |     private String password;
39 | 
40 |     @ApiModelProperty(value = "盐")
41 |     private String salt;
42 | 
43 |     @ApiModelProperty(value = "邮箱")
44 |     private String email;
45 | 
46 |     @ApiModelProperty(value = "手机号")
47 |     private String mobile;
48 | 
49 |     @ApiModelProperty(value = "状态  0：禁用   1：正常")
50 |     private Integer status;
51 | 
52 |     @ApiModelProperty(value = "创建者ID")
53 |     private Long createUserId;
54 | 
55 |     @ApiModelProperty(value = "创建时间")
56 |     private LocalDateTime createTime;
57 | 
58 | 
59 | }
60 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/mapper/SysUserMapper.java:
--------------------------------------------------------------------------------
 1 | package com.xh.mapper;
 2 | 
 3 | import com.xh.entity.SysUser;
 4 | import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 5 | 
 6 | /**
 7 |  * <p>
 8 |  * 系统用户 Mapper 接口
 9 |  * </p>
10 |  *
11 |  * @author xiaohe
12 |  * @since 2019-06-04
13 |  */
14 | public interface SysUserMapper extends BaseMapper<SysUser> {
15 | 
16 | }
17 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/service/SysUserService.java:
--------------------------------------------------------------------------------
 1 | package com.xh.service;
 2 | 
 3 | import com.xh.entity.SysUser;
 4 | import com.baomidou.mybatisplus.extension.service.IService;
 5 | 
 6 | /**
 7 |  * <p>
 8 |  * 系统用户 服务类
 9 |  * </p>
10 |  *
11 |  * @author xiaohe
12 |  * @since 2019-06-04
13 |  */
14 | public interface SysUserService extends IService<SysUser> {
15 | 
16 |     SysUser findUserByFirstDb(long id);
17 | 
18 |     SysUser findUserBySecondDb(long id);
19 | 
20 | }
21 | 


--------------------------------------------------------------------------------
/src/main/java/com/xh/service/impl/SysUserServiceImpl.java:
--------------------------------------------------------------------------------
 1 | package com.xh.service.impl;
 2 | 
 3 | import com.xh.datasource.CurDataSource;
 4 | import com.xh.datasource.DataSourceNames;
 5 | import com.xh.entity.SysUser;
 6 | import com.xh.mapper.SysUserMapper;
 7 | import com.xh.service.SysUserService;
 8 | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 9 | import org.springframework.stereotype.Service;
10 | 
11 | /**
12 |  * <p>
13 |  * 系统用户 服务实现类
14 |  * </p>
15 |  *
16 |  * @author xiaohe
17 |  * @since 2019-06-04
18 |  */
19 | @Service
20 | public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> implements SysUserService {
21 | 
22 |     @Override
23 |     public SysUser findUserByFirstDb(long id) {
24 |         return this.baseMapper.selectById(id);
25 |     }
26 | 
27 |     @CurDataSource(name = DataSourceNames.SECOND)
28 |     @Override
29 |     public SysUser findUserBySecondDb(long id) {
30 |         return this.baseMapper.selectById(id);
31 |     }
32 | 
33 | }
34 | 


--------------------------------------------------------------------------------
/src/main/resources/application.properties:
--------------------------------------------------------------------------------
 1 | 
 2 | server.port=8082
 3 | 
 4 | spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
 5 | spring.datasource.driverClassName=com.mysql.jdbc.Driver
 6 | # 数据源1
 7 | spring.datasource.druid.first.url=jdbc:mysql://localhost:3306/test1?allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=GMT%2B8
 8 | spring.datasource.druid.first.username=root
 9 | spring.datasource.druid.first.password=accp
10 | # 数据源2  需要创建对应数据库  更改该库中 sys_user 表
11 | spring.datasource.druid.second.url=jdbc:mysql://localhost:3306/test2?allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=GMT%2B8
12 | spring.datasource.druid.second.username=root
13 | spring.datasource.druid.second.password=accp
14 | 
15 | 


--------------------------------------------------------------------------------
/src/main/resources/doc/mysql.sql:
--------------------------------------------------------------------------------
 1 | -- 系统用户
 2 | CREATE TABLE `sys_user`
 3 | (
 4 |   `user_id`        bigint      NOT NULL AUTO_INCREMENT,
 5 |   `username`       varchar(50) NOT NULL COMMENT '用户名',
 6 |   `password`       varchar(100) COMMENT '密码',
 7 |   `salt`           varchar(20) COMMENT '盐',
 8 |   `email`          varchar(100) COMMENT '邮箱',
 9 |   `mobile`         varchar(100) COMMENT '手机号',
10 |   `status`         tinyint COMMENT '状态  0：禁用   1：正常',
11 |   `create_user_id` bigint(20) COMMENT '创建者ID',
12 |   `create_time`    datetime COMMENT '创建时间',
13 |   PRIMARY KEY (`user_id`),
14 |   UNIQUE INDEX (`username`)
15 | ) ENGINE = InnoDB
16 |   DEFAULT CHARSET = utf8 COMMENT ='系统用户';
17 | 
18 | -- 初始数据 
19 | INSERT INTO `sys_user` (`user_id`, `username`, `password`, `salt`, `email`, `mobile`, `status`, `create_user_id`,
20 |                         `create_time`)
21 | VALUES ('1', 'admin', '9ec9750e709431dad22365cabc5c625482e574c74adaebba7dd02f1129e4ce1d', 'YzcmCZNvbXocrsz9dm8e',
22 |         'root@renren.io', '13612345678', '1', '1', '2016-11-11 11:11:11');
23 | 


--------------------------------------------------------------------------------
/src/main/resources/mapper/SysUserMapper.xml:
--------------------------------------------------------------------------------
1 | <?xml version="1.0" encoding="UTF-8"?>
2 | <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
3 | <mapper namespace="com.xh.mapper.SysUserMapper">
4 | 
5 | </mapper>
6 | 


--------------------------------------------------------------------------------
/src/test/java/com/xh/SpringBootDynamicCurDataSourceApplicationTests.java:
--------------------------------------------------------------------------------
 1 | package com.xh;
 2 | 
 3 | import com.xh.entity.SysUser;
 4 | import com.xh.service.SysUserService;
 5 | import lombok.extern.slf4j.Slf4j;
 6 | import org.junit.Test;
 7 | import org.junit.runner.RunWith;
 8 | import org.springframework.beans.factory.annotation.Autowired;
 9 | import org.springframework.boot.test.context.SpringBootTest;
10 | import org.springframework.test.context.junit4.SpringRunner;
11 | 
12 | @Slf4j
13 | @RunWith(SpringRunner.class)
14 | @SpringBootTest
15 | public class SpringBootDynamicCurDataSourceApplicationTests {
16 | 
17 |     @Autowired
18 |     private SysUserService userService;
19 | 
20 |     @Test
21 |     public void contextLoads() {
22 |         SysUser user = userService.getById(1);
23 |         log.info(user.toString());
24 |     }
25 | 
26 |     @Test
27 |     public void test() {
28 |         SysUser user = userService.findUserByFirstDb(1);
29 |         log.info("第一个数据库 : [{}]", user.toString());
30 |         SysUser user2 = userService.findUserBySecondDb(1);
31 |         log.info("第二个数据库 : [{}]", user2.toString());
32 |     }
33 | }
34 | 


--------------------------------------------------------------------------------