From 27e6bc3df3e39e0d0b147b155a89ad6837ea972b Mon Sep 17 00:00:00 2001
From: cjl <909710561@qq.com>
Date: Mon, 06 Jan 2025 09:19:24 +0800
Subject: [PATCH] Merge branch 'cjl' into dev

---
 screen-api/src/main/java/com/moral/api/controller/AllocationController.java |  273 +++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 227 insertions(+), 46 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 a576d1a..df08b55 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,36 +1,62 @@
 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.aspectj.apache.bcel.generic.RET;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestMethod;
-import org.springframework.web.bind.annotation.RestController;
-
-import java.util.List;
-import java.util.Map;
-
-import com.moral.api.entity.Allocation;
+import com.alibaba.excel.EasyExcel;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.moral.api.entity.AllocationLog;
 import com.moral.api.entity.ResponsibilityUnit;
+import com.moral.api.pojo.enums.AllocationApproveEnum;
+import com.moral.api.pojo.enums.ChangeEnum;
+import com.moral.api.pojo.enums.YesOrNo;
+import com.moral.api.pojo.ext.allocation.AllocationExcelExt;
+import com.moral.api.pojo.ext.allocation.AllocationExt;
+import com.moral.api.pojo.ext.allocation.AllocationPageExt;
+import com.moral.api.pojo.query.ResponsibilityUnitQuery;
+import com.moral.api.pojo.query.allocation.*;
+import com.moral.api.pojo.query.allocationextension.AllocationExtensionAddCond;
+import com.moral.api.pojo.vo.allocation.AllocationListExcelVo;
+import com.moral.api.pojo.vo.allocation.AllocationPageVo;
+import com.moral.api.pojo.vo.allocation.AllocationVo;
+import com.moral.api.pojo.vo.file.FileAddressVo;
 import com.moral.api.service.AllocationService;
+import com.moral.api.service.ResponsibilityUnitService;
+import com.moral.api.utils.*;
 import com.moral.constant.Constants;
+import com.moral.constant.PageResult;
 import com.moral.constant.ResultMessage;
+import com.moral.util.DateUtils;
+import com.moral.util.WebUtils;
+import io.swagger.annotations.*;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.CollectionUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.util.ObjectUtils;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import javax.validation.Valid;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.net.URL;
+import java.net.URLEncoder;
+import java.util.*;
+import java.util.concurrent.atomic.AtomicReference;
 
 
 @Api(tags = {"������������"})
 @RestController
 @RequestMapping("allocation")
+@Slf4j
 public class AllocationController {
 
+    @Value("${file.path}")
+    private String basePath;
     @Autowired
     private AllocationService allocationService;
+    @Autowired
+    private ResponsibilityUnitService responsibilityUnitService;
 
 
     @ApiOperation(value = "������������", notes = "������������")
@@ -43,18 +69,6 @@
         return ResultMessage.ok(professions);
     }
 
-
-    @ApiOperation(value = "������������", notes = "������������")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String")
-    })
-    @RequestMapping(value = "reform", method = RequestMethod.GET)
-    public ResultMessage reform() {
-        List<Map<String, Object>> professions = allocationService.sysDictData(Constants.ZHENG_GAI_LEI_XING);
-        return ResultMessage.ok(professions);
-    }
-
-
     @ApiOperation(value = "������������", notes = "������������")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String")
@@ -65,46 +79,213 @@
         return ResultMessage.ok(responsibilityUnits);
     }
 
+
     /**
      * ���������������
      * @return
      */
     @PostMapping("insert")
