添加 TenantIdUtil 方便在非 web 场景使用

This commit is contained in:
卢春梦 2024-04-29 16:26:05 +08:00
parent de41acdd51
commit 8286b9f6ea
3 changed files with 70 additions and 3 deletions

View File

@ -42,7 +42,7 @@ public class BladeTenantHandler implements TenantLineHandler {
*/ */
@Override @Override
public Expression getTenantId() { public Expression getTenantId() {
return new StringValue(Func.toStr(SecureUtil.getTenantId(), TenantConstant.DEFAULT_TENANT_ID)); return new StringValue(Func.toStr(TenantIdUtil.get(), properties.getDefaultTenantId()));
} }
/** /**
@ -65,10 +65,10 @@ public class BladeTenantHandler implements TenantLineHandler {
public boolean ignoreTable(String tableName) { public boolean ignoreTable(String tableName) {
return !( return !(
( (
(properties.getTables().size() > 0 && properties.getTables().contains(tableName)) (!properties.getTables().isEmpty() && properties.getTables().contains(tableName))
|| properties.getBladeTables().contains(tableName) || properties.getBladeTables().contains(tableName)
) )
&& StringUtil.isNotBlank(SecureUtil.getTenantId()) && StringUtil.isNotBlank(TenantIdUtil.get())
); );
} }
} }

View File

@ -38,6 +38,11 @@ public class BladeTenantProperties {
*/ */
private String column = "tenant_id"; private String column = "tenant_id";
/**
* 默认的租户id默认为000000
*/
private String defaultTenantId = TenantConstant.DEFAULT_TENANT_ID;
/** /**
* 多租户数据表 * 多租户数据表
*/ */

View File

@ -0,0 +1,62 @@
/**
* Copyright (c) 2018-2028, Chill Zhuang 庄骞 (smallchill@163.com).
* <p>
* Licensed under the GNU LESSER GENERAL PUBLIC LICENSE 3.0;
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* <p>
* http://www.gnu.org/licenses/lgpl.html
* <p>
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springblade.core.boot.tenant;
import org.springblade.core.secure.utils.SecureUtil;
import org.springframework.util.Assert;
import java.util.function.Supplier;
/**
* TenantId 工具
*
* @author L.cm
*/
public class TenantIdUtil {
private static final ThreadLocal<String> tl = new ThreadLocal<>();
/**
* 获取租户id
*
* @return 租户id
*/
public static String get() {
String tenantId = tl.get();
if (tenantId != null) {
return tenantId;
}
return SecureUtil.getTenantId();
}
/**
* 使用租户 id 执行函数
*
* @param tenantId tenantId
* @param supplier supplier
* @param <R> 泛型
* @return R 函数返回
*/
public static <R> R use(String tenantId, Supplier<R> supplier) {
Assert.hasText(tenantId, "参数 tenantId 为空");
tl.set(tenantId);
try {
return supplier.get();
} finally {
tl.remove();
}
}
}