零基础尝试mybatis-plus读写分离

 看了好几篇博友写的文章，关于spring-boot整合mybatis-plus实现读写分离，不过都是缺这少那的，跑不起来，所以自己实操了一次，做个记录
  实现方式为使用Aop切面
1、增加数据库枚举类

1. /**
2. * 数据库类型
3. */
4. public enum DBTypeEnum {
5.     /**
6.      * 主节点
7.      */
8.     MASTER,
9.     /**
10.      * 从
11.      */
12.     SLAVE
13. }

复制代码

2、配置数据源

1. /**
2. * 多数据源配置
3. */
4. @AutoConfigureBefore(DruidDataSourceAutoConfigure.class)
5. @Configuration
6. @ConfigurationProperties(prefix = "spring.datasource.druid")
7. @Data
8. public class DataSourceConfig {
10.     private int initialSize;
11.     private int maxActive;
12.     private int minIdle;
13.     private long maxWait;
14.     private long minEvictableIdleTimeMillis;
15.     private long timeBetweenEvictionRunsMillis;
16.     private boolean testWhileIdle;
18.     /**
19.      * 配置主数据源
20.      *
21.      * @return 数据源
22.      */
23.     @Bean(name = "masterDataSource")
24.     @ConfigurationProperties(prefix = "spring.datasource.druid.master" )
25.     public DataSource masterDataSource() {
26.         DruidDataSource druidDataSource = new DruidDataSource();
27.         parseDruidConfig(druidDataSource);
28.         return druidDataSource;
29.     }
31.     /**
32.      * 配置从数据源
33.      *
34.      * @return 数据源
35.      */
36.     @Bean(name = "slaveDataSource")
37.     @ConfigurationProperties(prefix = "spring.datasource.druid.slave")
38.     public DataSource slaveDataSource() {
39.         DruidDataSource druidDataSource = new DruidDataSource();
40.         parseDruidConfig(druidDataSource);
41.         return druidDataSource;
42.     }
44.     private void parseDruidConfig(DruidDataSource dataSource) {
45.         dataSource.setInitialSize(initialSize);
46.         dataSource.setMaxActive(maxActive);
47.         dataSource.setMinIdle(minIdle);
48.         dataSource.setMaxWait(maxWait);
49.         dataSource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
50.         dataSource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
51.         dataSource.setTestWhileIdle(testWhileIdle);
52.     }
54.     /**
55.      * 配置路由数据源
56.      *
57.      * @param masterDataSource 主节点
58.      * @param slaveDataSource 从节点
59.      * @return 数据源
60.      */
61.     @Bean(name = "myRoutingDataSource")
62.     @DependsOn({"masterDataSource", "slaveDataSource"})
63.     @Primary
64.     public DataSource myRoutingDataSource(@Qualifier("masterDataSource") DataSource masterDataSource,
65.                                           @Qualifier("slaveDataSource") DataSource slaveDataSource) {
66.         Map<Object, Object> targetDataSources = new HashMap<>(3);
67.         targetDataSources.put(DBTypeEnum.MASTER, masterDataSource);
68.         targetDataSources.put(DBTypeEnum.SLAVE, slaveDataSource);
69.         MyRoutingDataSource myRoutingDataSource = new MyRoutingDataSource();
70.         //设置默认数据源
71.         myRoutingDataSource.setDefaultTargetDataSource(masterDataSource);
72.         myRoutingDataSource.setTargetDataSources(targetDataSources);
73.         return myRoutingDataSource;
74.     }
75. }

复制代码

3、线程轮循切换主从数据库(多主多从的情况下适用)

1. /**
2. * 通过ThreadLocal将数据源设置到每个线程上下文中
3. */
4. public class DataSourceContextHolder {
5.     private static final ThreadLocal<DBTypeEnum> CONTEXT_HOLDER = new ThreadLocal<>();
7.     private static final AtomicInteger COUNTER = new AtomicInteger(-1);
9.     public static void set(DBTypeEnum dbType) {
10.         CONTEXT_HOLDER.set(dbType);
11.     }
13.     public static DBTypeEnum get() {
14.         return CONTEXT_HOLDER.get();
15.     }
17.     public static void clear(){
18.         CONTEXT_HOLDER.remove();
19.     }
21.     public static void master() {
22.         set(DBTypeEnum.MASTER);
23.         System.out.println("切换到master");
24.     }
26.     public static void slave() {
27.         //  轮询
28.         int index = COUNTER.getAndIncrement() % 2;
29.         if (COUNTER.get() > 9999) {
30.             COUNTER.set(-1);
31.         }
32. //        if (index == 0) {
33. //            set(DBTypeEnum.SLAVE1);
34. //            System.out.println("切换到slave1");
35. //        } else {
36. //            set(DBTypeEnum.SLAVE2);
37. //            System.out.println("切换到slave2");
38. //        }
39.         set(DBTypeEnum.SLAVE);
40.         System.out.println("切换到slave2");
41.     }
42. }

复制代码

4、声明路由数据源key(多主多从的情况下适用)