-    public  ResultMessage insert(@RequestBody Allocation allocation){
+    @ApiOperation("������")
+    public  ResultMessage insert(@Valid @RequestBody AllocationAddCond allocation){
         allocationService.insertAllocation(allocation);
         return ResultMessage.ok();
     }
 
 
-    /**
-     * ���������������
-     * @return
-     */
-    @GetMapping("check")
-    public ResultMessage check(Integer id){
-        allocationService.check(id);
-        return ResultMessage.ok();
-    }
+
 
     /**
      * ������������
-     * @param allocation
+     * @param allocationUpdateCond
      * @return
+     *
+     *
      */
     @PostMapping("update")
-    public ResultMessage update(@RequestBody Allocation allocation){
-        allocationService.updateAll(allocation);
+    @ApiOperation("������")
+
+    public ResultMessage update(@Valid @RequestBody AllocationUpdateCond allocationUpdateCond){
+        allocationService.updateAll(allocationUpdateCond);
         return ResultMessage.ok();
     }
 
+    @PostMapping("change")
+    @ApiOperation("������")
+    public ResultMessage changeAllocation(@Valid @RequestBody AllocationChangeCond changeCond){
+        allocationService.changeAllocation(changeCond);
+        return ResultMessage.ok();
+    }
+
+    @PostMapping("check")
+    @ApiOperation("������")
+    public ResultMessage checkAllocation(@Valid @RequestBody AllocationCheckCond checkCond){
+        allocationService.checkAllocation(checkCond);
+        return ResultMessage.ok();
+    }
+
+
+    @PostMapping("/page")
+    @ApiOperation("������")
+    public ResultMessage<PageResult<AllocationPageVo>> page(@Valid @RequestBody AllocationPageCond allocationPageCond) {
+        Page<AllocationPageExt> page = allocationService.extPage(allocationPageCond);
+        PageResult<AllocationPageVo> result = new PageResult<>(page);
+        result.setList(AllocationPageVo.convert(page.getRecords()));
+        return ResultMessage.ok(result);
+    }
+    @GetMapping("/detail")
+    @ApiOperation("������")
+    public ResultMessage<AllocationVo> get(Integer id) {
+        AllocationExt allocationext = allocationService.extOne(id);
+        AllocationVo allocationVo = AllocationVo.convert(allocationext);
+        allocationVo.setFileBaseList(allocationext.getFileBaseList());
+        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")
+    @ApiOperation("������")
+    public ResultMessage removeById(Integer id){
+        allocationService.removeById(id);
+        return ResultMessage.ok();
+    }
+    @GetMapping("invalid")
+    @ApiOperation("������")
+    public ResultMessage invalidResult(@RequestParam @ApiParam(value = "id",name = "������id") Integer id,
+                                       @RequestParam @ApiParam(value = "invalidReason",name = "������������") String invalidReason){
+        allocationService.invalidResult(id,invalidReason);
+        return ResultMessage.ok();
+    }
+
+    @PostMapping("/applyfor")
+    @ApiOperation("������������")
+    public ResultMessage applyFor(@Valid @RequestBody AllocationExtensionAddCond allocationExtensionAddCond) {
+        return allocationService.applyFor(allocationExtensionAddCond) ? ResultMessage.ok() : ResultMessage.fail();
+    }
+
+
     /**
-     * ������������������
+     * ������������������
      * @return
      */
-    @GetMapping("selectAll")
-    public ResultMessage selectAll(Map<String,Object> map){
-        allocationService.selectAll(map);
+    @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);
+    }
+
+
+
+    @GetMapping("listExcel")
+    @ApiOperation("������������")
+    public void listExcel(@RequestParam @ApiParam(value = "id",name = "������id") List<Integer> id,HttpServletResponse response){
+
+        List<AllocationExcelExt> extList = allocationService.listExcel(id);
+        try {
+            genImageExcel(extList,response);
+        }catch (Exception e){
+            log.error("error result ",e);
+        }
+
+    }
+
+
+    private void genImageExcel(List<AllocationExcelExt> list, HttpServletResponse response) throws IOException {
+        if (CollectionUtils.isEmpty(list)) {
+            return;
+        }
+        List<AllocationListExcelVo> demoDataList = new ArrayList<>();
+        for(int i = 0;i<list.size();i++){
+            AllocationExcelExt excelExt = list.get(i);
+            AllocationListExcelVo item = AllocationListExcelVo.convert(excelExt);
+            String stateStr = Objects.nonNull(excelExt.getState())?AllocationApproveEnum.getByValue(excelExt.getState()).getName():"���������������";
+            item.setStateStr(stateStr);
+            List<String> urls = new ArrayList<>();
+            for(FileAddressVo z : excelExt.getFileBaseList()){
+                urls.add(handleFileRealPath(z.getFileAddress()));
+            }
+            item.setChangeTypeStr(Objects.isNull(excelExt.getChangeType())?" ":ChangeEnum.getByValue(excelExt.getChangeType()).getName());
+            item.setChangeStr(Objects.isNull(excelExt.getIsChange())?"���": YesOrNo.getByValue(excelExt.getIsChange()).getName());
+            item.setWriteCellDataFile(urls);
+            List<String> urlsAfter = new ArrayList<>();
+            for(FileAddressVo z : excelExt.getFileChangeList()){
+                urlsAfter.add(handleFileRealPath(z.getFileAddress()));
+            }
+            item.setWriteCellDataFileResult(urlsAfter);
+            item.setEscalationTimeStr(DateUtils.dateToDateString(excelExt.getEscalationTime(),DateUtils.yyyy_MM_dd_EN));
+            item.setChangeTimeStr(Objects.isNull(excelExt.getChangeTime())?" ":DateUtils.dateToDateString(excelExt.getChangeTime(),DateUtils.yyyy_MM_dd_EN));
+            demoDataList.add(item);
+        }
+        try {
+            response.setHeader("Content-disposition", "attachment;filename*=utf-8''"
+                    + URLEncoder.encode("���������������", "utf-8") + ".xlsx");
+            response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
+        EasyExcel.write(response.getOutputStream(), AllocationListExcelVo.class)
+                .registerWriteHandler(new CustomImageModifyHandler()).sheet("���������").doWrite(demoDataList);
+            response.getOutputStream().close();
+        }catch (Exception e){
+            log.error("error result ",e);
+        }
+    }
+    private String handleFileRealPath(String path) {
+        return basePath.replaceAll(StringUtils.BACKSLASH.concat(StringUtils.BACKSLASH), StringUtils.SLASH).concat(StringUtils.SLASH).concat(path);
+    }
+
+
+    @GetMapping("unitQuery")
+    @ApiOperation("������������������")
+    public ResultMessage unitQuery(){
+        List<ResponsibilityUnitQuery> responsibilityUnitQueries = responsibilityUnitService.unitQuery();
+        return ResultMessage.ok(responsibilityUnitQueries);
+    }
+
 }

--
Gitblit v1.8.0