From bb6856a82a588f05040b0410a9747ae8029bb3be Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Fri, 13 Oct 2023 10:06:51 +0800
Subject: [PATCH] Merge branch 'dev' of http://blit.7drlb.com:8888/r/moral into wb

---
 screen-api/src/main/java/com/moral/api/service/impl/AllocationServiceImpl.java |   54 ++++++++++++++++++++++++++++++++++--------------------
 1 files changed, 34 insertions(+), 20 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/AllocationServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/AllocationServiceImpl.java
index 7a8b8f9..5949515 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/AllocationServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/AllocationServiceImpl.java
@@ -211,32 +211,46 @@
     public Map<String, Object> selectUnitView(Map<String, Object> map) {
         QueryWrapper<Allocation> wrapper = new QueryWrapper<>();
         //������������������
-        Map<String, Object> userInfo = (Map<String, Object>) TokenUtils.getUserInfo();
-        Object unitId = userInfo.get("unitId");
-        Map<String, Object> orgInfo = (Map<String, Object>) userInfo.get("organization");
-        Integer orgId = (Integer) orgInfo.get("id");
+        List<Integer> list = this.unitResult();
         if (!ObjectUtils.isEmpty(map.get("unitId"))){
             wrapper.in("unit_id",map.get("unitId").toString());
         }else {
-            if (orgId!=24){
-                if (ObjectUtils.isEmpty(unitId)){
+            if (!ObjectUtils.isEmpty(list)){
+                Integer integer = list.get(0);
+                if (integer==0){
                     return null;
-                }
-                ResponsibilityUnit responsibilityUnit1 = responsibilityUnitMapper.selectById(Integer.parseInt(unitId.toString()));
-                Integer areaCode = responsibilityUnit1.getAreaCode();
-                List<ResponsibilityUnit> responsibilityUnits = responsibilityUnitService.selectUnit(areaCode);
-                if (ObjectUtils.isEmpty(responsibilityUnits)){
-                    List<ResponsibilityUnit> responsibilityUnits1 = responsibilityUnitService.selectAreaUnit(areaCode);
-                    Map<Integer, List<ResponsibilityUnit>> collect = responsibilityUnits1.stream().collect(Collectors.groupingBy(o -> o.getUnitId()));
-                    List<Integer> collect1 = collect.keySet().stream().collect(Collectors.toList());
-                    wrapper.in("unit_id",collect1);
                 }else {
-                    Map<Integer, List<ResponsibilityUnit>> collect = responsibilityUnits.stream().collect(Collectors.groupingBy(o -> o.getUnitId()));
-                    List<Integer> collect1 = collect.keySet().stream().collect(Collectors.toList());
-                    wrapper.in("unit_id",collect1);
+                    wrapper.in("unit_id",list);
                 }
             }
         }
+
+//        Map<String, Object> userInfo = (Map<String, Object>) TokenUtils.getUserInfo();
+//        Object unitId = userInfo.get("unitId");
+//        Map<String, Object> orgInfo = (Map<String, Object>) userInfo.get("organization");
+//        Integer orgId = (Integer) orgInfo.get("id");
+//        if (!ObjectUtils.isEmpty(map.get("unitId"))){
+//            wrapper.in("unit_id",map.get("unitId").toString());
+//        }else {
+//            if (orgId!=24){
+//                if (ObjectUtils.isEmpty(unitId)){
+//                    return null;
+//                }
+//                ResponsibilityUnit responsibilityUnit1 = responsibilityUnitMapper.selectById(Integer.parseInt(unitId.toString()));
+//                Integer areaCode = responsibilityUnit1.getAreaCode();
+//                List<ResponsibilityUnit> responsibilityUnits = responsibilityUnitService.selectUnit(areaCode);
+//                if (ObjectUtils.isEmpty(responsibilityUnits)){
+//                    List<ResponsibilityUnit> responsibilityUnits1 = responsibilityUnitService.selectAreaUnit(areaCode);
+//                    Map<Integer, List<ResponsibilityUnit>> collect = responsibilityUnits1.stream().collect(Collectors.groupingBy(o -> o.getUnitId()));
+//                    List<Integer> collect1 = collect.keySet().stream().collect(Collectors.toList());
+//                    wrapper.in("unit_id",collect1);
+//                }else {
+//                    Map<Integer, List<ResponsibilityUnit>> collect = responsibilityUnits.stream().collect(Collectors.groupingBy(o -> o.getUnitId()));
+//                    List<Integer> collect1 = collect.keySet().stream().collect(Collectors.toList());
+//                    wrapper.in("unit_id",collect1);
+//                }
+//            }
+//        }
 
         HashMap<String, Object> rsMap = new HashMap<>();
         ArrayList<Map<String, Object>> polluteArrayList = new ArrayList<>();
@@ -612,7 +626,7 @@
     public List<Integer>  unitResult() {
         QxUser user = UserHelper.getCurrentUser();
         Integer unitId = Objects.nonNull(user.getUnitId())?user.getUnitId():0;
-        List<Integer> userList = responsibilityUnitMapper.selectCodeList("JBD",user.getUserId());
+        List<Integer> userList = responsibilityUnitMapper.selectCodeList(SysDictTypeEnum.SYS_JBD.getValue(),user.getUserId());
         if(CollectionUtils.isNotEmpty(userList)){
             return new ArrayList<>();
         }
@@ -626,7 +640,7 @@
     @Override
     public Integer unitAreaCode() {
         QxUser user = UserHelper.getCurrentUser();
-        List<Integer> userList = responsibilityUnitMapper.selectCodeList("JBD",user.getUserId());
+        List<Integer> userList = responsibilityUnitMapper.selectCodeList(SysDictTypeEnum.SYS_JBD.getValue(),user.getUserId());
         if(CollectionUtils.isNotEmpty(userList)){
             return 1;
         }

--
Gitblit v1.8.0