一、定义我们自己的切库注解类

自定义注解有几点需要注意:

1)@Target 是作用的目标,接口、方法、类、字段、包等等,具体看:ElementType

2)@Retention 是注解存在的范围,RUNTIME代表的是注解会在class字节码文件中存在,在运行时可以通过反射获取到,具体看:RetentionPolicy

3)允许的变量,通常都要给定默认值,比如我们使用一个service时,可以@Service,也可以

@Service("xxxx")
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD
})
public @interface RoutingDataSource {String value() default DataSources.MASTER_DB;
}

二、定义需要使用的数据库及配置

1、数据库配置:application.properties,这里要注意不同db的前缀区别

## datasource master #
spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.driver-class-name=com.mysql.jdbc.Driver
spring.datasource.url=jdbc:mysql://localhost:3306/master?characterEncoding=UTF-8
spring.datasource.username=root
spring.datasource.password=466420182## datasource slave #
spring.datasourceSlave.type=com.alibaba.druid.pool.DruidDataSource
spring.datasourceSlave.driver-class-name=com.mysql.jdbc.Driver
spring.datasourceSlave.url=jdbc:mysql://localhost:3306/slave?characterEncoding=UTF-8
spring.datasourceSlave.username=root
spring.datasourceSlave.password=466420182

2、定义支持的数据源id

public interface DataSources {String MASTER_DB = "masterDB";String SLAVE_DB = "slaveDB";
}

3、定义数据库实体类并配置为多数据源的形式

这里不要忽略了通过 MapperScan 指定需要扫描的mybatis的接口类

@Configuration
public class DatasourceConfig {//destroy-method="close"的作用是当数据库连接不使用的时候,就把该连接重新放到数据池中,方便下次使用调用.@Bean(destroyMethod =  "close", name = DataSources.MASTER_DB)@ConfigurationProperties(prefix = "spring.datasource")public DataSource dataSource() {return DataSourceBuilder.create().type(DruidDataSource.class).build();}@Bean(destroyMethod =  "close", name = DataSources.SLAVE_DB)@ConfigurationProperties(prefix = "spring.datasourceSlave")public DataSource dataSourceSlave() {return DataSourceBuilder.create().type(DruidDataSource.class).build();}
}

4、配置成动态数据源

@Configuration
@MapperScan(basePackages = {"com.xxx.dao"})  // 这里需要替换为实际的路径
public class MybatisConfig {@Autowired@Qualifier(Datasources.MASTER_DB)private DataSource masterDB;@Autowired@Qualifier(DataSources.SLAVE_DB)private DataSource slaveDB;/*** 动态数据源*/@Bean(name = "dynamicDataSource")public DataSource dynamicDataSource() {DynamicDataSource dynamicDataSource = new DynamicDataSource();// 默认数据源
        dynamicDataSource.setDefaultTargetDataSource(masterDB);// 配置多数据源Map<Object, Object> dsMap = Maps.newHashMap();dsMap.put(DataSources.MASTER_DB, masterDB);dsMap.put(DataSources.SLAVE_DB, slaveDB);dynamicDataSource.setTargetDataSources(dsMap);return dynamicDataSource;}@Bean@ConfigurationProperties(prefix = "mybatis")public SqlSessionFactoryBean sqlSessionFactoryBean() {SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();// 配置数据源,此处配置为关键配置,如果没有将 dynamicDataSource 作为数据源则不能实现切换
        sqlSessionFactoryBean.setDataSource(dynamicDataSource());return sqlSessionFactoryBean;}
}

三、使用ThreadLocal安全的管理当前进程使用的数据源连接

@Slf4j
public class DataSourceContextHolder {/*** 默认数据源*/public static final String DEFAULT_DATASOURCE = DataSources.MASTER_DB;private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();// 设置数据源名public static void setDB(String dbType) {log.debug("切换到{}数据源", dbType);contextHolder.set(dbType);}// 获取数据源名public static String getDB() {return (contextHolder.get());}// 清除数据源名public static void clearDB() {contextHolder.remove();}
}

四、通过编写切面,对所有我们自定义切库注解的方法进行拦截,动态的选择数据源

这里是为下一步提供铺垫,动态调整DataSourceContextHolder里存储的值,使用threadLocal来管理是为了避免多线程之间互相影响。

