质量管理bug修改

cement-business/src/main/java/cn/wise/sc/cement/business/controller/QualityApplyController.java

 package cn.wise.sc.cement.business.controller;
 
+import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.StrUtil;
 import cn.wise.sc.cement.business.entity.QualityApply;
@@ -24,6 +25,8 @@ import java.time.LocalDate;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -40,37 +43,51 @@ import java.util.List;
 @RequestMapping("/business/qualityApply")
 public class QualityApplyController {
 
-	final
-	IQualityApplyService iQualityApplyService;
+    final
+    IQualityApplyService iQualityApplyService;
 
-	public QualityApplyController(IQualityApplyService iQualityApplyService) {
-		this.iQualityApplyService = iQualityApplyService;
-	}
+    public QualityApplyController(IQualityApplyService iQualityApplyService) {
+        this.iQualityApplyService = iQualityApplyService;
+    }
 
-	@PostMapping("/apply")
-	@ApiOperation("质量详情审核")
-	public BaseResponse<Boolean> applyQuality(@RequestBody List<QualityApply> qualityApply) {
-		boolean save = iQualityApplyService.saveBatch(qualityApply);
-		if (save) {
-			return BaseResponse.okData(true);
-		} else {
-			return BaseResponse.errorMsg("添加失败!");
-		}
-	}
+    @PostMapping("/apply")
+    @ApiOperation("质量详情审核")
+    public BaseResponse<Boolean> applyQuality(@RequestBody List<QualityApply> qualityApply) {
 
-	@PostMapping("/export")
-	@ApiOperation("导出质量监控")
-	public void exportQuality(String startDate, String endDate, HttpServletResponse response) {
+        Set<Integer> groupIds = qualityApply
+                .stream().map(QualityApply::getTeamGroupId).collect(Collectors.toSet());
+        QueryWrapper<QualityApply> qw = new QueryWrapper<>();
+        Set<Integer> projectIds = qualityApply
+                .stream().map(QualityApply::getProjectId).collect(Collectors.toSet());
+        if (CollectionUtil.isEmpty(qualityApply) && CollectionUtil.isEmpty(groupIds)
+                && CollectionUtil.isEmpty(projectIds)) {
+            return BaseResponse.okData(true);
+        }
 
-		Date startParse = null;
-		if (StrUtil.isNotBlank(startDate)) {
-			startParse = DateUtil.parse(startDate);
-		}
-		Date endParse = null;
-		if (StrUtil.isNotBlank(endDate)) {
-			endParse = DateUtil.parse(endDate);
-		}
-		iQualityApplyService.exportQuality(startParse, endParse, response);
-	}
+        qw.eq("project_id", projectIds.iterator().next());
+        qw.in("team_group_id", groupIds);
+        iQualityApplyService.remove(qw);
+        boolean save = iQualityApplyService.saveBatch(qualityApply);
+        if (save) {
+            return BaseResponse.okData(true);
+        } else {
+            return BaseResponse.errorMsg("添加失败!");
+        }
+    }
+
+    @PostMapping("/export")
+    @ApiOperation("导出质量监控")
+    public void exportQuality(String startDate, String endDate, HttpServletResponse response) {
+
+        Date startParse = null;
+        if (StrUtil.isNotBlank(startDate)) {
+            startParse = DateUtil.parse(startDate);
+        }
+        Date endParse = null;
+        if (StrUtil.isNotBlank(endDate)) {
+            endParse = DateUtil.parse(endDate);
+        }
+        iQualityApplyService.exportQuality(startParse, endParse, response);
+    }
 }