Mybatis-Plus 租户使用

Mybatis-Plus 租户使用

文章目录

  • Mybatis-Plus 租户使用
  • 一. 前言
    • 1.1 租户存在的意义
    • 1.2 租户框架
  • 二. Mybatis-plus 租户
    • 2.1 租户处理器
    • 2.2 前置准备
      • 1. 依赖
      • 2. 表及数据准备
      • 3. 代码生成器
    • 2.3 使用
  • 三. 深入使用
    • 3.1 前言
    • 3.2 租户主体设值,取值
    • 3.3 部分表全量db操作
    • 3.4 全量db操作
    • 3.5 自由控制全量/租户对db操作
    • 3.6 feign传递租户
  • 四. 源码地址

一. 前言

1.1 租户存在的意义

多租户数据隔离与管理是在一个共享的软件应用程序中托管多个租户(客户)的数据,并确保每个租户的数据完全隔离、安全和可管理。需要支持相同数据(记录)可由具有不同容量的多个租户。

1.2 租户框架

​ mybatis-plus租户插件使用

由于业务数据,都需要通过对数据库的crud,而此时mybatis-plus作为orm框架,提供了租户功能,我们可以合理利用拓展,应用于自己的业务范畴中

二. Mybatis-plus 租户

2.1 租户处理器

我们通过实现TenantLineHandler,即可实现对数据库中的表,进行租户过滤

属性名类型默认值描述
TenantLineHandlerTenantLineHandler租户处理器( TenantId 行级 )

实现租户过滤

  1. 这里我们通过匿名内部类实现TenantLineHandler。(也可以新建立一个类实现)
  2. 向MybatisPlusInterceptor添加我们所实现的匿名内部类
  3. 注入bean

ps:配置完后,就可实现对表进行租户筛选,读者们可自行测试

​ 由于后面我对该代码进行了加工,该代码会出现一些自定义逻辑在里面,但原理是不变的


/*** @author zhanghp* @since 2023/11/20 14:29*/
@Configuration
public class TenantConfig {@Beanpublic MybatisPlusInterceptor mybatisPlusInterceptor(TenantProperties properties) {MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {@Overridepublic Expression getTenantId() {// 默认租户为1return new LongValue(1);}@Overridepublic String getTenantIdColumn() {return properties.getTenantId();}// 这是 default 方法,默认返回 false 表示所有表都需要拼多租户条件@Overridepublic boolean ignoreTable(String tableName) {return false;}}));return interceptor;}}

2.2 前置准备

1. 依赖

各依赖版本继承至父依赖,可自行查看

     <dependencies><!-- 启动 --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><!-- 配置解析 --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-configuration-processor</artifactId><optional>true</optional></dependency><!-- aop --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-aop</artifactId><scope>provided</scope></dependency><!-- test --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-test</artifactId><scope>test</scope></dependency><!-- mybatis-plus --><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-boot-starter</artifactId></dependency><!-- mysql jdbc --><dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><scope>runtime</scope></dependency><!-- 代码生成器 --><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-generator</artifactId></dependency><!-- 生成器模版引擎 --><dependency><groupId>org.apache.velocity</groupId><artifactId>velocity-engine-core</artifactId></dependency><!-- alibaba thread local --><dependency><groupId>com.alibaba</groupId><artifactId>transmittable-thread-local</artifactId></dependency><!-- lombok --><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId></dependency><!-- hutool all --><dependency><groupId>cn.hutool</groupId><artifactId>hutool-all</artifactId></dependency></dependencies>

2. 表及数据准备

完整配置:application.yml

这里我沿用了Spring自带的属性配置spring.sql.init,初始化数据库的schema,data。

建表sql | 数据

当启动项目后,自动进行建表,初始化表及相应的数据

spring.io官网对配置文件的概述:可自行搜索spring.sql.init解锁更多使用方式

