From 15369bf819ca82896bbcada782e6babf2a7d64c9 Mon Sep 17 00:00:00 2001
From: cjl <276999030@qq.com>
Date: Tue, 14 Nov 2023 13:49:26 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/cjl' into qa

---
 screen-api/src/main/java/com/moral/api/controller/AllocationController.java |  107 +++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 87 insertions(+), 20 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/controller/AllocationController.java b/screen-api/src/main/java/com/moral/api/controller/AllocationController.java
index 32958bf..86f968a 100644
--- a/screen-api/src/main/java/com/moral/api/controller/AllocationController.java
+++ b/screen-api/src/main/java/com/moral/api/controller/AllocationController.java
@@ -1,26 +1,47 @@
 package com.moral.api.controller;
 
 
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiImplicitParam;
+import io.swagger.annotations.ApiImplicitParams;
+import io.swagger.annotations.ApiOperation;
+import org.apache.commons.collections4.CollectionUtils;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.moral.api.entity.AllocationLog;
+import com.moral.api.pojo.enums.SysDictTypeEnum;
 import com.moral.api.pojo.ext.allocation.AllocationExt;
 import com.moral.api.pojo.ext.allocation.AllocationPageExt;
 import com.moral.api.pojo.query.allocation.*;
 import com.moral.api.pojo.query.allocationextension.AllocationExtensionAddCond;
+
 import com.moral.api.pojo.vo.allocation.AllocationPageVo;
 import com.moral.api.pojo.vo.allocation.AllocationVo;
+
 import com.moral.constant.PageResult;
 import io.swagger.annotations.*;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.util.ObjectUtils;
+
 import org.springframework.web.bind.annotation.*;
+
+import java.util.ArrayList;
 
 import java.util.List;
 import java.util.Map;
 
-import com.moral.api.entity.Allocation;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
 import com.moral.api.entity.ResponsibilityUnit;
+
 import com.moral.api.service.AllocationService;
+import com.moral.api.utils.EasyExcelUtils;
+import com.moral.api.utils.NoModelWriteData;
 import com.moral.constant.Constants;
 import com.moral.constant.ResultMessage;
+import com.moral.util.DateUtils;
+import com.moral.util.WebUtils;
+
 
 import javax.validation.Valid;
 
@@ -44,7 +65,6 @@
         return ResultMessage.ok(professions);
     }
 
-
     @ApiOperation(value = "������������", notes = "������������")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String")
@@ -54,6 +74,7 @@
         List<ResponsibilityUnit> responsibilityUnits = allocationService.seleteUnit();
         return ResultMessage.ok(responsibilityUnits);
     }
+
 
     /**
      * ���������������
@@ -67,15 +88,7 @@
     }
 
 
-    /**
-     * ���������������
-     * @return
-     */
-   /* @GetMapping("check")
-    public ResultMessage check(Integer id){
-        allocationService.check(id);
-        return ResultMessage.ok();
-    }*/
+
 
     /**
      * ������������
@@ -104,15 +117,7 @@
         allocationService.checkAllocation(checkCond);
         return ResultMessage.ok();
     }
-    /**
-     * ������������������
-     * @return
-     */
-    @GetMapping("selectAll")
-    public ResultMessage selectAll(Map<String,Object> map){
-        allocationService.selectAll(map);
-        return ResultMessage.ok();
-    }
+
 
     @PostMapping("/page")
     @ApiOperation("������")
@@ -131,6 +136,7 @@
         allocationVo.setFileChangeList(allocationext.getFileChangeList());
         allocationVo.setFileApproveList(allocationext.getFileApproveList());
         allocationVo.setApproveList(allocationext.getApproveList());
+        allocationVo.setTime(DateUtils.dateToDateString(allocationVo.getEscalationTime(), DateUtils.yyyy_MM_dd_CN));
         return ResultMessage.ok(allocationVo);
     }
     @GetMapping("remove")
@@ -153,4 +159,65 @@
         return allocationService.applyFor(allocationExtensionAddCond) ? ResultMessage.ok() : ResultMessage.fail();
     }
 
+
+    /**
+     * ������������������
+     * @return
+     */
+    @GetMapping("selectUnitView")
+    @ApiOperation("������������")
+    public ResultMessage selectUnitView(HttpServletRequest request){
+        Map<String, Object> params = WebUtils.getParametersStartingWith(request, null);
+        Map<String, Object> map1 = allocationService.selectUnitView(params);
+        return ResultMessage.ok(ObjectUtils.isEmpty(map1)? "0":map1);
+    }
+
+
+    @GetMapping("unitExel")
+    public void unitExel(HttpServletResponse response,HttpServletRequest request){
+        Map<String, Object> params = WebUtils.getParametersStartingWith(request, null);
+        //������������
+        List<Map<String, Object>> mapArrayList = allocationService.unitExel(params);
+        if (CollectionUtils.isEmpty(mapArrayList)) {
+            return;
+        }
+        Map<String, Object> map = mapArrayList.get(0);
+        List<String> list = new ArrayList<>();
+        for (String key : map.keySet()) {
+            list.add(key);
+        }
+        String[] s2 = new String[list.size()];
+        list.toArray(s2);
+        NoModelWriteData d = new NoModelWriteData();
+        d.setFileName("������������");
+        d.setHeadMap(s2);
+        d.setDataStrMap(s2);
+        d.setDataList(mapArrayList);
+        try {
+            EasyExcelUtils easyExcelUtils = new EasyExcelUtils();
+            easyExcelUtils.noModleWrite(d, response);
+        } catch (Exception e) {
+            int i = 0;
+        }
+    }
+
+    @GetMapping("updataUnit")
+    @ApiOperation("������������������")
+    public ResultMessage updataUnit(@RequestParam @ApiParam(value = "id",name = "������id") Integer id,
+                                    @RequestParam @ApiParam(value = "unitId",name = "������������Id") Integer unitId,
+                                    @RequestParam @ApiParam(value = "polluteType",name = "������������Id") Integer polluteType){
+        allocationService.updataUnit(id,unitId,polluteType);
+        return ResultMessage.ok();
+    }
+
+
+
+    @GetMapping("getLog")
+    @ApiOperation("������������")
+    public ResultMessage getLog(@RequestParam @ApiParam(value = "allocationNum",name = "������������") String allocationNum){
+        List<AllocationLog> log = allocationService.getLog(allocationNum);
+        return ResultMessage.ok(log);
+    }
+
+
 }

--
Gitblit v1.8.0