diff --git a/virtual-patient-model/src/main/java/com/supervision/vo/manage/TreatmentPlanTreeNode.java b/virtual-patient-model/src/main/java/com/supervision/vo/manage/TreatmentPlanTreeNode.java index b2dc2f29..0599b99b 100644 --- a/virtual-patient-model/src/main/java/com/supervision/vo/manage/TreatmentPlanTreeNode.java +++ b/virtual-patient-model/src/main/java/com/supervision/vo/manage/TreatmentPlanTreeNode.java @@ -47,4 +47,14 @@ public class TreatmentPlanTreeNode { } + + public void addNode(TreatmentPlanTreeNode disposalPlanNode){ + if (null == child){ + child = new ArrayList<>(); + } + boolean find = child.stream().anyMatch(node -> node.getName().equals(disposalPlanNode.getName())); + if (!find){ + child.add(disposalPlanNode); + } + } }