com.power4j.fist.boot.mybaits.tenant.TenantHandler Maven / Gradle / Ivy
/*
* Copyright 2021 ChenJun ([email protected] & https://github.com/John-Chan)
*
* 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
*
* http://www.gnu.org/licenses/lgpl.html
*
* 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 com.power4j.fist.boot.mybaits.tenant;
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.power4j.fist.data.tenant.isolation.TenantHolder;
import com.power4j.fist.boot.security.core.SecurityConstant;
import lombok.Setter;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.StringValue;
import java.util.Set;
/**
* @author CJ ([email protected])
* @date 2021/11/11
* @since 1.0
*/
public class TenantHandler implements TenantLineHandler {
private final String tenantColumn;
private final Set tables;
@Setter
private String defaultTenant = SecurityConstant.TENANT_ZERO;
public TenantHandler(String tenantColumn, Set tables) {
this.tenantColumn = tenantColumn;
this.tables = tables;
}
@Override
public Expression getTenantId() {
return new StringValue(TenantHolder.getTenant().orElse(defaultTenant));
}
@Override
public String getTenantIdColumn() {
return tenantColumn;
}
@Override
public boolean ignoreTable(String tableName) {
return !tables.contains(tableName);
}
}