xiongan-meeting/ics-system/src/main/java/com/ics/system/handlers/MyParkLineHandler.java

package com.ics.system.handlers;

import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.ics.common.utils.StringUtils;
import com.ics.system.service.ICurrentUserService;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;
import net.sf.jsqlparser.expression.NullValue;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Arrays;

@Slf4j
@Component
public class MyParkLineHandler implements TenantLineHandler {

    /**
     * 排除过滤的表
     */
    private static final String[] tableList = {"tables", "columns", "sys_tenant", "ics_park", "sys_config", "sys_dict_type", "sys_dict_data", "sys_districts",
            "sys_job", "sys_job_log", "sys_login_info", "sys_menu", "sys_notice", "sys_oper_log", "sys_oss", "sys_role", "sys_role_dept", "sys_role_menu",
            "sys_sn", "sys_user_role", "sys_dept", "ics_customer_contract_room", "ics_park", "ics_apply_room", "ics_customer_contract_refund_room",  "ics_apply_park_file",
            "ics_apply_settle_file", "ics_apply_move_in_file", "ics_activity","ics_customer_staff","tb_customer_ticket","tb_reservation","tb_reservation_person",
            "tb_room_content","tb_room_item","tb_room_item_by_room","tb_room_serve_by_room","tb_room_serve","tb_equipment","tb_staff_customer",
            "tb_room_equipment","tb_room_record","tb_room_serve","tb_showroom","tb_showroom_record","tb_ticket","tb_user_equipment","tb_visitor_person"
    };

    /**
     * 多租户标识
     */
    private static final String PARK_ID = "park_id";

    /**
     * 排除过滤的表前缀
     */
    private static final String[] tablePrefix = {"qrtz", "gen"};

    @Autowired
    private ICurrentUserService userOnlineService;

    /**
     * 返回当前用户的租户ID
     */
    @Override
    public Expression getTenantId() {
        // 从当前系统上下文中取出当前请求的服务商ID，通过解析器注入到SQL中。
        Long parkId = userOnlineService.getParkId();
        if (parkId == null) {
            return new NullValue();
        }
        return new LongValue(parkId);
    }

    /**
     * 租户字段名
     *
     * @return
     */
    @Override
    public String getTenantIdColumn() {
        return PARK_ID;
    }

    /**
     * 跳过不需要加多租户的表
     */
    @Override
    public boolean ignoreTable(String tableName) {
        String prefix = StringUtils.substringBefore(tableName, "_");
        if (Arrays.asList(tableList).contains(tableName) || Arrays.asList(tablePrefix).contains(prefix) || getParkId() == null) {
            //无租户模式
            return true;
        }
        return false;
    }

    /**
     * 当前部门Id
     */
    private Long getParkId() {
        log.debug("当前园区为{}", userOnlineService.getParkId());
        return userOnlineService.getParkId();
    }

}