SpringBoot+Mybatis-Plus实现mysql读写分离方案的示例代码

时间:2021-09-28 15:47:37

1. 引入mybatis-plus相关包,pom.xml文件

SpringBoot+Mybatis-Plus实现mysql读写分离方案的示例代码

2. 配置文件application.property增加多库配置

mysql 数据源配置

  1. spring.datasource.primary.jdbc-url=jdbc:mysql://xx.xx.xx.xx:3306/portal?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=UTC&characterEncoding=utf8&serverTimezone=GMT%2B8
  2. spring.datasource.primary.username=root
  3. spring.datasource.primary.password=root
  4. spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver
  5. #mysql slave 数据源配置
  6.  
  7. spring.datasource.slave.jdbc-url=jdbc:mysql://xx.xx.xx.xx:3306/portal?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=UTC&characterEncoding=utf8&serverTimezone=GMT%2B8
  8. spring.datasource.slave.username=root
  9. spring.datasource.slave.password=root
  10. spring.datasource.slave.driver-class-name=com.mysql.cj.jdbc.Driver

3. 配置数据源及注解

数据源配置 MultiDataSourceConfig.Java

  1. /**
  2. * 配置多数据源
  3. */
  4. @Profile("dev")//开发模式配置文件
  5. @Configuration
  6. @MapperScan(basePackages = "com.csc.portal.mapper")//扫描包
  7. public class MultiDataSourceConfig {
  8.  
  9. /**
  10. * 主数据源
  11. * @return
  12. */
  13. @Bean
  14. @ConfigurationProperties(prefix = "spring.datasource.primary")
  15. public DataSource masterDataSource() {
  16. return DataSourceBuilder.create().build();
  17. }
  18.  
  19. /**
  20. * 从数据源
  21. * @return
  22. */
  23. @Bean
  24. @ConfigurationProperties(prefix = "spring.datasource.slave")
  25. public DataSource slaveDataSource() {
  26. return DataSourceBuilder.create().build();
  27. }
  28.  
  29. /**
  30. * 路由数据源,前面两个数据源是为了创建此数据源
  31. * @param masterDataSource 主数据源
  32. * @param slaveDataSource 从数据源
  33. * @return
  34. */
  35. @Bean
  36. public DataSource myRoutingDataSource(@Qualifier("masterDataSource") DataSource masterDataSource,
  37. @Qualifier("slaveDataSource") DataSource slaveDataSource) {
  38. Map<Object, Object> targetDataSources = new HashMap<>();
  39. targetDataSources.put(DBTypeEnum.MASTER, masterDataSource);
  40. targetDataSources.put(DBTypeEnum.SLAVE, slaveDataSource);
  41. MyRoutingDataSource myRoutingDataSource = new MyRoutingDataSource();
  42. myRoutingDataSource.setDefaultTargetDataSource(slaveDataSource);//设置默认数据源
  43. myRoutingDataSource.setTargetDataSources(targetDataSources);//设置路由表,使用map的key,value方式得到对应数据源
  44. return myRoutingDataSource;
  45. }

数据库枚举类

  1. public enum DBTypeEnum {
  2. MASTER, SLAVE;
  3. }

注解

  1. @Target({ElementType.METHOD})
  2. @Retention(RetentionPolicy.RUNTIME)
  3. public @interface Master {
  4. }
  1. @Target({ElementType.METHOD})
  2. @Retention(RetentionPolicy.RUNTIME)
  3. public @interface Slave {
  4. }

4. Mybatis-plus配置

  1. @EnableTransactionManagement
  2. @Configuration
  3. @MapperScan(basePackages = "com.csc.portal.mapper")
  4. public class MybatisPlusConfig {
  5. /**
  6. * 分页插件
  7. */
  8. @Bean
  9. public PaginationInterceptor paginationInterceptor() {
  10. return new PaginationInterceptor();
  11. }
  12.  
  13. @Resource(name = "myRoutingDataSource")
  14. private DataSource myRoutingDataSource;
  15. /**
  16. * 使用MyBatis Plus的sqlSessionFactory代替,
  17. * 此处注意mybatis与mybatisPlus的配置不同,不然扫描不到对数据操作的方法。会报未绑定错误
  18. * @return sqlSessionFactory
  19. * @throws Exception
  20. */
  21. @Bean
  22. public SqlSessionFactory sqlSessionFactory() throws Exception {
  23. MybatisSqlSessionFactoryBean sqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
  24. sqlSessionFactoryBean.setDataSource(myRoutingDataSource);
  25. sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/*.xml"));
  26. MybatisConfiguration mybatisConfiguration = new MybatisConfiguration();
  27. sqlSessionFactoryBean.setConfiguration(mybatisConfiguration);
  28. return sqlSessionFactoryBean.getObject();
  29. }
  30.  
  31. /**
  32. * 此处为使用mybatis时的sqlsessionFactory配置
  33. * @return
  34. * @throws Exception
  35. */
  36. /*
  37. @Bean
  38. public SqlSessionFactory sqlSessionFactory() throws Exception {
  39. SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
  40. sqlSessionFactoryBean.setDataSource(myRoutingDataSource);
  41. sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mapper/*.xml"));
  42. return sqlSessionFactoryBean.getObject();
  43. }
  44. */
  45.  
  46. /**
  47. * 事务配置
  48. * @return 事务管理器
  49. */
  50. @Bean
  51. public DataSourceTransactionManager transactionManager() {
  52. DataSourceTransactionManager tx = new DataSourceTransactionManager();
  53. tx.setDataSource(myRoutingDataSource);
  54. return tx;
  55. }

5. 增加数据源管理类

DBContextHolder.java

  1. public class DBContextHolder {
  2.  
  3. /**
  4. * 外部一个请求将会产生一个线程与之对应,每个线程的变量可用ThreadLocal进行存储
  5. */
  6. private static final ThreadLocal<DBTypeEnum> contextHolder = new ThreadLocal<>();
  7.  
  8. public static void set(DBTypeEnum dbType) {
  9. contextHolder.set(dbType);
  10. }
  11.  
  12. public static DBTypeEnum get() {
  13. return contextHolder.get();
  14. }
  15.  
  16. public static void master() {
  17. set(DBTypeEnum.MASTER);
  18. System.out.println("切换到master");
  19. }
  20.  
  21. public static void slave() {
  22. set(DBTypeEnum.SLAVE);
  23. System.out.println("切换到slave");
  24. }
  25.  
  26. }

指定选择数据源

MyRoutingDataSource.java 方法determineCurrentLookupKey决定最终使用哪个数据源进行操作,若为空则使用默认数据源。

  1. public class MyRoutingDataSource extends AbstractRoutingDataSource {
  2. @Nullable
  3. @Override
  4. protected Object determineCurrentLookupKey() {
  5. System.out.println("线程名:"+Thread.currentThread().getName()+":"+DBContextHolder.get());
  6. return DBContextHolder.get();
  7. /* if (DBContextHolder.get() != null) {
  8. System.out.println("线程名:"+Thread.currentThread().getName()+":"+DBContextHolder.get());
  9. return DBContextHolder.get();
  10. } else {
  11. System.out.println("未匹配到指定数据库,默认切换到Master");
  12. return DBTypeEnum.MASTER;
  13. }*/
  14. //return DBContextHolder.get();
  15. }
  16.  
  17. }

6. 增加aop切面

  1. @Aspect
  2. @Component
  3. @Order(0)//配置注解优先级,优于事物注解@Transactional先进行数据源切换,
  4. //不然在事物中进行数据源切换无效
  5. public class DataSourceAop {
  6.  
  7. @Pointcut(/*"!@annotation(com.csc.portal.annotation.Master) " +
  8. "&& (execution(* com.csc.portal.service..*.select*(..)) " +
  9. "|| execution(* com.csc.portal.service..*.get*(..))"+*/
  10. " @annotation(com.csc.portal.annotation.Slave)")
  11. public void readPointcut() {
  12.  
  13. }
  14.  
  15. @Pointcut("@annotation(com.csc.portal.annotation.Master) " //+
  16. /* "|| execution(* com.csc.portal.service..*.insert*(..)) " +
  17. "|| execution(* com.csc.portal.service..*.add*(..)) " +
  18. "|| execution(* com.csc.portal.service..*.update*(..)) " +
  19. "|| execution(* com.csc.portal.service..*.edit*(..)) " +
  20. "|| execution(* com.csc.portal.service..*.delete*(..)) " +
  21. "|| execution(* com.csc.portal.service..*.remove*(..))"*/)
  22. public void writePointcut() {
  23.  
  24. }
  25.  
  26. @Before("readPointcut()")
  27. public void read() {
  28. //获取拦截类
  29. DBContextHolder.slave();
  30. System.out.println(Thread.currentThread().getName()+DBContextHolder.get());
  31. }
  32.  
  33. @Before("writePointcut()")
  34. public void write() {
  35. //获取拦截类
  36. /* String className = pjp.getTarget().getClass().getName();
  37. System.out.println("当前线程"+Thread.currentThread().getName()+" 拦截类为:" + className);
  38.  
  39. //获取拦截的方法名
  40. MethodSignature msig = (MethodSignature) pjp.getSignature();
  41. Method currentMethod = null;
  42. try {
  43. currentMethod = pjp.getTarget().getClass().getMethod(msig.getName(), msig.getParameterTypes());
  44. } catch (NoSuchMethodException e) {
  45. e.printStackTrace();
  46. }
  47. String methodName = currentMethod.getName();
  48. System.out.println("拦截方法名为:" + methodName);*/
  49. DBContextHolder.master();
  50. System.out.println(Thread.currentThread().getName()+DBContextHolder.get());
  51. }
  52. }

6. 实际应用

  1. 在service层方法前增加注解@Master表示使用主库,进行增删改的操作使用主库。
  2. 在service层方法前增加注解@Slave表示使用从库,进行查的操作使用从库,默认使用从库,可不配置。
  3. @ Transactional注解加到service层,增加了@Transactional注解后,启用事务后,一个事务内部的connection是复用的,所以就算AOP切了数据源字符串,但是数据源并不会被真正修改。所以@Transactional注解不要写在controller层,不然在service层也切换不了数据源。
  4. @Transactional与@Master可同时使用,已经配置@Master注解的优先级较高,先切换数据源后执行事务。

到此这篇关于SpringBoot+Mybatis-Plus实现mysql读写分离方案的示例代码的文章就介绍到这了,更多相关SpringBoot Mybatis-Plus mysql读写分离内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://blog.csdn.net/AAAAABBBBBYYYYY/article/details/110875245