diff --git a/blade-core-boot/src/main/java/org/springblade/core/boot/tenant/TenantIdUtil.java b/blade-core-boot/src/main/java/org/springblade/core/boot/tenant/TenantIdUtil.java index 8400627..c52f148 100644 --- a/blade-core-boot/src/main/java/org/springblade/core/boot/tenant/TenantIdUtil.java +++ b/blade-core-boot/src/main/java/org/springblade/core/boot/tenant/TenantIdUtil.java @@ -61,4 +61,20 @@ public class TenantIdUtil { } } + /** + * 使用租户 id 执行函数 + * + * @param tenantId tenantId + * @param runnable Runnable + */ + public static void use(String tenantId, Runnable runnable) { + Assert.hasText(tenantId, "参数 tenantId 为空"); + tl.set(tenantId); + try { + runnable.run(); + } finally { + tl.remove(); + } + } + }