package org.jeecg.modules.eam.service.impl;
|
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.jeecg.common.api.vo.CommonGenericTree;
|
import org.jeecg.modules.eam.entity.ShutdownCauseCategory;
|
import org.jeecg.modules.eam.mapper.ShutDownCauseCategoryMapper;
|
import org.jeecg.modules.eam.service.IShutDownCauseCategoryService;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @Description: mom_eam_shutdown_cause_category
|
* @Author: jeecg-boot
|
* @Date: 2023-03-14
|
* @Version: V1.0
|
*/
|
@Service
|
public class ShutDownCauseCategoryServiceImpl extends ServiceImpl<ShutDownCauseCategoryMapper, ShutdownCauseCategory> implements IShutDownCauseCategoryService {
|
|
@Override
|
public List<CommonGenericTree> loadTree() {
|
List<ShutdownCauseCategory> shutdownCauseCategoryList = this.lambdaQuery().eq(ShutdownCauseCategory::getDelFlag, "0").list();
|
List<CommonGenericTree> commonGenericTrees = loadTree(shutdownCauseCategoryList);
|
return commonGenericTrees;
|
}
|
|
@SuppressWarnings("unchecked")
|
public List<CommonGenericTree> loadTree(List<ShutdownCauseCategory> shutdownCauseCategoryList) {
|
@SuppressWarnings("unused")
|
ShutdownCauseCategory shutdownCauseCategory = new ShutdownCauseCategory();
|
List<CommonGenericTree> list = new ArrayList<>();
|
Map<String, CommonGenericTree> map = new HashMap<>();
|
Map<String, CommonGenericTree> map2 = new HashMap<>();
|
Map<String, CommonGenericTree> map3 = new HashMap<>();
|
Map<String, CommonGenericTree> map4 = new HashMap<>();
|
Map<String, CommonGenericTree> map5 = new HashMap<>();
|
Map<String, CommonGenericTree> map6 = new HashMap<>();
|
Map<String, CommonGenericTree> map7 = new HashMap<>();
|
CommonGenericTree<ShutdownCauseCategory> node = new CommonGenericTree<>();
|
node.setKey("-1");
|
node.setValue("-1");
|
node.setTitle("停机原因");
|
node.setRField1("");
|
node.setRField2("");
|
node.setDisabled(false);
|
node.setEntity(new ShutdownCauseCategory());
|
list.add(node);
|
if (CollectionUtils.isNotEmpty(list)) {
|
CommonGenericTree<ShutdownCauseCategory> pcNode;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (causeCategory.getParentId().equals("-1")) {
|
pcNode = new CommonGenericTree<>();
|
pcNode.setKey(causeCategory.getId());
|
pcNode.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
pcNode.setParentId(node.getKey());
|
pcNode.setIcon("");
|
pcNode.setType(0);
|
pcNode.setValue(causeCategory.getId());
|
pcNode.setDisabled(false);
|
pcNode.setRField1(causeCategory.getNum());
|
pcNode.setRField2(causeCategory.getName());
|
pcNode.setEntity(causeCategory);
|
node.addChildren(pcNode);
|
map.put(causeCategory.getId(), pcNode);
|
}
|
}
|
CommonGenericTree<ShutdownCauseCategory> childNode;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map.containsKey(causeCategory.getParentId())) {
|
pcNode = map.get(causeCategory.getParentId());
|
childNode = new CommonGenericTree<>();
|
childNode.setKey(causeCategory.getId());
|
childNode.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
childNode.setParentId(causeCategory.getParentId());
|
childNode.setIcon("");
|
childNode.setType(0);
|
childNode.setValue(causeCategory.getId());
|
childNode.setDisabled(false);
|
childNode.setRField1(causeCategory.getNum());
|
childNode.setRField2(causeCategory.getName());
|
childNode.setEntity(causeCategory);
|
pcNode.addChildren(childNode);
|
map2.put(causeCategory.getId(), childNode);
|
}
|
}
|
if (map2.size() > 0) {
|
CommonGenericTree<ShutdownCauseCategory> child2Node;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map2.containsKey(causeCategory.getParentId())) {
|
pcNode = map2.get(causeCategory.getParentId());
|
child2Node = new CommonGenericTree<>();
|
child2Node.setKey(causeCategory.getId());
|
child2Node.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
child2Node.setParentId(causeCategory.getParentId());
|
child2Node.setIcon("");
|
child2Node.setType(0);
|
child2Node.setValue(causeCategory.getId());
|
child2Node.setDisabled(false);
|
child2Node.setRField1(causeCategory.getNum());
|
child2Node.setRField2(causeCategory.getName());
|
child2Node.setEntity(causeCategory);
|
pcNode.addChildren(child2Node);
|
map3.put(causeCategory.getId(), child2Node);
|
}
|
}
|
if (map3.size() > 0) {
|
CommonGenericTree<ShutdownCauseCategory> child3Node;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map3.containsKey(causeCategory.getParentId())) {
|
pcNode = map3.get(causeCategory.getParentId());
|
child3Node = new CommonGenericTree<>();
|
child3Node.setKey(causeCategory.getId());
|
child3Node.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
child3Node.setParentId(causeCategory.getParentId());
|
child3Node.setIcon("");
|
child3Node.setType(0);
|
child3Node.setValue(causeCategory.getId());
|
child3Node.setDisabled(false);
|
child3Node.setRField1(causeCategory.getNum());
|
child3Node.setRField2(causeCategory.getName());
|
child3Node.setEntity(causeCategory);
|
pcNode.addChildren(child3Node);
|
map4.put(causeCategory.getId(), child3Node);
|
}
|
}
|
if (map4.size() > 0) {
|
CommonGenericTree<ShutdownCauseCategory> child4Node;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map4.containsKey(causeCategory.getParentId())) {
|
pcNode = map4.get(causeCategory.getParentId());
|
child4Node = new CommonGenericTree<>();
|
child4Node.setKey(causeCategory.getId());
|
child4Node.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
child4Node.setParentId(causeCategory.getParentId());
|
child4Node.setIcon("");
|
child4Node.setType(0);
|
child4Node.setValue(causeCategory.getId());
|
child4Node.setDisabled(false);
|
child4Node.setRField1(causeCategory.getNum());
|
child4Node.setRField2(causeCategory.getName());
|
child4Node.setEntity(causeCategory);
|
pcNode.addChildren(child4Node);
|
map5.put(causeCategory.getId(), child4Node);
|
}
|
}
|
if (map5.size() > 0) {
|
CommonGenericTree<ShutdownCauseCategory> child5Node;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map5.containsKey(causeCategory.getParentId())) {
|
pcNode = map5.get(causeCategory.getParentId());
|
child5Node = new CommonGenericTree<>();
|
child5Node.setKey(causeCategory.getId());
|
child5Node.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
child5Node.setParentId(causeCategory.getParentId());
|
child5Node.setIcon("");
|
child5Node.setType(0);
|
child5Node.setValue(causeCategory.getId());
|
child5Node.setDisabled(false);
|
child5Node.setRField1(causeCategory.getNum());
|
child5Node.setRField2(causeCategory.getName());
|
child5Node.setEntity(causeCategory);
|
pcNode.addChildren(child5Node);
|
map6.put(causeCategory.getId(), child5Node);
|
}
|
}
|
if (map6.size() > 0) {
|
CommonGenericTree<ShutdownCauseCategory> child6Node;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map6.containsKey(causeCategory.getParentId())) {
|
pcNode = map6.get(causeCategory.getParentId());
|
child6Node = new CommonGenericTree<>();
|
child6Node.setKey(causeCategory.getId());
|
child6Node.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
child6Node.setParentId(causeCategory.getParentId());
|
child6Node.setIcon("");
|
child6Node.setType(0);
|
child6Node.setValue(causeCategory.getId());
|
child6Node.setDisabled(false);
|
child6Node.setRField1(causeCategory.getNum());
|
child6Node.setRField2(causeCategory.getName());
|
child6Node.setEntity(causeCategory);
|
pcNode.addChildren(child6Node);
|
map7.put(causeCategory.getId(), child6Node);
|
}
|
}
|
if (map7.size() > 0) {
|
CommonGenericTree<ShutdownCauseCategory> child7Node;
|
for (ShutdownCauseCategory causeCategory : shutdownCauseCategoryList) {
|
if (map.containsKey(causeCategory.getParentId())) {
|
pcNode = map7.get(causeCategory.getParentId());
|
child7Node = new CommonGenericTree<>();
|
child7Node.setKey(causeCategory.getId());
|
child7Node.setTitle(causeCategory.getNum() + "/" + causeCategory.getName());
|
child7Node.setParentId(causeCategory.getParentId());
|
child7Node.setIcon("");
|
child7Node.setType(0);
|
child7Node.setValue(causeCategory.getId());
|
child7Node.setDisabled(false);
|
child7Node.setRField1(causeCategory.getNum());
|
child7Node.setRField2(causeCategory.getName());
|
child7Node.setEntity(causeCategory);
|
// pcNode.addChildren(child7Node);
|
// map7.put(cause.getId(), child7Node);
|
}
|
}
|
}
|
}
|
}
|
}
|
}
|
}
|
}
|
return list;
|
}
|
}
|