自定义注解,核心的处理就是写处理这个注解的逻辑,然后通过指定的拦截方案根据当前的数据做一些动态的处理。比如Spring提供的@Controller、@Service等注解,都是需要我们在配置文件里配置好需要扫描的路径,然后项目启动时,spring根据配置去指定路径读取这些配置,然后这些类才可以被spring进行管理。

这里不要忽略了默认数据源要选择主库,如果切库出现什么问题,比如配置错误等,可以保证访问主库来得到正确的结果;另外,请求完了不要忘记调用提供的clearDB的操作,防止threadLocal误用带来的内存泄露。

@Aspect
@Component
@Slf4j
public class DynamicDataSourceAspect {@Before("@annotation(RoutingDataSource)")public void beforeSwitchDS(JoinPoint point){//获得当前访问的classClass<?> className = point.getTarget().getClass();//获得访问的方法名String methodName = point.getSignature().getName();//得到方法的参数的类型Class[] argClass = ((MethodSignature)point.getSignature()).getParameterTypes();String dataSource = DataSourceContextHolder.DEFAULT_DATASOURCE;try {// 得到访问的方法对象Method method = className.getMethod(methodName, argClass);// 判断是否存在@DS注解if (method.isAnnotationPresent(RoutingDataSource.class)) {RoutingDataSource annotation = method.getAnnotation(RoutingDataSource.class);// 取出注解中的数据源名dataSource = annotation.value();}} catch (Exception e) {log.error("routing datasource exception, " + methodName, e);}// 切换数据源
        DataSourceContextHolder.setDB(dataSource);}@After("@annotation(RoutingDataSource)")public void afterSwitchDS(JoinPoint point){DataSourceContextHolder.clearDB();}
}

五、动态的取出我们在切面里设置的数据源的字符串即可

这里需要把原理介绍一下,在连接数据库时其实是先选择一个配置好的spring管理的datasource的id,就是我们之前在 DatasourceConfig 类里定义的Datasource实体类的id:masterDB 和 slaveDB。然后根据id去spring的上下文选择配置,进行数据库连接。有兴趣的可以看一下源码。
@Slf4j
public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() {log.debug("数据源为{}", DataSourceContextHolder.getDB());return DataSourceContextHolder.getDB();}
}

六、取消自动配置数据源,使用我们这里定义的数据源配置

在SpringBoot启动类上通常直接使用@SpringBootApplication就可以了,这里需要调整为:

@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class
})

其实这个配置就是告诉springboot不使用默认的数据源配置。

七、使用

@Service
public class DataSourceRoutingService {@Resourceprivate SysUserMapper sysUserMapper;@RoutingDataSource(DataSources.MASTER_DB) // 这个注解这时是可以省略,因为默认就是访问主库public SysUser test1(int id) {return sysUserMapper.selectByPrimaryKey(id);}@RoutingDataSource(DataSources.SLAVE_DB)public SysUser test2(int id) {return sysUserMapper.selectByPrimaryKey(id);}
}

如此,数据库切库就OK了。如果你的系统已经有主库、从库之分了,那么赶紧在你的系统里利用起来吧。

八、扩展

这里呢,还可以支持多个扩展。比如现在一个主库后面有多个从库,在切面拿到需要切换从库时,还可以选择随机选择一个,或者根据类名、方法名或业务配置等选择某一个从库,这样不但可以分担每个从库的压力,也可以有针对性的让指定的读请求打到指定的从库上。如果有多个主库,也可以有更多的选择~

