From 663120b055dfdced4a67ce1d4160ae8ca19b4190 Mon Sep 17 00:00:00 2001
From: chenhj <1263187585@qq.com>
Date: 星期四, 12 三月 2026 18:04:21 +0800
Subject: [PATCH] 修改工单所需数量

---
 src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java |   15 +++++++++++++--
 1 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java b/src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java
index 35d0366..e4ef9d9 100644
--- a/src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java
+++ b/src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java
@@ -1,6 +1,8 @@
 package com.ruoyi.production.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.production.dto.ProductStructureDto;
 import com.ruoyi.production.mapper.ProductStructureMapper;
 import com.ruoyi.production.pojo.ProductStructure;
 import com.ruoyi.production.service.ProductStructureService;
@@ -20,7 +22,16 @@
     private  ProductStructureMapper productStructureMapper;
 
     @Override
-    public List<ProductStructure> listByProductId(Long productId) {
-        return productStructureMapper.listByProductId( productId);
+    public Boolean addProductStructureDto(ProductStructureDto productStructureDto) {
+        this.remove(new QueryWrapper<ProductStructure>().lambda().eq(ProductStructure::getBomId, productStructureDto.getBomId()));
+        productStructureDto.getProductStructureList().forEach(productStructure -> {
+            productStructure.setBomId(productStructureDto.getBomId());
+        });
+        return this.saveBatch(productStructureDto.getProductStructureList());
+    }
+
+    @Override
+    public List<ProductStructureDto> listBybomId(Long bomId) {
+        return productStructureMapper.listBybomId(bomId);
     }
 }

--
Gitblit v1.9.3