  • 指定数据库配置
  • 指定出初始化的建表sql,及对应的数据
spring:datasource:driver-class-name: com.mysql.cj.jdbc.Driverusername: rootpassword: zhp.1221url: jdbc:mysql://127.0.0.1:3306/test?useSSL=false&useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTCsql:init:# 建表sqlschema-locations: classpath:db/schema.sql# 数据data-locations: classpath:db/data.sql# 生成模式mode: always

3. 代码生成器

调用代码生成Generate.java,自动生成表实体类的相关类

这里就不进行展示了,自动代码生成若有不懂的地方可看Mybatis plus 自动生成代码与自定义模板

2.3 使用

🌳MpTest.java:测试对db的增删改查


/*** @author zhanghp* @since 2023/11/20 14:21*/
public class MpTest extends AppTest {@Resourceprivate DemoMapper demoMapper;@Testpublic void select() {demoMapper.selectList(null);}@Testpublic void insert() {demoMapper.insert(Demo.builder().id(100).age(18).name("草莓熊").build());}@Testpublic void update() {demoMapper.updateById(Demo.builder().id(100).age(20).name("皮卡").build());}@Testpublic void delete() {demoMapper.deleteById(Demo.builder().id(100).build());}}

select查询结果日志:

image-20231124182325830

insert插入日志:

在这里插入图片描述

update修改日志:

在这里插入图片描述

delete删除日志:

在这里插入图片描述

三. 深入使用

3.1 前言

Saas多租户使用时,其实面临着许多技术问题,我拿几个公司所遇见的经典问题进行举例

  1. 如何传递tenat_id
  2. 有一些配置表,永远不需要租户过滤
  3. 在一个请求里,不需要租户过滤
  4. 在一个请求里,部分数据db需要租户过滤,部分数据db不需要租户过滤
  5. 微服务feign传递(由于我的代码项目是单体架构,我只简单赘述如何实现)

3.2 租户主体设值,取值

一个租户的设值,一般分为两种:

