package com.ruoyi.production.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.production.dto.ProductionOrderStructureDto;
|
import com.ruoyi.production.mapper.ProductionOrderStructureMapper;
|
import com.ruoyi.production.pojo.ProductionOrderStructure;
|
import com.ruoyi.production.service.IProductionOrderStructureService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Objects;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 生产订单绑定的BOM子表 服务实现类
|
* </p>
|
*
|
* @author deslrey
|
* @since 2026-03-20
|
*/
|
@Slf4j
|
@Service
|
public class ProductionOrderStructureServiceImpl extends ServiceImpl<ProductionOrderStructureMapper, ProductionOrderStructure> implements IProductionOrderStructureService {
|
|
@Override
|
public List<ProductionOrderStructureDto> listByOrderId(Long orderId) {
|
List<ProductionOrderStructureDto> dtoList = baseMapper.listByOrderId(orderId);
|
|
Map<Long, ProductionOrderStructureDto> map = new HashMap<>();
|
for (ProductionOrderStructureDto node : dtoList) {
|
map.put(node.getId(), node);
|
}
|
|
List<ProductionOrderStructureDto> tree = new ArrayList<>();
|
for (ProductionOrderStructureDto node : dtoList) {
|
Long parentId = node.getParentId();
|
if (parentId == null || !map.containsKey(parentId)) {
|
tree.add(node);
|
} else {
|
map.get(parentId).getChildren().add(node);
|
}
|
}
|
return tree;
|
}
|
|
@Override
|
public void addOrUpdateBomStructs(Long orderId, List<ProductionOrderStructureDto> list) {
|
// 扁平化前端传入的树
|
List<ProductionOrderStructureDto> flatList = new ArrayList<>();
|
for (ProductionOrderStructureDto root : list) {
|
flatList.add(root);
|
flattenTree(root.getChildren(), flatList);
|
}
|
|
// 查询数据库已有数据
|
List<ProductionOrderStructure> dbList = list(new LambdaQueryWrapper<ProductionOrderStructure>()
|
.eq(ProductionOrderStructure::getOrderId, orderId));
|
|
// 前端已有id集合
|
Set<Long> frontendIds = flatList.stream()
|
.map(ProductionOrderStructureDto::getId)
|
.filter(Objects::nonNull)
|
.collect(Collectors.toSet());
|
|
// 需要删除的节点
|
Set<Long> deleteIds = dbList.stream()
|
.map(ProductionOrderStructure::getId)
|
.filter(id -> !frontendIds.contains(id))
|
.collect(Collectors.toSet());
|
if (!deleteIds.isEmpty()) {
|
removeByIds(deleteIds);
|
}
|
|
List<ProductionOrderStructure> insertList = new ArrayList<>();
|
List<ProductionOrderStructure> updateList = new ArrayList<>();
|
Map<String, ProductionOrderStructure> tempEntityMap = new HashMap<>();
|
|
for (ProductionOrderStructureDto dto : flatList) {
|
ProductionOrderStructure entity = new ProductionOrderStructure();
|
org.springframework.beans.BeanUtils.copyProperties(dto, entity);
|
entity.setOrderId(orderId);
|
if (dto.getId() == null) {
|
entity.setId(null);
|
entity.setParentId(null);
|
insertList.add(entity);
|
if (dto.getTempId() != null) {
|
tempEntityMap.put(dto.getTempId(), entity);
|
}
|
} else {
|
updateList.add(entity);
|
}
|
}
|
|
if (!insertList.isEmpty()) {
|
saveBatch(insertList);
|
}
|
|
// 回写新增节点的 parentId
|
List<ProductionOrderStructure> parentFixList = new ArrayList<>();
|
Long realParentId;
|
for (ProductionOrderStructureDto dto : flatList) {
|
if (dto.getId() != null) continue;
|
ProductionOrderStructure child = tempEntityMap.get(dto.getTempId());
|
if (child == null) continue;
|
String parentTempId = dto.getParentTempId();
|
if (parentTempId != null && !parentTempId.isEmpty()) {
|
if (tempEntityMap.containsKey(parentTempId)) {
|
realParentId = tempEntityMap.get(parentTempId).getId();
|
} else {
|
try {
|
realParentId = Long.valueOf(parentTempId);
|
} catch (NumberFormatException e) {
|
realParentId = 0L;
|
}
|
}
|
child.setParentId(realParentId);
|
} else {
|
child.setParentId(0L);
|
}
|
parentFixList.add(child);
|
}
|
|
if (!parentFixList.isEmpty()) {
|
updateBatchById(parentFixList);
|
}
|
if (!updateList.isEmpty()) {
|
updateBatchById(updateList);
|
}
|
}
|
|
private void flattenTree(List<ProductionOrderStructureDto> source, List<ProductionOrderStructureDto> result) {
|
if (source == null) return;
|
for (ProductionOrderStructureDto node : source) {
|
result.add(node);
|
flattenTree(node.getChildren(), result);
|
}
|
}
|
}
|