SpringBoot+SpringAOP+Java自定义注解+mybatis实现切库读写分离相关推荐

  1. 自定义注解完成数据库切库

      为什么会有数据库切库一说 首先,许多项目都有主库与从库,有的主库后面甚至会有很多个从库,主从库之间的通常同步也很快,这为数据库切库提供了一个基础,因为可以去不同的数据库查询,得到相同的结果(如果不 ...

  2. springboot项目中自定义注解的使用总结、java自定义注解实战(常用注解DEMO)

    初学spring的时候使用注解总觉得使用注解很神奇,加一个注解就能实现想要的功能,很好奇,也想自己根据需要写一些自己实现的自定义注解.问题来了,自定义注解到底是什么?肯定会有人和我一样有这个疑惑,我根 ...

  3. Java 自定义注解及使用场景

    文章目录 1 什么是注解(Annotation) 2 元注解 2.1 @Target 2.2 @Retention 2.3 @Repeatable 2.4 @Inherited 示例1 反射获取注解 ...

  4. java 自定义注解_两步实现Java自定义注解

    什么是注解? 注解就是为容器提供元数据,例如@Controller 注解则是标记了该Bean需要交给Spring容器进行管理. 那么我们怎么去实现自个的注解,也就是自定义注解呢? 一.自定义一个注解( ...

  5. java自定义注解实现前后台参数校验

    其实是可以通过@Constraint来限定自定义注解的方法. @Constraint(validatedBy = xxxx.class) 下面是我做的 java自定义注解实现前后台参数校验 的代码示例 ...

  6. java自定义注解类

    一.前言 今天阅读帆哥代码的时候,看到了之前没有见过的新东西, 比如java自定义注解类,如何获取注解,如何反射内部类,this$0是什么意思? 于是乎,学习并整理了一下. 二.代码示例 import ...

  7. java 自定义注解实现不同对象之间的拷贝(支持大小写、驼峰转换)

    java 自定义注解实现不同对象之间的拷贝(支持大小写.驼峰转换) 1.需求 要实现两个属性名称完全不同的对象之间的拷贝.具体如下 原对象 /*** 原实体*/ @Data public class ...

  8. java中注解动态传参_Java自定义注解源码+原理解释(使用Java自定义注解校验bean传入参数合法性)...

    Java自定义注解源码+原理解释(使用Java自定义注解校验bean传入参数合法性)java 前言:因为前段时间忙于写接口,在接口中须要作不少的参数校验,本着简洁.高效的原则,便写了这个小工具供本身使 ...

  9. Java 自定义注解简单使用

    Java 自定义注解简单使用 文章目录 Java 自定义注解简单使用 第一步:在 pom.xml 下引入反射框架 第二步:自定义注解 第三步:写两个测试实体类 第四步:写个工具类扫描被注解的类(核心) ...

  10. java 自定义注解 教程_Java注解教程及自定义注解

    Java注解提供了关于代码的一些信息,但并不直接作用于它所注解的代码内容.在这个教程当中,我们将学习Java的注解,如何定制注解,注解的使用以及如何通过反射解析注解. Java1.5引入了注解,当前许 ...

最新文章

  1. pythonassertbug_还在 Bug 不断?不妨试试这 2 个装X技巧
  2. 链表问题(6)-----排序
  3. [译]ChipMunk 教程1 - 设置
  4. 只有房子没有工作能贷款吗?
  5. Autodesk 360 Mobile不能显示图片?
  6. 多重 for 循环,如何提高效率?
  7. 必备天气预报界面APP应用设计灵感,出门瞅一瞅~
  8. 我是如何在 10 分钟内搞砸 IT 面试的
  9. 遍历字典时用与不用iter的区别
  10. 使用寻路算法写的一个小项目
  11. grep和正则表达式
  12. 将文本写在图片上,自定义字体,自动换行,自定义行间距
  13. 最最最最最最最最基础的C---函数
  14. Linux下PHP+MySQL+CoreSeek中文检索引擎配置
  15. win10 网卡驱动正常搜索不到WiFi信号 解决方法
  16. urp教务系统php,URP高校教务管理系统
  17. pdfbox、icepdf工具实现pdf转图片
  18. 转:淘宝客搜索链接技巧首度分享
  19. 408数据结构考研笔记!超级详细!23最新考纲
  20. 苹果x微信为什么不出定位服务器,苹果x微信发动态为什么显示不了位置

热门文章

  1. 【图像去噪】基于matlab中值+均值+高斯+Laplacian+Sobel+Prewitt图像去噪【含Matlab源码 025期】
  2. x射线计算机断层成像_医疗保健中的深度学习-X射线成像(第4部分-类不平衡问题)...
  3. android 结算权限,如何在Android应用程序中实施应用内结算?
  4. sql与mysql数据对接_SQL Server 对接MySQL 数据库
  5. vfp生成菜单时文件不存在_手把手教你搞定权限管理,结合Vue实现菜单的动态权限控制!...
  6. 常州工学院计算机网络考试题库,用Winsock实现在线考试
  7. layui多文件上传讲解_layui文件上传的实际应用实例
  8. 高并发大流量专题---1、高并发大流量解决方案总结
  9. HDU4825/5536 [01 字典树/简单字典树更新]
  10. java volatile 和Transient 关键字