From bf84fec7d8dd5e7fbcc93030b8d83da7ec0ece91 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期五, 03 四月 2026 15:39:36 +0800
Subject: [PATCH] yys

---
 src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java |   21 ++++++++++++++++++++-
 1 files changed, 20 insertions(+), 1 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 381b642..5bb0fc6 100644
--- a/src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java
+++ b/src/main/java/com/ruoyi/production/service/impl/ProductStructureServiceImpl.java
@@ -129,9 +129,11 @@
     }
 
 
+
+
     @Override
     public List<ProductStructureDto> listBybomId(Integer bomId) {
-        List<ProductStructureDto> list = productStructureMapper.listBybomId(bomId);
+        List<ProductStructureDto> list = productStructureMapper.listBybomId(bomId,false);
 
         Map<Long, ProductStructureDto> map = new HashMap<>();
         for (ProductStructureDto node : list) {
@@ -154,4 +156,21 @@
         return tree;
     }
 
+    @Override
+    public List<ProductStructureDto> listBybomIdIsParent(Integer bomId) {
+        return productStructureMapper.listBybomId(bomId,true);
+    }
+
+    @Override
+    public Long countBybomId(Integer bomId) {
+        ProductStructure productStructure = productStructureMapper.selectOne(new LambdaQueryWrapper<ProductStructure>()
+                .eq(ProductStructure::getBomId, bomId)
+                .isNull(ProductStructure::getParentId));
+        if(productStructure != null){
+           return productStructureMapper.selectCount(new LambdaQueryWrapper<ProductStructure>()
+                    .eq(ProductStructure::getParentId, productStructure.getId()));
+        }
+        return 0L;
+    }
+
 }

--
Gitblit v1.9.3