From 7ef4892f9f24f941aca37e6b3991b808a0aca619 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期五, 08 九月 2023 11:16:35 +0800
Subject: [PATCH] 优化

---
 ycl-platform/src/main/java/com/ycl/controller/epuipment/EquipmentBayonetController.java |   77 +++++++++++---------------------------
 1 files changed, 23 insertions(+), 54 deletions(-)

diff --git a/ycl-platform/src/main/java/com/ycl/controller/epuipment/EquipmentBayonetController.java b/ycl-platform/src/main/java/com/ycl/controller/epuipment/EquipmentBayonetController.java
index 8f2ef89..59cbc6f 100644
--- a/ycl-platform/src/main/java/com/ycl/controller/epuipment/EquipmentBayonetController.java
+++ b/ycl-platform/src/main/java/com/ycl/controller/epuipment/EquipmentBayonetController.java
@@ -1,30 +1,23 @@
 package com.ycl.controller.epuipment;
 
 
-import com.alibaba.excel.EasyExcel;
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.ycl.annotation.LogSave;
 import com.ycl.api.CommonResult;
 import com.ycl.controller.BaseController;
-import com.ycl.entity.dict.DataDictionary;
 import com.ycl.entity.equipment.EquipmentBayonet;
 import com.ycl.service.dict.IDataDictionaryService;
 import com.ycl.service.equipment.IEquipmentBayonetService;
+import com.ycl.service.region.ISccgRegionService;
+import com.ycl.utils.EasyExcelUtils;
 import com.ycl.vo.equipment.EquipmentBayonetVO;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.SneakyThrows;
-import org.apache.commons.lang3.StringUtils;
-import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletResponse;
-import java.net.URLEncoder;
-import java.time.LocalDateTime;
-import java.time.format.DateTimeFormatter;
-import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -43,73 +36,49 @@
     IEquipmentBayonetService iEquipmentBayonetService;
     @Autowired
     IDataDictionaryService iDataDictionaryService;
+    @Autowired
+    ISccgRegionService iSccgRegionService;
 
     @GetMapping("/query")
     @ApiOperation("鏌ヨ")
-    @LogSave
+    @LogSave(operationType = "鍗″彛绠$悊", contain = "鏌ヨ鍗″彛")
     public CommonResult search(@RequestParam(required = false) Integer size,
                                @RequestParam(required = false) Integer current,
                                @RequestParam(required = false) String bayonetName) {
         Page<EquipmentBayonet> equipmentBayonetPage = new Page<>();
         equipmentBayonetPage.setCurrent(current);
         equipmentBayonetPage.setSize(size);
-        return CommonResult.success(iEquipmentBayonetService
-                .page(equipmentBayonetPage, new LambdaQueryWrapper<EquipmentBayonet>()
-                        .like(StringUtils.isNotBlank(bayonetName), EquipmentBayonet::getBayonetName, bayonetName))
-                .getRecords()
-                .stream()
-                .map(item -> {
-                    EquipmentBayonetVO equipmentBayonetVO = new EquipmentBayonetVO();
-                    BeanUtils.copyProperties(item, equipmentBayonetVO);
-                    equipmentBayonetVO.setFrontEndType(iDataDictionaryService.getOne(new LambdaQueryWrapper<DataDictionary>().eq(DataDictionary::getId, item.getFrontEndType())).getName());
-                    equipmentBayonetVO.setInOutCityType(iDataDictionaryService.getOne(new LambdaQueryWrapper<DataDictionary>().eq(DataDictionary::getId, item.getInOutCityType())).getName());
-                    return equipmentBayonetVO;
-                }).collect(Collectors.toList()));
+        return CommonResult.success(iEquipmentBayonetService.selectEquipmentPage(equipmentBayonetPage, bayonetName));
     }
 
     @PostMapping("/addition")
     @ApiOperation("娣诲姞")
