From 2ba93435afce9b04a5247739041902861b20596c Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Tue, 24 Oct 2023 10:24:26 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/wb' into qa

---
 screen-api/src/main/java/com/moral/api/service/impl/AllocationServiceImpl.java |   67 +++++++++++++++++++--------------
 1 files changed, 39 insertions(+), 28 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 003041c..1dbffad 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
@@ -64,6 +64,8 @@
     private AllocationExtensionService allocationExtensionService;
     @Autowired
     private ResponsibilityUnitService responsibilityUnitService;
+    @Autowired
+    private SysDictTypeService sysDictTypeService;
 
     /**
      * ������������������������������������
@@ -280,13 +282,14 @@
             escalationTypeList.add(typeMap);
         }
         //������������������������
+        SysDictType sysDictType = sysDictTypeService.slectFind(Constants.WU_RAN_LEI_XING);
         Map<Integer, List<Allocation>> polluteTypeMap = allocations.stream().collect(Collectors.groupingBy(o -> o.getPolluteType()));
         Set<Integer> polluteTypes = polluteTypeMap.keySet();
         for (Integer polluteType : polluteTypes) {
             HashMap<String, Object> typeMap = new HashMap<>();
             List<Allocation> allocations1 = polluteTypeMap.get(polluteType);
             QueryWrapper<SysDictData> sysDictDataQueryWrapper = new QueryWrapper<>();
-            sysDictDataQueryWrapper.eq("dict_type_id",32);
+            sysDictDataQueryWrapper.eq("dict_type_id",sysDictType.getId());
             sysDictDataQueryWrapper.eq("dataKey",polluteType);
             SysDictData sysDictData = sysDictDataMapper.selectOne(sysDictDataQueryWrapper);
             typeMap.put("name",sysDictData.getDataValue());
@@ -617,57 +620,65 @@
         //������������������
         List<Integer> list = unitResult();
         if (!ObjectUtils.isEmpty(list)){
-            Integer integer = list.get(0);
-            if (integer==0){
-                return null;
-            }else {
-                wrapper.in(Allocation::getUnitId,list);
-            }
+            wrapper.in(Allocation::getUnitId,list);
         }
         Integer integer = unitAreaCode();
-        //��������������� ���������
-        if (integer==0 || integer==1){
+        // ���������
+        if (integer==1){
             if (state==3){
-                wrapper.in(Allocation::getState,40,50);
-            }else if (state==2){
-                wrapper.eq(Allocation::getState,30);
+                wrapper.in(Allocation::getState,AllocationApproveEnum.PASS.value,AllocationApproveEnum.REFUSE.value);
             }else if (state==1){
-                wrapper.eq(Allocation::getState,20);
+                wrapper.in(Allocation::getState,AllocationApproveEnum.UNDER_RECTIFICATION.value,AllocationApproveEnum.DRAFT.value,AllocationApproveEnum.IN_APPROVAL.value);
+            }else {
+                wrapper.eq(Allocation::getState,0);
             }
-        //������������
-        }else if (integer==2){
+        }else if(integer == 0) {
             if (state==3){
-                wrapper.in(Allocation::getState,40,50);
+                wrapper.in(Allocation::getState,AllocationApproveEnum.PASS.value,AllocationApproveEnum.REFUSE.value);
             }else if (state==2){
-                wrapper.eq(Allocation::getState,20);
+                wrapper.eq(Allocation::getState,AllocationApproveEnum.IN_APPROVAL.value);
             }else if (state==1){
-                wrapper.eq(Allocation::getState,30);
+                wrapper.in(Allocation::getState,AllocationApproveEnum.UNDER_RECTIFICATION.value,AllocationApproveEnum.DRAFT.value);
+            }
+        }else if(integer == 2){
+            if (state==3){
+                wrapper.in(Allocation::getState,AllocationApproveEnum.PASS.value,AllocationApproveEnum.REFUSE.value);
+            }else if (state==2){
+                wrapper.in(Allocation::getState,AllocationApproveEnum.UNDER_RECTIFICATION.value,AllocationApproveEnum.DRAFT.value);
+            }else if (state==1){
+                wrapper.eq(Allocation::getState,AllocationApproveEnum.IN_APPROVAL.value);
             }
         }
         if (!ObjectUtils.isEmpty(startTime) && !ObjectUtils.isEmpty(endTime)){
             wrapper.between(Allocation::getEscalationTime,startTime,endTime);
         }
         wrapper.eq(Allocation::getIsDel,0).eq(Allocation::getIsInvalid,0);
-        wrapper.orderByDesc(Allocation::getEscalationTime);
+        wrapper.orderByDesc(Allocation::getEscalationTime,Allocation::getAllocationId);
         List<Allocation> allocations = allocationMapper.selectList(wrapper);
         for (Allocation allocation : allocations) {
             AllocationFindVo allocationFindVo = new AllocationFindVo();
             BeanUtils.copyProperties(allocation,allocationFindVo);
             Integer state1 = allocationFindVo.getState();
-            if (integer==0 || integer==1){
-                if (state1==20){
+            if ( integer==2){
+                if (state1.equals(AllocationApproveEnum.DRAFT.value)||state1.equals(AllocationApproveEnum.UNDER_RECTIFICATION.value)){
+                    allocationFindVo.setStateName(AppAllocationStateEnum.TO_BE_COMPLETED.name);
+                }else if (state1.equals(AllocationApproveEnum.IN_APPROVAL.value)){
                     allocationFindVo.setStateName(AppAllocationStateEnum.TO_BE_PROCESSED.name);
-                }else if (state1==30){
-                    allocationFindVo.setStateName(AppAllocationStateEnum.TO_BE_PROCESSED.name);
-                }else if (state1==40 ||state1==50){
+                }else if (state1.equals(AllocationApproveEnum.PASS.value)||state1.equals(AllocationApproveEnum.REFUSE.value)){
                     allocationFindVo.setStateName(AppAllocationStateEnum.COMPLETED.name);
                 }
-            }else {
-                if (state1==20){
+            }else if(integer==1){
+                if (state1.equals(AllocationApproveEnum.DRAFT.value)||state1.equals(AllocationApproveEnum.UNDER_RECTIFICATION.value)||state1.equals(AllocationApproveEnum.IN_APPROVAL.value)){
                     allocationFindVo.setStateName(AppAllocationStateEnum.TO_BE_PROCESSED.name);
-                }else if (state1==30){
+                }else if (state1.equals(AllocationApproveEnum.PASS.value)||state1.equals(AllocationApproveEnum.REFUSE.value)){
+                    allocationFindVo.setStateName(AppAllocationStateEnum.COMPLETED.name);
+                }
+            }else if(integer==0) {
+                if (state1.equals(AllocationApproveEnum.DRAFT.value)||state1.equals(AllocationApproveEnum.UNDER_RECTIFICATION.value)){
                     allocationFindVo.setStateName(AppAllocationStateEnum.TO_BE_PROCESSED.name);
-                }else if (state1==40||state1==50){
+                }else if (state1.equals(AllocationApproveEnum.IN_APPROVAL.value)){
+                    allocationFindVo.setStateName(AppAllocationStateEnum.TO_BE_COMPLETED.name);
+                }else if (state1.equals(AllocationApproveEnum.PASS.value)||state1.equals(AllocationApproveEnum.REFUSE.value)){
                     allocationFindVo.setStateName(AppAllocationStateEnum.COMPLETED.name);
                 }
             }

--
Gitblit v1.8.0