package com.ruoyi.common.tenant.handle;
|
|
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
|
import com.ruoyi.common.core.utils.StringUtils;
|
import com.ruoyi.common.satoken.utils.LoginHelper;
|
import com.ruoyi.common.tenant.helper.TenantHelper;
|
import com.ruoyi.common.tenant.properties.TenantProperties;
|
import lombok.AllArgsConstructor;
|
import net.sf.jsqlparser.expression.Expression;
|
import net.sf.jsqlparser.expression.LongValue;
|
import net.sf.jsqlparser.expression.NullValue;
|
|
import java.util.List;
|
|
/**
|
* 自定义租户处理器
|
*
|
* @author Lion Li
|
*/
|
@AllArgsConstructor
|
public class PlusTenantLineHandler implements TenantLineHandler {
|
|
private final TenantProperties tenantProperties;
|
|
@Override
|
public Expression getTenantId() {
|
String tenantId = LoginHelper.getTenantId();
|
if (StringUtils.isBlank(tenantId)) {
|
return new NullValue();
|
}
|
String dynamicTenantId = TenantHelper.getDynamic();
|
if (StringUtils.isNotBlank(dynamicTenantId)) {
|
// 返回动态租户
|
return new LongValue(dynamicTenantId);
|
}
|
// 返回固定租户
|
return new LongValue(tenantId);
|
}
|
|
@Override
|
public boolean ignoreTable(String tableName) {
|
String tenantId = LoginHelper.getTenantId();
|
// 判断是否有租户
|
if (StringUtils.isNotBlank(tenantId)) {
|
// 不需要过滤租户的表
|
List<String> excludes = tenantProperties.getExcludes();
|
// 非业务表
|
excludes.addAll(List.of(
|
"gen_table",
|
"gen_table_column"
|
));
|
return excludes.contains(tableName);
|
}
|
return true;
|
}
|
|
}
|