-    @LogSave(operationType = "鍗″彛绠$悊",contain = "娣诲姞鍗″彛")
+    @LogSave(operationType = "鍗″彛绠$悊", contain = "娣诲姞鍗″彛")
     public CommonResult add(@RequestBody EquipmentBayonet equipmentBayonet) {
         return CommonResult.success(iEquipmentBayonetService.save(equipmentBayonet));
     }
 
     @PutMapping("/modification")
     @ApiOperation("淇敼")
-    @LogSave(operationType = "鍗″彛绠$悊",contain = "淇敼鍗″彛")
+    @LogSave(operationType = "鍗″彛绠$悊", contain = "淇敼鍗″彛")
     public CommonResult modify(@RequestBody EquipmentBayonet equipmentBayonet) {
         return CommonResult.success(iEquipmentBayonetService.updateById(equipmentBayonet));
     }
 
-    @GetMapping("/export")
-    @ApiOperation("瀵煎嚭")
-    @SneakyThrows
-    @LogSave(operationType = "鍗″彛绠$悊",contain = "鍒犻櫎鍗″彛")
-    public void export(HttpServletResponse response,
-                       @RequestParam(required = false) Integer size,
-                       @RequestParam(required = false) Integer current,
-                       @RequestParam(required = false) String bayonetName) {
-        Page<EquipmentBayonet> equipmentBayonetPage = new Page<>();
-        equipmentBayonetPage.setCurrent(current);
-        equipmentBayonetPage.setSize(size);
-        String fileName ="鍗″彛鏁版嵁";
-        response.setContentType("application/vnd.ms-excel");
-        response.setCharacterEncoding("utf-8");
-        response.setHeader("Content-disposition", "attachment;filename=" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")) + URLEncoder.encode(fileName + ".xlsx", "utf-8"));
-        EasyExcel.write(response.getOutputStream(), EquipmentBayonetVO.class)
-                .sheet("鏁版嵁")
-                .doWrite(iEquipmentBayonetService
-                        .page(equipmentBayonetPage, new LambdaQueryWrapper<EquipmentBayonet>()
-                                .like(StringUtils.isNotBlank(bayonetName), EquipmentBayonet::getBayonetName, bayonetName))
-                        .getRecords()
-                        .stream()
-                        .map(item -> {
-                            EquipmentBayonetVO equipmentBayonetVO = new EquipmentBayonetVO();
-                            BeanUtils.copyProperties(item, equipmentBayonetVO);
-                            equipmentBayonetVO.setFrontEndType(iDataDictionaryService.getOne(new LambdaQueryWrapper<DataDictionary>().eq(DataDictionary::getId, item.getFrontEndType())).getName());
-                            equipmentBayonetVO.setInOutCityType(iDataDictionaryService.getOne(new LambdaQueryWrapper<DataDictionary>().eq(DataDictionary::getId, item.getInOutCityType())).getName());
-                            return equipmentBayonetVO;
-                        }).collect(Collectors.toList()));
+    @DeleteMapping("/delete/{id}")
+    @ApiOperation("鍒犻櫎")
+    @LogSave(operationType = "鍗″彛绠$悊", contain = "鍒犻櫎鍗″彛")
+    public CommonResult delete(@PathVariable Integer id) {
+        return CommonResult.success(iEquipmentBayonetService.removeById(id));
     }
 
-}
+
+    @PostMapping("/export")
+    @ApiOperation("瀵煎嚭")
+    @SneakyThrows
+    @LogSave(operationType = "鍗″彛绠$悊", contain = "瀵煎嚭鍗″彛")
+    public void export(HttpServletResponse response,
+                       @RequestParam(required = false) String bayonetName) {
+        EasyExcelUtils.export(response, "鏁版嵁", EquipmentBayonetVO.class, iEquipmentBayonetService.selectEquipmentList(bayonetName));
+    }
+}
\ No newline at end of file

--
Gitblit v1.8.0