当前位置: 首页 > news >正文

如何设计一个@TenantIgnore功能,使得被该注解注解的方法可以使用户绕过自己的租户id查询全局信息(所有租户的信息)

什么是@TelnantIgnore

有些系统设计时,将不同公司(租户)的数据放在同一个数据库(表)中,因此会员A在查询数据时,应只查询本人所在公司的数据,因此会在某些表每个执行的sql语句中默认加上"where tenantid = ?"来限制用户的行为。但是也有一些全局的查询,要查询所有公司的所有数据,因此需要在该Controller的方法上加上@TelnantIgnore,使该方法绕过加"where tenantid = ?"这个步骤。


如何做

这是一个非常经典的AOP + ThreadLocal + ORM 插件协同工作的架构设计案例。

如果你要从零设计一个@TenantIgnore功能,让被注解的方法能够绕过租户限制查询全局数据,你需要构建一条完整的指令传输链

我们可以把这个过程分为4 个核心步骤,就像设计一套“免检通行”系统:


第一步:制作通行证(定义注解)

首先,你需要定义一个注解,作为一个“标记”。

// 1. 定义注解 @TenantIgnore@Target({ElementType.METHOD,ElementType.TYPE})// 既能加在方法上,也能加在类上@Retention(RetentionPolicy.RUNTIME)// 运行时必须存在public@interfaceTenantIgnore{// 可以扩展属性,比如 boolean value() default true;}

第二步:准备口袋(定义上下文 Holder)

你需要一个地方来存放“是否免检”这个状态。因为 Web 请求是基于线程的,所以必须用ThreadLocal

// 2. 定义上下文容器 TenantContextHolderpublicclassTenantContextHolder{// 使用 ThreadLocal 保证线程隔离privatestaticfinalThreadLocal<Boolean>IGNORE=newThreadLocal<>();// 设置状态publicstaticvoidsetIgnore(Booleanignore){IGNORE.set(ignore);}// 获取状态(供 MyBatis 插件读取)publicstaticbooleanisIgnore(){returnBoolean.TRUE.equals(IGNORE.get());}// 清理(防止内存泄漏和线程污染)publicstaticvoidclear(){IGNORE.remove();}}

第三步:门口安检员(编写 AOP 切面)

这是核心逻辑。你需要拦截所有贴了@TenantIgnore的方法,在方法执行把“免检卡”塞进口袋,执行把卡拿出来。

// 3. 定义切面 TenantIgnoreAspect@Aspect@ComponentpublicclassTenantIgnoreAspect{@Around("@annotation(tenantIgnore)")// 拦截所有带 @TenantIgnore 的方法publicObjectaround(ProceedingJoinPointjoinPoint,TenantIgnoretenantIgnore)throwsThrowable{// A. 存档:先记住进门前的状态(防止嵌套调用时,把外层的状态搞丢)booleanoldIgnore=TenantContextHolder.isIgnore();try{// B. 塞卡:标记当前线程为“忽略租户”TenantContextHolder.setIgnore(true);// C. 放行:执行真正的业务逻辑(查询数据库)returnjoinPoint.proceed();}finally{// D. 复原:无论成功失败,必须把状态恢复成进门前的样子// 这一步至关重要!否则线程池复用会导致下一个请求数据泄露!TenantContextHolder.setIgnore(oldIgnore);}}}

第四步:修改数据库规则(配置 MyBatis Plus Handler)

最后,你需要告诉 MyBatis Plus:“生成 SQL 之前,先去口袋里摸一下,看看有没有免检卡。”

// 4. 配置 MyBatis Plus 的多租户拦截器@ComponentpublicclassMyTenantLineHandlerimplementsTenantLineHandler{@OverridepublicExpressiongetTenantId(){// 返回当前用户的租户ID,例如 1returnnewLongValue(UserContext.getTenantId());}@OverridepublicStringgetTenantIdColumn(){return"tenant_id";// 数据库里的列名}// 【核心】决定是否要忽略表(不加 WHERE tenant_id = ?)@OverridepublicbooleanignoreTable(StringtableName){// 1. 系统表(如字典、菜单)默认忽略if("sys_dict".equals(tableName)){returntrue;}// 2. 【关键】去口袋(ThreadLocal)里查状态// 如果 Aspect 设置了 true,这里就返回 true -> MyBatis Plus 就不会拼接租户条件了if(TenantContextHolder.isIgnore()){returntrue;}// 3. 默认不忽略returnfalse;}}

🌟 总结:完整的执行流

当你写下这段代码时:

@TenantIgnore// 1. 贴上标记publicList<User>getAllUsers(){returnuserMapper.selectList(null);// 2. 查库}

系统底层发生了什么:

  1. AOP 介入:Aspect 看到注解,调用TenantContextHolder.setIgnore(true)。此时当前线程的ThreadLocal变为true
  2. 业务执行:代码跑到userMapper.selectList
  3. MP 拦截:MyBatis Plus 准备生成 SQL,问MyTenantLineHandler:“我要不要加租户条件?”
  4. Handler 检查:Handler 调用TenantContextHolder.isIgnore(),发现是true
  5. SQL 生成:Handler 说“不用加”。MP 生成SELECT * FROM user(全表查询)。
  6. AOP 收尾:方法返回,Aspect 在finally块中清理ThreadLocal

这就是通过注解驱动实现动态数据权限控制的标准架构设计。

http://www.jsqmd.com/news/103145/

相关文章:

  • EmotiVoice模型微调指南:针对特定领域优化语音表现
  • 【心电图信号处理】心电图信号处理(含基础波形检测、信号去噪、信号重建指标)【含Matlab源码 14715期】
  • windows的任务管理器中如何查看与硬盘相关的指标?
  • EmotiVoice语音合成服务高并发架构设计
  • 【情绪识别】基于matlab心率变异性信号的持续情绪识别方法【含Matlab源码 14718期】
  • 【情绪识别】心率变异性信号的持续情绪识别方法【含Matlab源码 14718期】
  • 如何训练自定义情感模型以扩展EmotiVoice能力?
  • 【预编码】基于matlab深度学习的带有有限字母表信令MIMO通道线性预编码【含Matlab源码 14717期】
  • 从零到一:构建智能红蓝对抗训练可视化系统——一场技术与战术的完美融合
  • EmotiVoice语音合成质量评估标准与优化建议
  • 任务管理器和资源管理器的关系和区别
  • 6、自旋、量子比特与纠缠:量子世界的奇妙之旅
  • 43、Red Hat系统安全维护与监控指南
  • EmotiVoice能否替代专业配音演员?我们做了测试
  • 1.3 衡量AI的标准:图灵测试、中文房间与认知科学的视角
  • EmotiVoice语音合成缓存机制设计提升性能
  • 37、深入探究:Linux内核的重建与优化
  • 1、量子计算基础:从比特到量子比特
  • EmotiVoice语音合成与ASR系统的闭环测试实践
  • 38、Linux 系统管理与 Perl 编程全解析
  • HTTP的数据报格式
  • 2、量子世界的奥秘:电子自旋与光子偏振
  • 18、量子认知、理性与共同知识:经典与量子视角的探讨
  • 33、邮件操作、DHCP服务器搭建与系统管理
  • EmotiVoice语音合成在影视后期制作中的辅助作用
  • 开发者必看:EmotiVoice API接口调用详细说明
  • 35、系统日志管理与数据备份全攻略
  • 20、量子音乐、量子艺术及其感知
  • EmotiVoice语音合成在语音翻译软件中的情感保留能力探究
  • EmotiVoice语音合成在心理治疗音频内容生成中的价值