From 482033ad6accee24ae2a530ad5fbc42febb0f2f1 Mon Sep 17 00:00:00 2001
From: huminmin <mac@MacBook-Pro.local>
Date: 星期一, 16 三月 2026 13:31:56 +0800
Subject: [PATCH] 质量管理：优化更新逻辑，增加原料导出、下载接口
---
src/main/java/com/ruoyi/quality/service/impl/RawMaterialServiceImpl.java | 185 ++++++++++++++++++++++++++++++++++++++++-----
1 files changed, 162 insertions(+), 23 deletions(-)
diff --git a/src/main/java/com/ruoyi/quality/service/impl/RawMaterialServiceImpl.java b/src/main/java/com/ruoyi/quality/service/impl/RawMaterialServiceImpl.java
index 23f3587..035648f 100644
--- a/src/main/java/com/ruoyi/quality/service/impl/RawMaterialServiceImpl.java
+++ b/src/main/java/com/ruoyi/quality/service/impl/RawMaterialServiceImpl.java
@@ -4,6 +4,8 @@
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.deepoove.poi.XWPFTemplate;
+import com.deepoove.poi.config.Configure;
import com.ruoyi.basic.mapper.ProductMapper;
import com.ruoyi.basic.mapper.ProductModelMapper;
import com.ruoyi.basic.pojo.Product;
@@ -11,11 +13,16 @@
import com.ruoyi.common.enums.RawMaterialCheckResult;
import com.ruoyi.common.enums.RawMaterialCheckType;
import com.ruoyi.common.enums.RawMaterialInspectState;
+import com.ruoyi.common.utils.HackLoopTableRenderPolicy;
import com.ruoyi.common.utils.bean.BeanUtils;
+import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.inspectiontask.mapper.QualityInspectItemMapper;
import com.ruoyi.inspectiontask.pojo.QualityInspectItem;
import com.ruoyi.quality.dto.RawMaterialDto;
+import com.ruoyi.quality.dto.RawMaterialQualityInspectItemDto;
import com.ruoyi.quality.mapper.RawMaterialMapper;
+import com.ruoyi.quality.pojo.QualityInspect;
+import com.ruoyi.quality.pojo.QualityInspectParam;
import com.ruoyi.quality.pojo.RawMaterial;
import com.ruoyi.quality.pojo.RawMaterialQualityInspectItem;
import com.ruoyi.quality.service.RawMaterialQualityInspectItemService;
@@ -25,9 +32,11 @@
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import javax.servlet.http.HttpServletResponse;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.URLEncoder;
+import java.util.*;
import java.util.stream.Collectors;
@@ -90,8 +99,12 @@
.eq(RawMaterialQualityInspectItem::getRawMaterialId, id));
Map<Long, RawMaterialQualityInspectItem> map = rawQualityList.stream().collect(Collectors.toMap(RawMaterialQualityInspectItem::getQualityInspectItemId, qualityInspectItem -> qualityInspectItem));
// 鑾峰彇妫�楠岄」鐩�
- List<QualityInspectItem> qualityInspectItems = qualityInspectItemMapper.selectList(new LambdaQueryWrapper<QualityInspectItem>()
- .in(QualityInspectItem::getId, rawQualityList.stream().map(RawMaterialQualityInspectItem::getQualityInspectItemId).collect(Collectors.toList())));
+ List<QualityInspectItem> qualityInspectItems = new ArrayList<>();
+ if (CollectionUtils.isNotEmpty(rawQualityList)) {
+ List<Long> qualityInspectItemIds = rawQualityList.stream().map(RawMaterialQualityInspectItem::getQualityInspectItemId).collect(Collectors.toList());
+ qualityInspectItems = qualityInspectItemMapper.selectList(new LambdaQueryWrapper<QualityInspectItem>()
+ .in(QualityInspectItem::getId, qualityInspectItemIds));
+ }
// 鏇挎崲鍖栭獙鍊�
qualityInspectItems.forEach(qualityInspectItem -> {
qualityInspectItem.setTestValue(map.get(qualityInspectItem.getId()).getTestValue());
@@ -128,24 +141,150 @@
}
@Override
+ @Transactional(rollbackFor = Exception.class)
public boolean update(RawMaterialDto rawMaterialDto) {
- // 鍒犻櫎鎵�鏈夊叧鑱旈」鐩噸鏂板姞杞�
- rawMaterialQualityInspectItemService.remove(new LambdaQueryWrapper<RawMaterialQualityInspectItem>()
- .eq(RawMaterialQualityInspectItem::getRawMaterialId, rawMaterialDto.getId()));
- int i = rawMaterialMapper.updateById(rawMaterialDto);
- // 鏂板妫�娴嬮」鍏宠仈鍏崇郴
- if (CollectionUtils.isNotEmpty(rawMaterialDto.getQualityInspectItem())) {
- List<RawMaterialQualityInspectItem> rawMaterialQualityInspectItems = rawMaterialDto.getQualityInspectItem().stream()
- .map(qualityInspectItem -> {
- RawMaterialQualityInspectItem item = new RawMaterialQualityInspectItem();
- item.setRawMaterialId(rawMaterialDto.getId());
- item.setQualityInspectItemId(qualityInspectItem.getId());
- item.setTestValue(qualityInspectItem.getTestValue());
- return item;
- })
- .collect(Collectors.toList());
- rawMaterialQualityInspectItemService.saveBatch(rawMaterialQualityInspectItems);
+ // 鏇存柊鍘熸枡
+ if (rawMaterialMapper.updateById(rawMaterialDto) <= 0) {
+ return false;
}
- return i > 0;
+
+ Long rawMaterialId = rawMaterialDto.getId();
+ List<QualityInspectItem> newQualityInspectItems = rawMaterialDto.getQualityInspectItem();
+
+ // 鑾峰彇鐜版湁妫�楠岄」鍏宠仈璁板綍
+ List<RawMaterialQualityInspectItem> existingItems = rawMaterialQualityInspectItemService.list(
+ new LambdaQueryWrapper<RawMaterialQualityInspectItem>()
+ .eq(RawMaterialQualityInspectItem::getRawMaterialId, rawMaterialId)
+ );
+
+ // 鍒犻櫎涓嶅瓨鍦ㄧ殑鍏宠仈璁板綍
+ if (CollectionUtils.isEmpty(newQualityInspectItems)) {
+ // 濡傛灉娌℃湁妫�楠岄」锛屽垹闄ゆ墍鏈夌幇鏈夊叧鑱旇褰�
+ if (CollectionUtils.isNotEmpty(existingItems)) {
+ List<Long> existingIds = existingItems.stream()
+ .map(RawMaterialQualityInspectItem::getId)
+ .collect(Collectors.toList());
+ rawMaterialQualityInspectItemService.removeByIds(existingIds);
+ }
+ return true;
+ }
+
+ // 鑾峰彇鏂版楠岄」鐨処D闆嗗悎鍜屾祴璇曞�兼槧灏�
+ Set<Long> newItemIds = newQualityInspectItems.stream()
+ .map(QualityInspectItem::getId)
+ .collect(Collectors.toSet());
+
+ Map<Long, String> testValueMap = newQualityInspectItems.stream()
+ .collect(Collectors.toMap(QualityInspectItem::getId, QualityInspectItem::getTestValue));
+
+ // 鍒犻櫎涓嶅瓨鍦ㄧ殑鍏宠仈璁板綍
+ List<Long> toDelete = existingItems.stream()
+ .filter(item -> !newItemIds.contains(item.getQualityInspectItemId()))
+ .map(RawMaterialQualityInspectItem::getId)
+ .collect(Collectors.toList());
+
+ if (CollectionUtils.isNotEmpty(toDelete)) {
+ rawMaterialQualityInspectItemService.removeByIds(toDelete);
+ }
+
+ // 澶勭悊鏇存柊鏁版嵁
+ List<RawMaterialQualityInspectItem> toUpdate = existingItems.stream()
+ .filter(item -> newItemIds.contains(item.getQualityInspectItemId()))
+ .peek(item -> item.setTestValue(testValueMap.get(item.getQualityInspectItemId())))
+ .collect(Collectors.toList());
+
+ if (CollectionUtils.isNotEmpty(toUpdate)) {
+ rawMaterialQualityInspectItemService.updateBatchById(toUpdate);
+ }
+
+ // 澶勭悊鏂板鏁版嵁
+ Set<Long> existingItemIds = existingItems.stream()
+ .map(RawMaterialQualityInspectItem::getQualityInspectItemId)
+ .collect(Collectors.toSet());
+
+ List<RawMaterialQualityInspectItem> toCreate = newQualityInspectItems.stream()
+ .filter(item -> !existingItemIds.contains(item.getId()))
+ .map(item -> {
+ RawMaterialQualityInspectItem newItem = new RawMaterialQualityInspectItem();
+ newItem.setRawMaterialId(rawMaterialId);
+ newItem.setQualityInspectItemId(item.getId());
+ newItem.setTestValue(item.getTestValue());
+ return newItem;
+ })
+ .collect(Collectors.toList());
+
+ if (CollectionUtils.isNotEmpty(toCreate)) {
+ rawMaterialQualityInspectItemService.saveBatch(toCreate);
+ }
+
+ return true;
}
-}
+
+ @Override
+ public void export(HttpServletResponse response, RawMaterialDto rawMaterialDto) {
+ List<RawMaterialDto> rawMaterials = rawMaterialMapper.listPage(new Page<>(1, Integer.MAX_VALUE), rawMaterialDto).getRecords();
+ ExcelUtil<RawMaterialDto> util = new ExcelUtil<RawMaterialDto>(RawMaterialDto.class);
+ util.exportExcel(response, rawMaterials, "鍘熸枡妫�楠屽鍑�");
+ }
+
+ @Override
+ public void down(HttpServletResponse response, RawMaterialDto rawMaterialDto) {
+ RawMaterial rawMaterial = rawMaterialMapper.selectById(rawMaterialDto.getId());
+ // 鑾峰彇浜у搧鍨嬪彿淇℃伅
+ ProductModel productModel = productModelMapper.selectById(rawMaterial.getProductModelId());
+ Product product = productMapper.selectById(productModel.getProductId());
+ RawMaterialDto inspect = new RawMaterialDto();
+ BeanUtils.copyProperties(rawMaterial, inspect);
+ inspect.setModel(productModel.getModel());
+ inspect.setProductName(product.getProductName());
+ inspect.setCheckTypeText(RawMaterialCheckType.fromValue(rawMaterial.getCheckType()) == null ? "" : RawMaterialCheckType.fromValue(rawMaterial.getCheckType()).getValue());
+ inspect.setCheckResultText(RawMaterialCheckResult.fromValue(rawMaterial.getCheckResult()) == null ? "" : RawMaterialCheckResult.fromValue(rawMaterial.getCheckResult()).getValue());
+
+ List<RawMaterialQualityInspectItem> paramList = rawMaterialQualityInspectItemService.list(
+ new LambdaQueryWrapper<RawMaterialQualityInspectItem>()
+ .eq(RawMaterialQualityInspectItem::getRawMaterialId, rawMaterialDto.getId())
+ );
+ List<RawMaterialQualityInspectItemDto> inspectItems = new ArrayList<>();
+
+ int index = 1;
+ for (RawMaterialQualityInspectItem item : paramList) {
+ RawMaterialQualityInspectItemDto dto = new RawMaterialQualityInspectItemDto();
+ BeanUtils.copyProperties(item, dto);
+ QualityInspectItem qualityInspectItem = qualityInspectItemMapper.selectById(item.getQualityInspectItemId());
+ dto.setName(qualityInspectItem.getName());
+ dto.setUnit(qualityInspectItem.getUnit());
+ dto.setStandardValue(qualityInspectItem.getStandardValue());
+ dto.setInternalControl(qualityInspectItem.getInternalControl());
+ dto.setIndex(index);
+ inspectItems.add(dto);
+ index++;
+ }
+
+ InputStream inputStream = this.getClass().getResourceAsStream("/static/report-template.docx");
+ Configure configure = Configure.builder()
+ .bind("paramList", new HackLoopTableRenderPolicy())
+ .build();
+ XWPFTemplate template = XWPFTemplate.compile(inputStream, configure).render(
+ new HashMap<String, Object>() {{
+ put("inspect", inspect);
+ put("paramList", inspectItems);
+ }});
+
+ try {
+ response.setContentType("application/msword");
+ String fileName = URLEncoder.encode(
+ "妫�楠屾姤鍛�", "UTF-8");
+ response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
+ response.setHeader("Content-disposition",
+ "attachment;filename=" + fileName + ".docx");
+ OutputStream os = response.getOutputStream();
+ template.write(os);
+ os.flush();
+ os.close();
+ inputStream.close();
+ } catch (Exception e) {
+ e.printStackTrace();
+ throw new RuntimeException("瀵煎嚭澶辫触");
+ }
+ }
+}
\ No newline at end of file
--
Gitblit v1.9.3