81 lines
1.9 KiB
Java
81 lines
1.9 KiB
Java
package com.youlai.boot.common.tenant;
|
||
|
||
import lombok.extern.slf4j.Slf4j;
|
||
|
||
/**
|
||
* 租户上下文工具类
|
||
* <p>
|
||
* 使用 ThreadLocal 存储当前线程的租户ID，确保线程安全
|
||
* </p>
|
||
*
|
||
* @author Ray.Hao
|
||
* @since 3.0.0
|
||
*/
|
||
@Slf4j
|
||
public class TenantContextHolder {
|
||
|
||
/**
|
||
* 租户ID线程本地变量
|
||
*/
|
||
private static final ThreadLocal<Long> TENANT_ID_HOLDER = new ThreadLocal<>();
|
||
|
||
/**
|
||
* 忽略租户标志（用于某些场景下临时跳过租户过滤）
|
||
*/
|
||
private static final ThreadLocal<Boolean> IGNORE_TENANT_HOLDER = new ThreadLocal<>();
|
||
|
||
/**
|
||
* 设置当前租户ID
|
||
*
|
||
* @param tenantId 租户ID
|
||
*/
|
||
public static void setTenantId(Long tenantId) {
|
||
if (tenantId != null) {
|
||
TENANT_ID_HOLDER.set(tenantId);
|
||
log.debug("设置当前租户ID: {}", tenantId);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取当前租户ID
|
||
*
|
||
* @return 租户ID，如果未设置则返回 null
|
||
*/
|
||
public static Long getTenantId() {
|
||
return TENANT_ID_HOLDER.get();
|
||
}
|
||
|
||
/**
|
||
* 设置忽略租户标志
|
||
*
|
||
* @param ignore 是否忽略
|
||
*/
|
||
public static void setIgnoreTenant(boolean ignore) {
|
||
IGNORE_TENANT_HOLDER.set(ignore);
|
||
log.debug("设置忽略租户标志: {}", ignore);
|
||
}
|
||
|
||
/**
|
||
* 是否忽略租户
|
||
*
|
||
* @return true-忽略，false-不忽略
|
||
*/
|
||
public static boolean isIgnoreTenant() {
|
||
Boolean ignore = IGNORE_TENANT_HOLDER.get();
|
||
return ignore != null && ignore;
|
||
}
|
||
|
||
/**
|
||
* 清除当前线程的租户上下文
|
||
* <p>
|
||
* 必须在请求结束时调用，避免线程池复用导致的数据泄露
|
||
* </p>
|
||
*/
|
||
public static void clear() {
|
||
TENANT_ID_HOLDER.remove();
|
||
IGNORE_TENANT_HOLDER.remove();
|
||
log.debug("清除租户上下文");
|
||
}
|
||
}
|
||
|