  1. 一个请求里,附带租户值,一般放在请求头(header)
  2. 自定义设值引用

设值后,需要找个地方存储,这里引用alibaba的TransmittableThreadLocal进行值的存储,也方便后续取值,所以这里新建立一个TenantContextHolder类

TENAN_ID:进行值的存储与读取

TENANT_SKIP:进行租户是否过滤,也就是对db是否进行租户过滤的判断(where tenant_id = ?),下一节会讲到

@UtilityClass:对所有方法加static,对类加final,不允许new


/*** @author zhanghp* @since 2023/11/20 20:28*/
@Data
@UtilityClass
public class TenantContextHolder {/*** tenant_id*/private final ThreadLocal<Long> TENAT_ID = new TransmittableThreadLocal<>();/*** 获取租户* @return 租户id*/public Long getTenantId() {return TENAT_ID.get();}/*** 设置租户* @param tenantId 租户*/public void setTenantId(Long tenantId) {TENAT_ID.set(tenantId);}/*** 清空租户信息*/public void clear() {TENAT_ID.remove();}}

​ 当有了这个类后,那我们需要重新实现TenantLineHandler,由于之前已经定义了一个TenantConfig的bean来对租户处理器并注入到容器中,若我们再次定义一个会导致bean冗余,让spring找不到使用哪个,进而导致启动报错。


那么问题来了,有没有一种方式我既要能控制新定义的TenangLineHandler存在,老的也兼容呢?

这时我们使用一下@ConditionalOnMissingBean使用在TenantConfig上,和@Order使用在DeepTenantConfig即可,读者可自行浏览

DeepTenantConfig

@ConditionalOnProperty(prefix = “custom.tenant”, value = “deep-config”, havingValue = “true”):
代表我是否要使用这个bean注入,为什么要有这个注解来控制bean注入呢?是因为要兼容上一章的租户使用TenantConfig,选择注入哪个,若没有该注解,则都会默认注入DeepTenantConfig的bean,控制该bean的注入在application.yml里的custom.tenant.deep-config


/*** @author zhanghp* @since 2023/11/24 17:51*/
@Configuration
public class DeepTenantConfig {@Bean@Order(Integer.MIN_VALUE)@ConditionalOnProperty(prefix = "custom.tenant", value = "deep-config", havingValue = "true")public MybatisPlusInterceptor mybatisPlusInterceptor() {MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {@Overridepublic Expression getTenantId() {// 获取租户Long tenantId = TenantContextHolder.getTenantId();// 租户为空则返回空if (tenantId == null) {return new NullValue();}// 返回租户return new LongValue(tenantId);}// 这是 default 方法,默认返回 false 表示所有表都需要拼多租户条件@Overridepublic boolean ignoreTable(String tableName) {// 租户为空,则对该表不进行租户的操作Long tenantId = TenantContextHolder.getTenantId();if (tenantId == null) {return Boolean.TRUE;}return false;}}));return interceptor;}}

这些都处理好后,我们在配置中把custom.tenant.deep-config设置为true,这样读取的是我们上面定义的bean


我们在创建一个请求处理器类,进行租户的处理

TenantInterceptor

当一次请求进来,会进行租户值当处理,并存储在刚才我们所建立的TenantContextHolder类里,进行租户值处理


/*** @author zhanghp* @date 2023-11-25 13:48*/
public class TenantInterceptor implements HandlerInterceptor {@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {if (StrUtil.isNotBlank("tenant-id")) {TenantContextHolder.setTenantId(Convert.toLong(tenantId));}return true;}@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {TenantContextHolder.clear();}
}

将拦截器添加到拦截器注册类里


/*** @author zhanghp* @date 2023-11-25 16:53*/
@Configuration
public class CustomInterceptorConfig implements WebMvcConfigurer {@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(getRequestContextInterceptor());}@Beanpublic TenantInterceptor getRequestContextInterceptor() {return new TenantInterceptor();}
}

现在我们创建一个controller


/*** <p>* mybatis - demo表 前端控制器* </p>** @author zhp* @since 2023-11-20*/
@RestController
@RequestMapping("/zhanghp/demo")
public class DemoController {@Resourceprivate DemoService demoService;@GetMapping("/get")public void getall(){List<Demo> all = demoService.getAll();if (IterUtil.isEmpty(all)) {return;}all.forEach(System.out::println);}
}

现在我们打开apifox/postman,进行一次请求操作,看是否会生效

  • http://localhost:8888/zhanghp/demo/get
  • 请求头添加tenant_id的值

发送请求
在这里插入图片描述

结果日志打印

在这里插入图片描述


3.3 部分表全量db操作

⭐️场景

其实在业务中,有一些配置表,字典表其实是不需要租户的,就比如城市配置表,城市就这些,加租户也是没有任何意义的

⭐️解决

实现其实也是很简单,只需要在DeepTenantConfig里的的ignoreTables方法进行一些逻辑处理就可以

实现步骤

