From 346288bc38bdab442721d7609caed86d03b9d2d4 Mon Sep 17 00:00:00 2001
From: cjl <276999030@qq.com>
Date: Tue, 26 Sep 2023 19:25:24 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/wb' into qa

---
 screen-api/src/main/java/com/moral/api/controller/AllocationController.java |  239 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 239 insertions(+), 0 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
new file mode 100644
index 0000000..67e7547
--- /dev/null
+++ b/screen-api/src/main/java/com/moral/api/controller/AllocationController.java
@@ -0,0 +1,239 @@
+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 org.aspectj.apache.bcel.generic.RET;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+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.web.bind.annotation.*;
+
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import com.moral.api.entity.Allocation;
+import com.moral.api.entity.ResponsibilityUnit;
+import com.moral.api.pojo.dto.allocation.AllocationUnitViewDto;
+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.WebUtils;
+
+import javax.validation.Valid;
+
+
+@Api(tags = {"������������"})
+@RestController
+@RequestMapping("allocation")
+public class AllocationController {
+
+    @Autowired
+    private AllocationService allocationService;
+
+
+    @ApiOperation(value = "������������", notes = "������������")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String")
+    })
+    @RequestMapping(value = "contaminate", method = RequestMethod.GET)
+    public ResultMessage contaminate() {
+        List<Map<String, Object>> professions = allocationService.sysDictData(Constants.WU_RAN_LEI_XING);
+        return ResultMessage.ok(professions);
+    }
+
+
+    @ApiOperation(value = "������������", notes = "������������")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String")
+    })
+    @RequestMapping(value = "unit", method = RequestMethod.GET)
+    public ResultMessage unit() {
+        List<ResponsibilityUnit> responsibilityUnits = allocationService.seleteUnit();
+        return ResultMessage.ok(responsibilityUnits);
+    }
+
+    /**
+     * ���������������
+     * @return
+     */
+    @PostMapping("insert")
+    @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 allocationUpdateCond
+     * @return
+     *
+     *
+     */
+    @PostMapping("update")
+    @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();
+    }
+    /**
+     * ������������������
+     * @return
+     */
+    @GetMapping("selectAll")
+    public ResultMessage selectAll(Map<String,Object> map){
+        allocationService.selectAll(map);
+        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());
+        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("selectUnitView")
+    public ResultMessage selectUnitView(HttpServletRequest request){
+        Map<String, Object> params = WebUtils.getParametersStartingWith(request, null);
+        Map<String, Object> map1 = allocationService.selectUnitView(params);
+        return ResultMessage.ok(map1);
+    }
+
+
+    @GetMapping("unitExel")
+    public void unitExel(HttpServletResponse response,HttpServletRequest request){
+        Map<String, Object> params = WebUtils.getParametersStartingWith(request, null);
+        //������������
+        Map<String, Object> map1 = allocationService.selectUnitView(params);
+        List<AllocationUnitViewDto> unitView = (List<AllocationUnitViewDto>) map1.get("unitView");
+
+
+        ArrayList<Map<String, Object>> mapArrayList = new ArrayList<>();
+        for (AllocationUnitViewDto allocationUnitViewDto : unitView) {
+            Map<String, Object> map = entityToMap(allocationUnitViewDto);
+            mapArrayList.add(map);
+        }
+        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;
+        }
+    }
+
+    /**
+     ������������Map
+     */
+    public static Map<String, Object> entityToMap(Object object) {
+        Map<String, Object> map = new HashMap<>();
+        for (Field field : object.getClass().getDeclaredFields()) {
+            try {
+                boolean flag = field.isAccessible();
+                field.setAccessible(true);
+                Object o = field.get(object);
+                map.put(field.getName(), o);
+                field.setAccessible(flag);
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+        return map;
+    }
+}

--
Gitblit v1.8.0