package com.zhgd.mybatis; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.convert.Convert; import cn.hutool.core.util.ReflectUtil; import cn.hutool.core.util.StrUtil; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.handler.DataPermissionHandler; import com.zhgd.annotation.DataScope; import com.zhgd.jeecg.common.util.SpringContextUtils; import com.zhgd.xmgl.modules.baotou.entity.UserDevGroup; import com.zhgd.xmgl.modules.baotou.service.IUserDevGroupService; import com.zhgd.xmgl.modules.baotou.service.IUserDeviceService; import com.zhgd.xmgl.modules.basicdata.entity.SystemUser; import com.zhgd.xmgl.modules.basicdata.enums.SystemUserAccountTypeEnum; import com.zhgd.xmgl.modules.basicdata.service.ISystemUserService; import com.zhgd.xmgl.modules.bigdevice.mapper.TowerMapper; import com.zhgd.xmgl.modules.car.mapper.CarCameraMapper; import com.zhgd.xmgl.modules.environment.mapper.EnvironmentDevMapper; import com.zhgd.xmgl.modules.hangbasket.mapper.HangBasketDevMapper; import com.zhgd.xmgl.modules.poisonous.mapper.PoisonousGasDevMapper; import com.zhgd.xmgl.modules.project.entity.ProjectConfig; import com.zhgd.xmgl.modules.project.service.IProjectConfigService; import com.zhgd.xmgl.modules.safetyhat.mapper.SafetyHatDevMapper; import com.zhgd.xmgl.modules.video.mapper.AiAnalyseHardWareRecordMapper; import com.zhgd.xmgl.modules.video.service.IAiAnalyseHardWareRecordService; import com.zhgd.xmgl.modules.worker.mapper.EnterpriseInfoMapper; import com.zhgd.xmgl.modules.worker.service.impl.UserDevAuthorityServiceImpl; import com.zhgd.xmgl.modules.worker.service.impl.UserEnterpriseServiceImpl; import com.zhgd.xmgl.modules.xz.service.impl.XzSupplierQualificationApplyServiceImpl; import com.zhgd.xmgl.security.entity.UserInfo; import com.zhgd.xmgl.security.util.SecurityUtils; import com.zhgd.xmgl.util.EnvironmentUtil; import com.zhgd.xmgl.util.MapBuilder; import lombok.extern.slf4j.Slf4j; import net.sf.jsqlparser.JSQLParserException; import net.sf.jsqlparser.expression.Alias; import net.sf.jsqlparser.expression.Expression; import net.sf.jsqlparser.expression.Parenthesis; import net.sf.jsqlparser.expression.StringValue; import net.sf.jsqlparser.expression.operators.conditional.AndExpression; import net.sf.jsqlparser.expression.operators.conditional.OrExpression; import net.sf.jsqlparser.expression.operators.relational.EqualsTo; import net.sf.jsqlparser.expression.operators.relational.ExpressionList; import net.sf.jsqlparser.expression.operators.relational.InExpression; import net.sf.jsqlparser.expression.operators.relational.ItemsList; import net.sf.jsqlparser.parser.CCJSqlParserUtil; import net.sf.jsqlparser.schema.Column; import net.sf.jsqlparser.schema.Table; import net.sf.jsqlparser.statement.select.FromItem; import net.sf.jsqlparser.statement.select.Join; import net.sf.jsqlparser.statement.select.PlainSelect; import org.apache.commons.collections.CollectionUtils; import org.apache.commons.collections.MapUtils; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Lazy; import java.lang.reflect.Method; import java.util.*; import java.util.stream.Collectors; @Slf4j public class DataScopeHandler implements DataPermissionHandler { public static final String DATA_COLUMN = "dataField"; public static final String DEV_FIELD = "devField"; public static final String DEV_MAPPER = "dev_mapper"; public static final String ALIAS_NAME = "aliasName"; public static final String DEVICE_COLUMN = "devField"; public static final String PROJECT_COLUMN = "projectField"; @Lazy @Autowired EnvironmentUtil environmentUtil; @Lazy @Autowired IAiAnalyseHardWareRecordService aiAnalyseHardWareRecordService; @Lazy @Autowired private XzSupplierQualificationApplyServiceImpl xzSupplierQualificationApplyService; @Lazy @Autowired private EnterpriseInfoMapper enterpriseInfoMapper; @Lazy @Autowired private UserDevAuthorityServiceImpl userDevAuthorityService; @Lazy @Autowired private UserEnterpriseServiceImpl userEnterpriseService; @Lazy @Autowired private IProjectConfigService projectConfigService; @Lazy @Autowired private ISystemUserService systemUserService; @Lazy @Autowired private IUserDeviceService userDeviceService; @Lazy @Autowired private IUserDevGroupService userDevGroupService; @Override public Expression getSqlSegment(Expression where, String mappedStatementId) { return null; } protected String getProjectSn() { return "project_sn"; } /** * 获取设置的过滤的表名和字段名 * * @return */ private HashMap getFieldEnterpriseTables() { HashMap tables = new HashMap<>(16); tables.put("worker_admission", "certificate_issuing_unit"); tables.put("worker_info", "enterprise_id"); tables.put("team_info", "enterprise_id"); tables.put("department_info", "enterprise_id"); tables.put("xz_task_progress_content", "enterprise_id"); tables.put("xz_task_progress_alarm", "enterprise_id"); tables.put("xz_task_progress", "enterprise_id"); tables.put("quality_inspection_record", "enterprise_id"); tables.put("worker_info_audit_record", "enterprise_id"); tables.put("xz_material", "enterprise_id"); tables.put("inspect_task_record", "enterprise_id"); tables.put("xz_security_inspect_task_record", "enterprise_id"); tables.put("xz_security_quality_inspection_record", "enterprise_id"); tables.put("exam_notice", "enterprise_id"); tables.put("exam_train_record", "enterprise_id"); tables.put("exam_course_record", "enterprise_id"); tables.put("xz_worker_safe_watch_alarm", "enterprise_id"); tables.put("project_fine_record", "enterprise_id"); tables.put("car_info", "enterprise_id"); tables.put("enterprise_info", "id"); tables.put("quality_problem", "enterprise_id"); tables.put("civilize_construction", "enterprise_id"); return tables; } /** * 合作单位ids设备 * * @return */ private HashMap getFieldEnterpriseIdsTables() { HashMap tables = new HashMap<>(16); tables.put("ai_analyse_hard_ware_record", "enterprise_ids"); tables.put("car_camera", "enterprise_ids"); tables.put("hang_basket_dev", "enterprise_ids"); tables.put("deep_excavation_engineering", "enterprise_ids"); tables.put("high_formwork_plane_figure", "enterprise_ids"); tables.put("high_formwork_measure_point", "enterprise_ids"); tables.put("tower", "enterprise_ids"); tables.put("safety_hat_dev", "enterprise_ids"); tables.put("concrete_monitor_dev", "enterprise_ids"); tables.put("environment_dev", "enterprise_ids"); tables.put("poisonous_gas_dev", "enterprise_ids"); tables.put("uface_dev", "enterprise_ids"); // tables.put("video_item", "enterprise_ids"); // tables.put("anti_pressure_fold_dev", "enterprise_ids"); // tables.put("bridge_erect_machine_dev", "enterprise_ids"); // tables.put("car_wash_dev", "enterprise_ids"); // tables.put("concrete_mix_station_dev", "enterprise_ids"); // tables.put("discharging_platform_dev", "enterprise_ids"); // tables.put("double_carbon_dev", "enterprise_ids"); // tables.put("electrical_dev", "enterprise_ids"); // tables.put("frontier_protection_dev", "enterprise_ids"); // tables.put("frontier_protection_no_net_dev", "enterprise_ids"); // tables.put("gt_material_device", "enterprise_ids"); // tables.put("high_formwork_measure_device", "enterprise_ids"); // tables.put("mass_rebound_measure_dev", "enterprise_ids"); // tables.put("ms_mixed_soil_dev", "enterprise_ids"); // tables.put("pave_compaction_dev", "enterprise_ids"); // tables.put("pave_dev", "enterprise_ids"); // tables.put("photovoltaic_power_dev", "enterprise_ids"); // tables.put("pitch_mix_station_dev", "enterprise_ids"); // tables.put("pressure_test_machine_dev", "enterprise_ids"); // tables.put("sewage_dev", "enterprise_ids"); // tables.put("smart_grout_dev", "enterprise_ids"); // tables.put("smart_tension_dev", "enterprise_ids"); // tables.put("smoke_dev", "enterprise_ids"); // tables.put("spray_dev", "enterprise_ids"); // tables.put("spray_rt_dev", "enterprise_ids"); // tables.put("stable_water_mix_station_dev", "enterprise_ids"); // tables.put("standard_dev", "enterprise_ids"); // tables.put("universal_test_dev", "enterprise_ids"); // tables.put("vehicle_position_dev", "enterprise_ids"); return tables; } /** * 视频监控 * * @return */ private HashMap getFieldVideoTables() { HashMap tables = new HashMap<>(16); tables.put("video_item", "item_id"); return tables; } /** * 合作单位ids设备的数据 * * @return */ private HashMap> getFieldEnterpriseDataTables() { HashMap> tables = new HashMap<>(16); tables.put("car_pass_record", new MapBuilder().put(DATA_COLUMN, "camera_id").put(DEV_FIELD, "cameraId").put(DEV_MAPPER, CarCameraMapper.class).build()); tables.put("ai_analyse_hard_ware_alarm_record", new MapBuilder().put(DATA_COLUMN, "hardware_id").put(DEV_FIELD, "hardwareId").put(DEV_MAPPER, AiAnalyseHardWareRecordMapper.class).build()); tables.put("hang_basket_detect_data", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, HangBasketDevMapper.class).build()); tables.put("hang_basket_displacement_data", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, HangBasketDevMapper.class).build()); tables.put("hang_basket_alarm_data", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, HangBasketDevMapper.class).build()); tables.put("tower_current_data", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, TowerMapper.class).build()); tables.put("tower_worker_attendance", new MapBuilder().put(DATA_COLUMN, "tower_dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, TowerMapper.class).build()); tables.put("tower_work_cycle", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, TowerMapper.class).build()); tables.put("tower_nut", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, TowerMapper.class).build()); tables.put("tower_alarm", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, TowerMapper.class).build()); tables.put("safety_hat_data", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, SafetyHatDevMapper.class).build()); tables.put("safety_hat_alarm", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, SafetyHatDevMapper.class).build()); tables.put("dust_noise_data", new MapBuilder().put(DATA_COLUMN, "device_id").put(DEV_FIELD, "deviceId").put(DEV_MAPPER, EnvironmentDevMapper.class).build()); tables.put("environment_alarm", new MapBuilder().put(DATA_COLUMN, "device_id").put(DEV_FIELD, "deviceId").put(DEV_MAPPER, EnvironmentDevMapper.class).build()); tables.put("poisonous_gas_dev_current_data", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, PoisonousGasDevMapper.class).build()); tables.put("poisonous_gas_dev_alarm", new MapBuilder().put(DATA_COLUMN, "dev_sn").put(DEV_FIELD, "devSn").put(DEV_MAPPER, PoisonousGasDevMapper.class).build()); //tables.put("deep_excavation_engineering", "enterprise_ids"); //tables.put("high_formwork_plane_figure", "enterprise_ids"); //tables.put("high_formwork_measure_point", "enterprise_ids"); //tables.put("concrete_monitor_dev", "enterprise_ids"); //tables.put("uface_dev", "enterprise_ids"); return tables; } /** * 装置和项目组 * * @return */ private HashMap> getFieldDevProjectTables() { HashMap> tables = new HashMap<>(16); tables.put("civil_engineering_quality_inspection_order", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("device_master_schedule_detail", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("device_master_schedule_year", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("lifting_operation", new MapBuilder().put(DEVICE_COLUMN, "operation_equipment").build()); tables.put("operationEquipment", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("main_project_complete_detail", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("main_project_complete_period", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("main_project_complete_total", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("milestone", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("nondestructive_test_order_ticket", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("pipeline_information", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("pipeline_pressure_test_bag", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("pipeline_welder", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("project_schedule", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").build()); tables.put("quality_supervise", new MapBuilder().put(DEVICE_COLUMN, "device_id").build()); tables.put("worker_admission_detail", new MapBuilder().put(DEVICE_COLUMN, "work_area").build()); tables.put("danger_environment_evaluate", new MapBuilder().put(DEVICE_COLUMN, "device_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("civilize_construction", new MapBuilder().put(DEVICE_COLUMN, "device_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("confined_space_operation", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("construction_plan_ledger", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("first_example_manage", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("pipeline_material_inspection", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("pipeline_welding_procedure_qualification", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("pipeline_welding_record", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("pouring_order_ledger", new MapBuilder().put(DEVICE_COLUMN, "device").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("quality_problem", new MapBuilder().put(DEVICE_COLUMN, "device_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("xz_security_quality_inspection_record", new MapBuilder().put(DEVICE_COLUMN, "device_unit_id").put(PROJECT_COLUMN, "project_group_id").build()); tables.put("green_construction_scheme", new MapBuilder().put(DEVICE_COLUMN, "device_unit").put(PROJECT_COLUMN, "project_group").build()); tables.put("dangerous_engineering_record", new MapBuilder().put(DEVICE_COLUMN, "device_unit").put(PROJECT_COLUMN, "project_group").build()); return tables; } /** * ai预警 * * @return */ private HashMap getFieldAiTables() { HashMap tables = new HashMap<>(16); tables.put("ai_analyse_hard_ware_alarm_record", "hardware_id"); return tables; } // /** // * 项目组 // * // * @return // */ // private HashMap getFieldGroupsTables() { // HashMap tables = new HashMap<>(16); // tables.put("danger_environment_evaluate", "project_group_id"); // tables.put("project_group", "id"); // return tables; // } // /** // * 装置 // * // * @return // */ // private HashMap getFieldDevUnitTables() { // HashMap tables = new HashMap<>(16); // tables.put("danger_environment_evaluate", "device_id"); // tables.put("device_unit", "id"); // return tables; // } // private HashMap getFieldSecurityTables() { // HashMap tables = new HashMap<>(16); // tables.put("xz_security_quality_inspection_record", "device_unit_id"); // tables.put("quality_problem", "device_id"); // tables.put("civilize_construction", "device_id"); // tables.put("device_unit", "id"); // return tables; // } public PlainSelect getSqlSegment(PlainSelect plainSelect, Object obj) { UserInfo user = SecurityUtils.getUser(); if (user == null) { return plainSelect; } return dataScopeFilterByProject(plainSelect, user, obj); // else if (type == 2 && user.getAccountType() == 4) { // return dataScopeFilterByProject(plainSelect, user); //} else if (type == 2) { // return dataScopeFilterByEnt(plainSelect, user); //} } private PlainSelect dataScopeFilterByProject(PlainSelect plainSelect, UserInfo user, Object obj) { JSONObject jo = (JSONObject) obj; Object parameter = jo.get("parameter"); DataScope ds = jo.getObject("ds", DataScope.class); this.init(plainSelect); //expressions List expressions = new ArrayList<>(); if (!DataScopeInterceptor.findIgnoreDataScope(parameter, ds)) { // ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); // HttpServletRequest request = requestAttributes.getRequest(); // request.getHeader("webSite"); //项目子账号 if (Objects.equals(user.getAccountType(), SystemUserAccountTypeEnum.PROJECT_SUB_ACCOUNT.getValue())) { List filterEnterprises = getNeedFilterLeftExpression(plainSelect, getFieldEnterpriseTables(), ds); if (CollUtil.isNotEmpty(filterEnterprises)) { List enterpriseIds = userEnterpriseService.getEnterpriseIdsIfSubProject(); for (String filterEnterprise : filterEnterprises) { inExpression(filterEnterprise, enterpriseIds, plainSelect); } } //合作单位ids设备 //and (FIND_IN_SET(100,xxx.enterprise_ids) or FIND_IN_SET(102,xxx.enterprise_ids)) or xxx.enterprise_ids = "" List filterEnterpriseIds = getNeedFilterLeftExpression(plainSelect, this.getFieldEnterpriseIdsTables(), ds); if (CollUtil.isNotEmpty(filterEnterpriseIds)) { List enterpriseIds = userEnterpriseService.getEnterpriseIdsIfSubProject(); // SystemUser su = systemUserService.getOne(new LambdaQueryWrapper().eq(SystemUser::getUserId, SecurityUtils.getUser().getUserId())); for (String filterEnterpriseId : filterEnterpriseIds) { List list1 = enterpriseIds.stream().map(s -> "FIND_IN_SET(" + s + "," + filterEnterpriseId + ")").collect(Collectors.toList()); String s1 = StrUtil.join(" or ", list1); String sql = " (" + s1 + ") or " + filterEnterpriseId + " = \"\""; Expression expression = null; try { expression = CCJSqlParserUtil.parseCondExpression(sql); expressions.add(expression); } catch (JSQLParserException e) { log.error("", e); } } } //合作单位ids设备的数据 List> devIdNames = getNeedFilterLeftExpression(plainSelect, ds, this.getFieldEnterpriseDataTables()); if (CollUtil.isNotEmpty(devIdNames)) { for (Map item : devIdNames) { String aliasName = MapUtils.getString(item, ALIAS_NAME); String fieldName = MapUtils.getString(item, DEV_FIELD); Class clz = (Class) item.get(DEV_MAPPER); Object mapperObj = SpringContextUtils.getBean(clz); Method selectListMethod = ReflectUtil.getMethod(mapperObj.getClass(), "selectList", QueryWrapper.class); Object queryResult = ReflectUtil.invoke(mapperObj, selectListMethod, Wrappers.query()); List list = (List) queryResult; List fieldVals = list.stream().map(o -> Convert.toStr(ReflectUtil.getFieldValue(o, fieldName))).collect(Collectors.toList()); if (CollUtil.isEmpty(fieldVals)) { fieldVals.add("-1"); } inExpression(aliasName, fieldVals, plainSelect); } } List filterItems = getNeedFilterLeftExpression(plainSelect, getFieldVideoTables(), ds); if (CollUtil.isNotEmpty(filterItems)) { List videoItems = userDevAuthorityService.getVideoItemsIfSubProject(); for (String item : filterItems) { inExpression(item, videoItems, plainSelect); } } List filterAis = getNeedFilterLeftExpression(plainSelect, getFieldAiTables(), ds, true); if (CollUtil.isNotEmpty(filterAis)) { SystemUser su = systemUserService.getOne(new LambdaQueryWrapper().eq(SystemUser::getUserId, SecurityUtils.getUser().getUserId())); ProjectConfig projectConfig = projectConfigService.getProjectConfigByProjectSn(su.getSn()); if (projectConfig != null && Objects.equals(projectConfig.getIsEnableRegionEnterprise(), 1)) { List videoItems = aiAnalyseHardWareRecordService.getAiAnalyseHardIdsByUserId(); if (CollUtil.isEmpty(videoItems)) { videoItems.add("0"); } videoItems = videoItems.stream().map(s -> "'" + s + "'").collect(Collectors.toList()); for (String filterAi : filterAis) { String sql = StrUtil.format(" ({}.hardware_id in ({}) OR ({}.quality_region_id in (select distinct quality_region_id from quality_region_to_user where user_id = {}))) ", filterAi, StrUtil.join(",", videoItems), filterAi, SecurityUtils.getUser().getUserId()); Expression expression = null; try { expression = CCJSqlParserUtil.parseCondExpression(sql); expressions.add(expression); } catch (JSQLParserException e) { log.error(e.getMessage(), e); } } } } //安全管理的装置筛选 // List filterDevUnits = getNeedFilterLeftExpression(plainSelect, getFieldSecurityTables(), ds); // if (CollUtil.isNotEmpty(filterDevUnits)) { // List deviceIds = userDeviceService.list(new LambdaQueryWrapper() // .eq(UserDevice::getUserId, SecurityUtils.getUser().getUserId())).stream().map(userDevice -> { // return userDevice.getDeviceId() + ""; // }).collect(Collectors.toList()); // if (CollUtil.isEmpty(deviceIds)) { // deviceIds.add("-1"); // } // for (String filterEnterprise : filterDevUnits) { // inExpression(filterEnterprise, deviceIds, plainSelect); // } // } // //项目组权限 // List groupNames = getNeedFilterLeftExpression(plainSelect, this.getFieldGroupsTables(), ds); // if (CollUtil.isNotEmpty(groupNames)) { // List groupIds = userDevGroupService.getGroupIds(); // for (String item : groupNames) { // inExpression(item, groupIds, plainSelect); // } // } // // //装置权限 // List unitNames = getNeedFilterLeftExpression(plainSelect, this.getFieldDevUnitTables(), ds); // if (CollUtil.isNotEmpty(unitNames)) { // List unitIds = userDevGroupService.getUnitIds(); // for (String item : unitNames) { // inExpression(item, unitIds, plainSelect); // } // } //装置、项目组权限 List> devMap = getNeedFilterLeftExpression(plainSelect, ds, this.getFieldDevProjectTables(), true); if (CollUtil.isNotEmpty(devMap)) { List list1 = userDevGroupService.list(new LambdaQueryWrapper() .eq(UserDevGroup::getUserId, SecurityUtils.getUser().getUserId())); for (Map item : devMap) { String projectField = MapUtils.getString(item, PROJECT_COLUMN); String deviceField = MapUtils.getString(item, DEVICE_COLUMN); String aliasName = MapUtils.getString(item, ALIAS_NAME); List devs = list1.stream().filter(o -> Objects.equals(o.getType(), 1)).collect(Collectors.toList()); if (StrUtil.isNotBlank(projectField)) { //装置和项目组都有 String sql; List s = new ArrayList<>(); for (UserDevGroup dev : devs) { List groupList = list1.stream().filter(o -> o.getNodeId().contains(dev.getDevGroupId().toString()) && Objects.equals(o.getType(), 2)).collect(Collectors.toList()); if (CollUtil.isNotEmpty(groupList)) { for (UserDevGroup group : groupList) { String s1 = StrUtil.format("({}.{}={} and {}.{}={})", aliasName, deviceField, dev.getDevGroupId(), aliasName, projectField, group.getDevGroupId()); s.add(s1); } } } if (CollUtil.isEmpty(s)) { sql = StrUtil.format(" ({}.{}=-1 and {}.{}=-1) ", aliasName, deviceField, aliasName, projectField); } else { sql = "(" + StrUtil.join(" OR ", s) + ")"; } Expression expression = null; try { expression = CCJSqlParserUtil.parseCondExpression(sql); expressions.add(expression); } catch (JSQLParserException e) { log.error(e.getMessage(), e); } } else { //只有装置的 List ss = devs.stream().map(userDevGroup -> userDevGroup.getDevGroupId().toString()).collect(Collectors.toList()); if (CollUtil.isEmpty(ss)) { ss.add("-1"); } this.inExpression(aliasName + "." + deviceField, ss, plainSelect); } } } } // else if (Objects.equals(user.getAccountType(), SystemUserAccountTypeEnum.SUPPLIER.getValue())) { // List filterEnterprises = getNeedFilterLeftExpression(plainSelect, getFieldEnterpriseTables(), ds); // EnterpriseInfo ei = enterpriseInfoMapper.getXzSupplierByUserId(SecurityUtils.getUser().getUserId()); // Long id; // if (ei == null) { // id = -1L; // } else { // id = ei.getId(); // } // for (String filterEnterprise : filterEnterprises) { // String sql = StrUtil.format(" ({} = {} OR {} IN ( SELECT DISTINCT t.enterprise_id FROM " + // "(SELECT t.id FROM project_enterprise t WHERE t.enterprise_id = {}) t2 join project_enterprise t on find_in_set( t2.id, ancestors ) )) ", // filterEnterprise, id, filterEnterprise, id); // Expression expression = null; // try { // expression = CCJSqlParserUtil.parseCondExpression(sql); // expressions.add(expression); // } catch (JSQLParserException e) { // log.error(e.getMessage(), e); // } // } // // //解析ai预警 // List filterAis = getNeedFilterLeftExpression(plainSelect, getFieldAiTables(), ds, true); // if (CollUtil.isNotEmpty(filterAis)) { // List videoItems = aiAnalyseHardWareRecordService.getAiAnalyseHardIdsByEnterpriseId(id); // if (CollUtil.isEmpty(videoItems)) { // videoItems.add("0"); // } // videoItems = videoItems.stream().map(s -> "'" + s + "'").collect(Collectors.toList()); // for (String filterAi : filterAis) { // String sql = StrUtil.format(" ({}.hardware_id in ({}) OR ({}.quality_region_id in (select distinct quality_region_id from quality_region_to_user where user_id = {}))) ", // filterAi, StrUtil.join(",", videoItems), filterAi, SecurityUtils.getUser().getUserId()); // Expression expression = null; // try { // expression = CCJSqlParserUtil.parseCondExpression(sql); // expressions.add(expression); // } catch (JSQLParserException e) { // log.error(e.getMessage(), e); // } // } // } // // } if (expressions.size() > 0) { Expression dataExpression; if (expressions.size() > 1) { //数据权限大于1个，之间用或 OrExpression orExpression = new OrExpression(expressions.get(0), expressions.get(1)); int i1 = 2; for (int i = i1; i < expressions.size(); i++) { orExpression = new OrExpression(orExpression, expressions.get(i)); } // 数据权限使用单独的括号 防止与其他条件冲突 dataExpression = new Parenthesis(orExpression); } else { dataExpression = expressions.get(0); } Expression where = plainSelect.getWhere(); if (where != null) { where = new AndExpression(where, dataExpression); } else { where = dataExpression; } plainSelect.setWhere(where); } } if (!DataScopeInterceptor.isNotSqlTest()) { String sql = " ('1qqq')"; try { Expression expression = CCJSqlParserUtil.parseCondExpression(sql); Expression where = plainSelect.getWhere(); if (where != null) { where = new AndExpression(where, expression); } else { where = expression; } plainSelect.setWhere(where); } catch (JSQLParserException e) { throw new RuntimeException(e); } } return plainSelect; } /** * 获取需要过滤的表别名或加字段 * * @param plainSelect * @param tables * @param ds * @param onlyAlas false：表别名.字段 true：表别名 * @return */ private List getNeedFilterLeftExpression(PlainSelect plainSelect, Map tables, DataScope ds, boolean onlyAlas) { ArrayList rtList = new ArrayList<>(); String[] dsArr = ds.includeTable(); if (dsArr.length > 0) { Map nt = new HashMap<>(16); for (String ds1 : dsArr) { if (tables.containsKey(ds1)) { nt.put(ds1, tables.get(ds1)); } } tables = nt; } FromItem fromItem = plainSelect.getFromItem(); if (fromItem instanceof Table) { Table table = (Table) fromItem; String name = table.getName(); if (tables.get(name) != null) { String aliasName = null; Alias alias = table.getAlias(); if (alias != null) { aliasName = alias.getName(); } String e; if (onlyAlas) { e = (StringUtils.isEmpty(aliasName) ? name : aliasName); } else { e = (StringUtils.isEmpty(aliasName) ? name : aliasName) + "." + tables.get(name); } rtList.add(e); } } List joins = plainSelect.getJoins(); //主表不是user表 if (!CollectionUtils.isEmpty(joins)) { //判断join的表里有没有user表 for (Join join : joins) { // 判断join里面是否存在user表，不存在则新增 FromItem rightItem = join.getRightItem(); if (rightItem instanceof Table) { Table joinTable = (Table) rightItem; if (tables.get(joinTable.getName()) != null) { String aliasName = null; Alias alias = joinTable.getAlias(); if (alias != null) { aliasName = alias.getName(); } String e; if (onlyAlas) { e = (StringUtils.isEmpty(aliasName) ? joinTable.getName() : aliasName); } else { e = (StringUtils.isEmpty(aliasName) ? joinTable.getName() : aliasName) + "." + tables.get(joinTable.getName()); } rtList.add(e); } } } } return rtList; } //private PlainSelect dataScopeFilterByEnt(PlainSelect plainSelect, SecurityUser user) { // init(plainSelect); // if (user.getAccountType() == 3) { // equalsTo(getAliasColumn(plainSelect, getEnterpriseSn()), user.getSn(), plainSelect); // } // if (user.getAccountType() == 2) { // List entSns = enterpriseService.getSnListForGov(user.getSn()); // if (entSns.size() == 0) { // entSns.add("0"); // } // inExpression(getAliasColumn(plainSelect, getEnterpriseSn()), entSns, plainSelect); // } // return plainSelect; //} private List getNeedFilterLeftExpression(PlainSelect plainSelect, Map tables, DataScope ds) { return getNeedFilterLeftExpression(plainSelect, tables, ds, false); } private List> getNeedFilterLeftExpression(PlainSelect plainSelect, DataScope ds, Map> param) { return getNeedFilterLeftExpression(plainSelect, ds, param, false); } private List> getNeedFilterLeftExpression(PlainSelect plainSelect, DataScope ds, Map> param, boolean onlyAlas) { ArrayList> rtList = new ArrayList<>(); String[] dsArr = ds.includeTable(); if (dsArr.length > 0) { Map> nt = new HashMap<>(16); for (String ds1 : dsArr) { if (param.containsKey(ds1)) { nt.put(ds1, param.get(ds1)); } } param = nt; } FromItem fromItem = plainSelect.getFromItem(); if (fromItem instanceof Table) { Table table = (Table) fromItem; String name = table.getName(); if (param.get(name) != null) { String aliasName = null; Alias alias = table.getAlias(); if (alias != null) { aliasName = alias.getName(); } Map map = param.get(name); String e; if (onlyAlas) { e = (StringUtils.isEmpty(aliasName) ? name : aliasName); } else { e = (StringUtils.isEmpty(aliasName) ? name : aliasName) + "." + map.get(DATA_COLUMN).toString(); } map.put(ALIAS_NAME, e); rtList.add(map); } } List joins = plainSelect.getJoins(); //主表不是user表 if (!CollectionUtils.isEmpty(joins)) { //判断join的表里有没有user表 for (Join join : joins) { // 判断join里面是否存在user表，不存在则新增 FromItem rightItem = join.getRightItem(); if (rightItem instanceof Table) { Table joinTable = (Table) rightItem; if (param.get(joinTable.getName()) != null) { String aliasName = null; Alias alias = joinTable.getAlias(); if (alias != null) { aliasName = alias.getName(); } Map map = param.get(aliasName); String e; if (onlyAlas) { e = (StringUtils.isEmpty(aliasName) ? joinTable.getName() : aliasName); } else { e = (StringUtils.isEmpty(aliasName) ? joinTable.getName() : aliasName) + "." + param.get(aliasName).get(DATA_COLUMN).toString(); } map.put(ALIAS_NAME, e); rtList.add(map); } } } } return rtList; } private void equalsTo(String leftExpression, String rightExpression, PlainSelect plainSelect) { EqualsTo equalsTo = new EqualsTo(); equalsTo.setLeftExpression(new Column(leftExpression)); equalsTo.setRightExpression(new StringValue(rightExpression)); AndExpression andExpression = new AndExpression(plainSelect.getWhere(), equalsTo); plainSelect.setWhere(andExpression); } private void inExpression(String leftExpression, List rightExpression, PlainSelect plainSelect) { InExpression inExpression = new InExpression(); ItemsList itemsList = new ExpressionList(rightExpression.stream().map(StringValue::new).collect(Collectors.toList())); inExpression.setLeftExpression(new Column(leftExpression)); inExpression.setRightItemsList(itemsList); AndExpression andExpression = new AndExpression(plainSelect.getWhere(), inExpression); plainSelect.setWhere(andExpression); } private String getAliasColumn(PlainSelect plainSelect, String sn) { FromItem fromItem = plainSelect.getFromItem(); Alias alias = fromItem.getAlias(); StringBuilder prefix = new StringBuilder(); if (alias != null) { prefix.append(alias).append("."); } return prefix.append(sn).toString(); } //protected void addParam(Insert insert, Object obj){ // SecurityUser user = SecurityUtil.getUser(); // if (user == null) { // return; // } // List columns = insert.getColumns(); // if (CollectionUtils.isEmpty(columns)) { // // 针对不给列名的insert 不处理 // return; // } // if (user.getAccountType() == 4 || user.getAccountType() == 3) { // if (!columns.stream().anyMatch(c -> c.toString().equals(getProjectSn()))) { // columns.add(new Column(getProjectSn())); // if (insert.getItemsList() != null) { // ItemsList itemsList = insert.getItemsList(); // if (itemsList instanceof MultiExpressionList) { // ((MultiExpressionList) itemsList).getExprList().forEach(el -> el.getExpressions().add(getProjectSn(obj))); // } else { // ((ExpressionList) itemsList).getExpressions().add(getProjectSn(obj)); // } // } else { // throw ExceptionUtils.mpe("Failed to process multiple-table update, please exclude the tableName or statementId"); // } // } // } //} private void init(PlainSelect plainSelect) { Expression envCondition = null; try { envCondition = CCJSqlParserUtil.parseCondExpression("1 == 1"); } catch (JSQLParserException e) { log.error(e.getMessage(), e); } Expression where = plainSelect.getWhere(); if (where == null) { plainSelect.setWhere(envCondition); } } }