1. /**
2. * 声明路由数据源key
3. */
4. public class MyRoutingDataSource extends AbstractRoutingDataSource {
5.     @Nullable
6.     @Override
7.     protected Object determineCurrentLookupKey() {
8.         return DataSourceContextHolder.get();
9.     }
10. }

复制代码

5、强制使用主/从数据库注解

1. /**
2. * 强制使用主数据库注解
3. */
4. @Retention(RetentionPolicy.RUNTIME)
5. @Target({ElementType.METHOD})
6. public @interface DataSourceMaster {
7. }

复制代码

1. /**
2. * 强制使用从数据库注解
3. */
4. @Retention(RetentionPolicy.RUNTIME)
5. @Target({ElementType.METHOD})
6. public @interface DataSourceSlave {
7. }

复制代码

6、切面类(从库注解会报错，不知道什么原因)

1. /**
2. * 使用aop实现数据源切换
3. */
4. @Aspect
5. @Component
6. public class DataSourceAop {
8. //    /**
9. //     * 需要读的方法,切面
10. //     */
11. //    @Pointcut("!@annotation(com.readWriteSeparation.annotation.DataSourceSlave)" +
12. //            "|| execution(* com.readWriteSeparation.service..*.select*(..)) " +
13. //            "|| execution(* com.readWriteSeparation.service..*.get*(..))" +
14. //            "|| execution(* com.readWriteSeparation.service..*.query*(..))" +
15. //            "|| execution(* com.readWriteSeparation.service..*.find*(..)))")
16. //    public void readPointcut() { }
18.     /**
19.      * 需要读的方法,切面
20.      */
21.     @Pointcut("execution(* com.readWriteSeparation.service..*.select*(..)) " +
22.             "|| execution(* com.readWriteSeparation.service..*.get*(..))" +
23.             "|| execution(* com.readWriteSeparation.service..*.query*(..))" +
24.             "|| execution(* com.readWriteSeparation.service..*.find*(..)))")
25.     public void readPointcut() { }
27.     /**
28.      * 写切面
29.      */
30.     @Pointcut("@annotation(com.readWriteSeparation.annotation.DataSourceMaster) " +
31.             "|| execution(* com.readWriteSeparation.service..*.insert*(..))" +
32.             "|| execution(* com.readWriteSeparation.service..*.save*(..))" +
33.             "|| execution(* com.readWriteSeparation.service..*.add*(..))" +
34.             "|| execution(* com.readWriteSeparation.service..*.update*(..))" +
35.             "|| execution(* com.readWriteSeparation.service..*.edit*(..))" +
36.             "|| execution(* com.readWriteSeparation.service..*.delete*(..))" +
37.             "|| execution(* com.readWriteSeparation.service..*.remove*(..))")
38.     public void writePointcut() { }
40.     @Before("readPointcut()")
41.     public void read() {
42.         DataSourceContextHolder.slave();
43.     }
45.     @Before("writePointcut()")
46.     public void write() {
47.         DataSourceContextHolder.master();
48.     }
50.     @After("readPointcut()")
51.     public void readAfter() {
52.         DataSourceContextHolder.clear();
53.     }
55.     @After("writePointcut()")
56.     public void writeAfter() {
57.         DataSourceContextHolder.clear();
58.     }
59. }

复制代码

7、.yml文件配置

1. spring:
2.   datasource:
3.     druid:
4.       master:
5.         username: root
6.         password: 12345678
7.         url: jdbc:mysql://192.168.10.15/zeroStart?characterEncoding=UTF-8&useSSL=true&requireSSL=false&zeroDateTimeBehavior=convertToNull&useSSL=false&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=Asia/Shanghai&allowMultiQueries=true
8.         driver-class-name: com.mysql.cj.jdbc.Driver
9.         type: com.zaxxer.hikari.HikariDataSource
10.       slave:
11.         username: root
12.         password: 12345678
13.         url: jdbc:mysql://localhost:3306/zeroStart?characterEncoding=UTF-8&useSSL=true&requireSSL=false&zeroDateTimeBehavior=convertToNull&useSSL=false&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=Asia/Shanghai&allowMultiQueries=true
14.         driver-class-name: com.mysql.cj.jdbc.Driver
15.         type: com.zaxxer.hikari.HikariDataSource

复制代码

8、引入的依赖

1.         <dependency>
2.             <groupId>com.baomidou</groupId>
3.             <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
4.             <version>2.4.2</version>
5.         </dependency>
6.         <dependency>
7.             <groupId>com.baomidou</groupId>
8.             <artifactId>mybatis-plus-extension</artifactId>
9.             <version>${mybatis-plus.version}</version>
10.         </dependency>
11.         
12.         <dependency>
13.             <groupId>com.alibaba</groupId>
14.             <artifactId>druid</artifactId>
15.             <version>1.2.7</version>
16.         </dependency>
18.         <dependency>
19.             <groupId>com.alibaba</groupId>
20.             <artifactId>druid-spring-boot-starter</artifactId>
21.             <version>1.2.16</version>
22.         </dependency>

复制代码

配置到这就好了，业务代码就不展示了
 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！