admin/wisdomisite-java
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

白色不要做ai和车辆的数据权限

Browse Source
This commit is contained in:
GUO 2024-05-18 20:00:26 +08:00
parent bb61add63c
commit 640fe5c635
2 changed files with 24 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/com/zhgd/mybatis/DataScopeHandler.java
View File

@ -13,6 +13,7 @@ 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 lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.expression.Alias;
@ -44,6 +45,9 @@ import java.util.stream.Collectors;
@Slf4j
public class DataScopeHandler implements DataPermissionHandler {
@Lazy
@Autowired
EnvironmentUtil environmentUtil;
@Lazy
@Autowired
IAiAnalyseHardWareRecordService aiAnalyseHardWareRecordService;
@ -85,15 +89,17 @@ public class DataScopeHandler implements DataPermissionHandler {
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("enterprise_info", "id");
tables.put("quality_inspection_record", "enterprise_id");
tables.put("worker_info_audit_record", "enterprise_id");
tables.put("car_info", "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("team_info", "enterprise_id");
if (!environmentUtil.isBaise()) {
tables.put("car_info", "enterprise_id");
tables.put("enterprise_info", "id");
}
return tables;
}
@ -105,7 +111,9 @@ public class DataScopeHandler implements DataPermissionHandler {
private HashMap<String, String> getFieldAiTables() {
HashMap<String, String> tables = new HashMap<>();
tables.put("ai_analyse_hard_ware_alarm_record", "hardware_id");
if (!environmentUtil.isBaise()) {
tables.put("ai_analyse_hard_ware_alarm_record", "hardware_id");
}
return tables;
}

17
src/main/java/com/zhgd/xmgl/util/EnvironmentUtil.java
View File

@ -17,15 +17,15 @@ public class EnvironmentUtil {
*/
private static String activeEnvironment;
public static String getActiveEnvironment() {
return activeEnvironment;
}
@Value("${active.environment.name}")
public void setActiveEnvironment(String activeEnvironment) {
EnvironmentUtil.activeEnvironment = activeEnvironment;
}
public static String getActiveEnvironment() {
return activeEnvironment;
}
/**
* 是否本地开发环境
*
@ -125,4 +125,13 @@ public class EnvironmentUtil {
return Stream.of("gxlt-prod", "gxlt-dev").anyMatch(s -> s.equals(activeEnvironment));
}
/**
* 是否baise环境
*
* @return
*/
public boolean isBaise() {
return Stream.of("baise").anyMatch(s -> s.equals(activeEnvironment));
}
}