package com.ruoyi.production.service.impl;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.production.mapper.ProductionLineMapper;
|
import com.ruoyi.production.pojo.ProductionLine;
|
import com.ruoyi.production.service.ProductionLineService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* @author :yys
|
* @date : 2025/12/4 11:19
|
*/
|
@Service
|
@Slf4j
|
public class ProductionLineServiceImpl extends ServiceImpl<ProductionLineMapper, ProductionLine> implements ProductionLineService {
|
|
@Autowired
|
private ProductionLineMapper productionLineMapper;
|
|
/**
|
* 递归查询产线树
|
* @return
|
*/
|
@Override
|
public List<ProductionLine> listTree() {
|
List<ProductionLine> productionLines = productionLineMapper.selectList(Wrappers.lambdaQuery(ProductionLine.class)
|
.eq(ProductionLine::getParentId, 0)
|
.eq(ProductionLine::getType, 1));
|
// 2. 递归为每个根节点加载子节点（工序/子产线）
|
for (ProductionLine rootLine : productionLines) {
|
loadChildren(rootLine,1);
|
}
|
return productionLines;
|
}
|
|
/**
|
* 递归加载子节点（核心方法）
|
* @param parentNode 父节点对象
|
*/
|
private void loadChildren(ProductionLine parentNode,Integer type) {
|
// 1. 查询当前父节点的所有子节点
|
List<ProductionLine> children = productionLineMapper.selectList(Wrappers.lambdaQuery(ProductionLine.class)
|
.eq(ProductionLine::getParentId, parentNode.getId())
|
.eq(ProductionLine::getType, type));
|
|
// 2. 若有子节点，继续递归加载子节点的子节点
|
if (!children.isEmpty()) {
|
parentNode.setChildren(children); // 设置子节点
|
for (ProductionLine child : children) {
|
loadChildren(child,type); // 递归调用，加载孙子节点
|
}
|
}
|
}
|
|
/**
|
* 按需提供：根据父节点ID查询单个子树（如查询某个产线下的所有工序）
|
*/
|
@Override
|
public IPage<ProductionLine> getSubTreeByParentId(Page page, ProductionLine productionLine) {
|
// 1. 查询父节点本身
|
ProductionLine parentNode = productionLineMapper.selectById(productionLine.getId());
|
if (parentNode == null) {
|
return null;
|
}
|
// 2. 递归加载子节点
|
loadChildren(productionLine,1);
|
List<Long> ids = new ArrayList<>();
|
ids.add(productionLine.getId());
|
if(CollectionUtils.isNotEmpty(productionLine.getChildren())){
|
// 递归获取所有子节点的id
|
getAllChildIds(productionLine,ids);
|
}
|
return productionLineMapper.selectPage(page, Wrappers.lambdaQuery(ProductionLine.class)
|
.in(ProductionLine::getParentId, ids)
|
.eq(ProductionLine::getType, 2));
|
}
|
|
/**
|
* 递归获取所有子节点的id
|
*/
|
public List<Long> getAllChildIds(ProductionLine node,List<Long> ids) {
|
if (node.getChildren() != null) {
|
for (ProductionLine child : node.getChildren()) {
|
ids.add(child.getId());
|
ids.addAll(getAllChildIds(child,ids));
|
}
|
}
|
return ids;
|
}
|
}
|
