package org.jeecg.modules.eam.service.impl; import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import org.jeecg.common.api.vo.CommonGenericTree; import org.jeecg.modules.eam.entity.ProcessParametersCategory; import org.jeecg.modules.eam.entity.RepairStandardCategory; import org.jeecg.modules.eam.mapper.RepairStandardCategoryMapper; import org.jeecg.modules.eam.service.IRepairStandardCategoryService; import org.springframework.stereotype.Service; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; /** * @Description: mom_eam_repair_standard_category * @Author: jeecg-boot * @Date: 2024-07-08 * @Version: V1.0 */ @Service public class RepairStandardCategoryServiceImpl extends ServiceImpl implements IRepairStandardCategoryService { @Override public List loadTree() { List repairStandardCategoryList = this.lambdaQuery().eq(RepairStandardCategory::getDelFlag, "0").list(); List commonGenericTrees = loadTree(repairStandardCategoryList); return commonGenericTrees; } @SuppressWarnings("unchecked") public List loadTree(List repairStandardCategoryList) { @SuppressWarnings("unused") RepairStandardCategory repairStandardCategory = new RepairStandardCategory(); List list = new ArrayList<>(); Map map = new HashMap<>(); Map map2 = new HashMap<>(); Map map3 = new HashMap<>(); Map map4 = new HashMap<>(); Map map5 = new HashMap<>(); Map map6 = new HashMap<>(); Map map7 = new HashMap<>(); CommonGenericTree node = new CommonGenericTree<>(); node.setKey("-1"); node.setValue("-1"); node.setTitle("维修标准分类"); node.setRField1(""); node.setRField2(""); node.setDisabled(false); node.setEntity(new RepairStandardCategory()); list.add(node); if (CollectionUtils.isNotEmpty(list)) { CommonGenericTree pcNode; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (standardCategory.getParentId().equals("-1")) { pcNode = new CommonGenericTree<>(); pcNode.setKey(standardCategory.getId()); pcNode.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); pcNode.setParentId(node.getKey()); pcNode.setIcon(""); pcNode.setType(0); pcNode.setValue(standardCategory.getId()); pcNode.setDisabled(false); pcNode.setRField1(standardCategory.getNum()); pcNode.setRField2(standardCategory.getName()); pcNode.setEntity(standardCategory); node.addChildren(pcNode); map.put(standardCategory.getId(), pcNode); } } CommonGenericTree childNode; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map.containsKey(standardCategory.getParentId())) { pcNode = map.get(standardCategory.getParentId()); childNode = new CommonGenericTree<>(); childNode.setKey(standardCategory.getId()); childNode.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); childNode.setParentId(standardCategory.getParentId()); childNode.setIcon(""); childNode.setType(0); childNode.setValue(standardCategory.getId()); childNode.setDisabled(false); childNode.setRField1(standardCategory.getNum()); childNode.setRField2(standardCategory.getName()); childNode.setEntity(standardCategory); pcNode.addChildren(childNode); map2.put(standardCategory.getId(), childNode); } } if(map2.size()>0){ CommonGenericTree child2Node; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map2.containsKey(standardCategory.getParentId())) { pcNode = map2.get(standardCategory.getParentId()); child2Node = new CommonGenericTree<>(); child2Node.setKey(standardCategory.getId()); child2Node.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); child2Node.setParentId(standardCategory.getParentId()); child2Node.setIcon(""); child2Node.setType(0); child2Node.setValue(standardCategory.getId()); child2Node.setDisabled(false); child2Node.setRField1(standardCategory.getNum()); child2Node.setRField2(standardCategory.getName()); child2Node.setEntity(standardCategory); pcNode.addChildren(child2Node); map3.put(standardCategory.getId(), child2Node); } } if(map3.size()>0){ CommonGenericTree child3Node; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map3.containsKey(standardCategory.getParentId())) { pcNode = map3.get(standardCategory.getParentId()); child3Node = new CommonGenericTree<>(); child3Node.setKey(standardCategory.getId()); child3Node.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); child3Node.setParentId(standardCategory.getParentId()); child3Node.setIcon(""); child3Node.setType(0); child3Node.setValue(standardCategory.getId()); child3Node.setDisabled(false); child3Node.setRField1(standardCategory.getNum()); child3Node.setRField2(standardCategory.getName()); child3Node.setEntity(standardCategory); pcNode.addChildren(child3Node); map4.put(standardCategory.getId(), child3Node); } } if(map4.size()>0){ CommonGenericTree child4Node; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map4.containsKey(standardCategory.getParentId())) { pcNode = map4.get(standardCategory.getParentId()); child4Node = new CommonGenericTree<>(); child4Node.setKey(standardCategory.getId()); child4Node.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); child4Node.setParentId(standardCategory.getParentId()); child4Node.setIcon(""); child4Node.setType(0); child4Node.setValue(standardCategory.getId()); child4Node.setDisabled(false); child4Node.setRField1(standardCategory.getNum()); child4Node.setRField2(standardCategory.getName()); child4Node.setEntity(standardCategory); pcNode.addChildren(child4Node); map5.put(standardCategory.getId(), child4Node); } } if(map5.size()>0){ CommonGenericTree child5Node; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map5.containsKey(standardCategory.getParentId())) { pcNode = map5.get(standardCategory.getParentId()); child5Node = new CommonGenericTree<>(); child5Node.setKey(standardCategory.getId()); child5Node.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); child5Node.setParentId(standardCategory.getParentId()); child5Node.setIcon(""); child5Node.setType(0); child5Node.setValue(standardCategory.getId()); child5Node.setDisabled(false); child5Node.setRField1(standardCategory.getNum()); child5Node.setRField2(standardCategory.getName()); child5Node.setEntity(standardCategory); pcNode.addChildren(child5Node); map6.put(standardCategory.getId(), child5Node); } } if(map6.size()>0){ CommonGenericTree child6Node; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map6.containsKey(standardCategory.getParentId())) { pcNode = map6.get(standardCategory.getParentId()); child6Node = new CommonGenericTree<>(); child6Node.setKey(standardCategory.getId()); child6Node.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); child6Node.setParentId(standardCategory.getParentId()); child6Node.setIcon(""); child6Node.setType(0); child6Node.setValue(standardCategory.getId()); child6Node.setDisabled(false); child6Node.setRField1(standardCategory.getNum()); child6Node.setRField2(standardCategory.getName()); child6Node.setEntity(standardCategory); pcNode.addChildren(child6Node); map7.put(standardCategory.getId(), child6Node); } } if(map7.size()>0){ CommonGenericTree child7Node; for (RepairStandardCategory standardCategory : repairStandardCategoryList) { if (map.containsKey(standardCategory.getParentId())) { pcNode = map7.get(standardCategory.getParentId()); child7Node = new CommonGenericTree<>(); child7Node.setKey(standardCategory.getId()); child7Node.setTitle(standardCategory.getNum() + "/" + standardCategory.getName()); child7Node.setParentId(standardCategory.getParentId()); child7Node.setIcon(""); child7Node.setType(0); child7Node.setValue(standardCategory.getId()); child7Node.setDisabled(false); child7Node.setRField1(standardCategory.getNum()); child7Node.setRField2(standardCategory.getName()); child7Node.setEntity(standardCategory); // pcNode.addChildren(child7Node); // map7.put(cause.getId(), child7Node); } } } } } } } } } return list; } }