  1. 在application.yml进行表的配置
  2. 读取配置,并在DeepTenantConfig添加自己的逻辑

这里由于读取了配置,我喜欢把自定义的配置定义一个类,通过该类对配置读取,读者可自行查看,这里就不赘述了

配置:

custom:tenant:# 是否注入deepTenantConfig的bean开关deep-config: true# 租户字段名称tenant-id: tenant_id# 忽略租户的表ignore-tables:- demo01- demo02

逻辑新增


/*** @author zhanghp* @since 2023/11/24 17:51*/
@Configuration
public class DeepTenantConfig {@Bean@Order(Integer.MIN_VALUE)@ConditionalOnProperty(prefix = "custom.tenant", value = "deep-config", havingValue = "true")public MybatisPlusInterceptor mybatisPlusInterceptor(TenantProperties properties) {MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {@Overridepublic Expression getTenantId() {// 获取租户Long tenantId = TenantContextHolder.getTenantId();// 租户为空则返回空if (tenantId == null) {return new NullValue();}// 返回租户return new LongValue(tenantId);}@Overridepublic String getTenantIdColumn() {return properties.getTenantId();}// 这是 default 方法,默认返回 false 表示所有表都需要拼多租户条件@Overridepublic boolean ignoreTable(String tableName) {Long tenantId = TenantContextHolder.getTenantId();if (tenantId == null) {return Boolean.TRUE;}// 忽略表对配置为空,则所有表都进行租户操作if (IterUtil.isEmpty(properties.getIgnoreTables())) {return false;}// 指定的表永远不进行租户操作if (properties.getIgnoreTables().contains(tableName)) {return true;}return false;}}));return interceptor;}
}

读者自行练习就可,就不再演示

3.4 全量db操作

⭐️场景

在一个请求里,这个请求涉及到的表是不需要租户过滤,那么读者就会想到,那在上一节的配置文件加啊。

这里的一个请求不需要对A表进行过滤,但另一个请求对A是需要进行租户过滤的,那这时加配置文件是行不通的

⭐️解决

  1. TenantContextHolder里定义一个过滤租户的threadlocal
  2. 在DeepTenantConfig的ignoreTables()方法新增逻辑判断
/*** @author zhanghp* @since 2023/11/20 20:28*/
@Data
@UtilityClass
public class TenantContextHolder {/*** tenant_id*/private final ThreadLocal<Long> TENAT_ID = new TransmittableThreadLocal<>();/*** 租户过滤标识*/private final ThreadLocal<Boolean> TENANT_SKIP = new TransmittableThreadLocal<>();/*** 获取租户* @return 租户id*/public Long getTenantId() {return TENAT_ID.get();}/*** 设置租户* @param tenantId 租户*/public void setTenantId(Long tenantId) {TENAT_ID.set(tenantId);}/*** 设置是否过滤的标识*/public void setTenantSkip() {TENANT_SKIP.set(Boolean.TRUE);}/*** 获取是否跳过租户过滤的标识* @return true-过滤 false-不过滤*/public Boolean getTenantSkip() {return TENANT_SKIP.get() != null && TENANT_SKIP.get();}/*** 清空租户信息*/public void clear() {TENAT_ID.remove();clearSkip();}/*** 清空租户过滤标识*/public void clearSkip(){TENANT_SKIP.remove();}}

DeepTenantConfig


/*** @author zhanghp* @since 2023/11/24 17:51*/
@Configuration
public class DeepTenantConfig {@Bean@Order(Integer.MIN_VALUE)@ConditionalOnProperty(prefix = "custom.tenant", value = "deep-config", havingValue = "true")public MybatisPlusInterceptor mybatisPlusInterceptor(TenantProperties properties) {MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {@Overridepublic Expression getTenantId() {// 获取租户Long tenantId = TenantContextHolder.getTenantId();// 租户为空则返回空if (tenantId == null) {return new NullValue();}// 返回租户return new LongValue(tenantId);}@Overridepublic String getTenantIdColumn() {return properties.getTenantId();}// 这是 default 方法,默认返回 false 表示所有表都需要拼多租户条件@Overridepublic boolean ignoreTable(String tableName) {// 1.是否跳过租户对该表对操作if (TenantContextHolder.getTenantSkip()) {return Boolean.TRUE;}// 2.租户为空,则对该表不进行租户的操作Long tenantId = TenantContextHolder.getTenantId();if (tenantId == null) {return Boolean.TRUE;}// 3.忽略表对配置为空,则所有表都进行租户操作if (IterUtil.isEmpty(properties.getIgnoreTables())) {return false;}// 4.指定的表永远不进行租户操作if (properties.getIgnoreTables().contains(tableName)) {return true;}return false;}}));return interceptor;}}

在controller层新加一个TenantContextHolder.setTenantSkip()即可

@GetMapping("/get2")public void getall2(){TenantContextHolder.setTenantSkip();List<Demo> all = demoService.list();if (IterUtil.isEmpty(all)) {return;}all.forEach(System.out::println);}

读者自行练习即可

3.5 自由控制全量/租户对db操作

⭐️场景

在一个请求里,进行了对A表和B表操作,但是A表是不需要租户,而B表又是需要租户

⭐️解决

  1. 对A表的操作提取为一个a方法,对B的操作提取出一个b方法
  2. 利用上节的TENANT_SKIP让a方法跳过租户
  3. 当a方法执行后,利用切面进行TENANT_SKIP的清空
  4. 这时b方法就会带着租户对B表进行处理

这里我写了一个注解@TenanClear

作用域:方法,类

作用:执行当前方法里的所有db操作,都不带租户,通过属性globalFlag进行控制后续方法是否添加租户过滤

ps:是=实现原理在 TenantClearAspect

/*** @author zhanghp* @since 2023/11/21 20:41*/
@Target({ ElementType.METHOD, ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface TenantClear {/*** <p>执行该方法后,是否全局过滤租户标志</p>* <ul>*     <li>true:执行完该线程之前,后续对db操作不添加租户操作</li>*     <li>false:执行完当前方法后,后续方法的db操作添加租户操作</li>* </ul>** @return 默认后续不添加租户操作*/boolean globalFlag() default true;
}

该注解的作用:当添加该注解后当前方法进行的所有db操作都不带租户处理,后续方法执行是否带租户,通过属性globalFlag决定


创建a方法,并创建b方法

在a方法添加@TenantClear(globalFlag = false),然后在controller层先调用a,在调用b

输出结果,a不带租户,b带租户


/*** <p>* mybatis - demo表 服务实现类* </p>** @author zhp* @since 2023-11-20*/
@Service
public class DemoServiceImpl extends ServiceImpl<DemoMapper, Demo> implements DemoService {@Overridepublic void b() {super.list();}@Override@TenantClear(globalFlag = false)public void a(){super.list();}}

在controller层调用

    @GetMapping("/tenant-clear")public void tenantClear(){Console.log("执行a方法");demoService.a();Console.log("执行b方法");demoService.b();}

结果打印

在这里插入图片描述

3.6 feign传递租户

由于本项目是单体项目,不涉及微服务,在这里只是简单概述

实现步骤:

  1. 实现feign自带的原生扩展RequestInterceptor类
  2. 重写apply方法,处理tenant-id的传递
public class CustomerFeignInterceptor implements RequestInterceptor {public CustomerFeignInterceptor() {}public void apply(RequestTemplate requestTemplate) {if (TenantContextHolder.getTenantId() == null) {log.debug("租户ID为空,feign不传递");} else {requestTemplate.header("tenant-id", new String[]{TenantContextHolder.getTenantId().toString()});}}
}

四. 源码地址

⭐️gitee:https://gitee.com/zhp1221/java/tree/master/lab_02_mybatis_plus/tenant

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/218409.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

基于 STM32 的温度测量与控制系统设计

本文介绍了如何基于 STM32 微控制器设计一款温度测量与控制系统。首先&#xff0c;我们将简要介绍 STM32 微控制器的特点和能力。接下来&#xff0c;我们将详细讨论温度传感器的选择与接口。然后&#xff0c;我们将介绍如何使用 STM32 提供的开发工具和相关库来进行温度测量和控…

PostgreSQL 分区表插入数据及报错:子表明明存在却报不存在以及column “xxx“ does not exist 解决方法

PostgreSQL 分区表插入数据及报错&#xff1a;子表明明存在却报不存在以及column “xxx“ does not exist 解决方法 问题1. 分区表需要先创建子表在插入&#xff0c;创建子表立马插入后可能会报错子表不存在&#xff1b;解决&#xff1a; 创建子表及索引后&#xff0c;sleep10毫…

Feast:概念

实体 实体是语义相关特征的集合。用户定义实体以映射其用例的领域。 例如&#xff0c;打车车服务可以将客户和司机作为实体&#xff0c;这些实体将相关特征分组&#xff0c;对应于这些客户和司机。 driver Entity(namedriver, join_keys[driver_id]) # 司机实体 customer E…

从0到0.01入门 Webpack| 007.精选 Webpack面试题

&#x1f90d; 前端开发工程师&#xff08;主业&#xff09;、技术博主&#xff08;副业&#xff09;、已过CET6 &#x1f368; 阿珊和她的猫_CSDN个人主页 &#x1f560; 牛客高级专题作者、在牛客打造高质量专栏《前端面试必备》 &#x1f35a; 蓝桥云课签约作者、已在蓝桥云…

CSS特效019:图标图片悬浮旋转一周

CSS常用示例100专栏目录 本专栏记录的是经常使用的CSS示例与技巧&#xff0c;主要包含CSS布局&#xff0c;CSS特效&#xff0c;CSS花边信息三部分内容。其中CSS布局主要是列出一些常用的CSS布局信息点&#xff0c;CSS特效主要是一些动画示例&#xff0c;CSS花边是描述了一些CSS…

使用SpringBoot集成FastDFS

使用SpringBoot集成FastDFS 这篇文章我们介绍如何使用 Spring Boot 将文件上传到分布式文件系统 FastDFS 中。 1、FastDFS FastDFS是一个开源的轻量级分布式文件系统&#xff0c;它对文件进行管理&#xff0c;功能包括&#xff1a;文件存储、文件同步、文件访问 &#xff0…

Linux关于定时任务crontab相关知识了解配合理解shell反弹远程控制

Linux关于定时任务crontab相关知识了解配合理解shell反弹远程控制 几点需要知道的信息 【1】crontab一般来说服务器都是有的&#xff0c;依赖crond服务&#xff0c;这个服务也是必须安装的服务&#xff0c;并且也是开机自启动的服务&#xff0c;也就是说&#xff0c;他基本上是…

Day41力扣打卡

打卡记录 第 N 位数字&#xff08;找规律&#xff09; 链接 class Solution:def findNthDigit(self, n: int) -> int:count, digit, start 9, 1, 1while n > count:n - countdigit 1start * 10count start * 9 * digitnum start (n - 1) // digitreturn int(str(n…

HCIA题目解析(1)

1、【多选题】关于动态 MAC 地址表说法正确的是&#xff1f; A、通过报文中的源MAC地址学习获得的动态MAC表项会老化 B、通过查看指定动态MAC地址表项的个数&#xff0c;可以获取接口下通信的用户数 C、在设备重启后&#xff0c;之前的动态表项会丢失 D、在设备重启后&…

电源控制系统架构(PCSA)之电源控制框架概览

目录 6 电源控制框架 6.1 电源控制框架概述 6.1.1 电源控制框架低功耗接口 6.1.2 电源控制框架基础设施组件 6 电源控制框架 电源控制框架是标准基础设施组件、接口和相关方法的集合&#xff0c;可用于构建SoC电源管理所需的基础设施。 本章介绍框架的主要组件和低功耗接…

ubuntu+Teslav100 环境配置

系统基本信息 nvidia-smi’ nvidia-smi 470.182.03 driver version:470.182.03 cuda version: 11.4 产看系统体系结构 uname -aUTC 2023 x86_64 x86_64 x86_64 GNU/Linux 下载miniconda https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/?CM&OA https://mi…

elk 简单操作手册

1.1. 基础概念 EFK不是一个软件,而是一套解决方案,开源软件之间的互相配合使用,高效的满足了很多场合的应用,是目前主流的一种日志系统。 EFK是三个开源软件的缩写,分别表示:Elasticsearch , Filebeat, Kibana , 其中Elasticsearch负责日志保存和搜索,Filebeat负责收集日志,Ki…