获取树形分类接口

ltkg-service/src/main/java/com/diagbot/dto/TreeNodeDTO.java

@@ -1,25 +1,73 @@
 package com.diagbot.dto;
 
-import lombok.Getter;
-import lombok.Setter;
-
 /**
  * @Description:
  * @author: gaodm
  * @time: 2020/3/18 9:18
  */
-@Getter
-@Setter
 public class TreeNodeDTO {
+
     private String icon;
+    private String iconOpen;
+    private String iconClose;
     private Integer id;
     private String name;
     private Integer pId;
 
-    public TreeNodeDTO(String icon, Integer id, String name, Integer pId) {
+    public TreeNodeDTO(String icon, String iconOpen, String iconClose, Integer id, String name, Integer pId) {
         this.icon = icon;
+        this.iconOpen = iconOpen;
+        this.iconClose = iconClose;
+        this.id = id;
+        this.name = name;
+        this.pId = pId;
+    }
+
+    public String getIcon() {
+        return icon;
+    }
+
+    public void setIcon(String icon) {
+        this.icon = icon;
+    }
+
+    public String getIconOpen() {
+        return iconOpen;
+    }
+
+    public void setIconOpen(String iconOpen) {
+        this.iconOpen = iconOpen;
+    }
+
+    public String getIconClose() {
+        return iconClose;
+    }
+
+    public void setIconClose(String iconClose) {
+        this.iconClose = iconClose;
+    }
+
+    public Integer getId() {
+        return id;
+    }
+
+    public void setId(Integer id) {
         this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
         this.name = name;
+    }
+
+    public Integer getpId() {
+        return pId;
+    }
+
+    public void setpId(Integer pId) {
         this.pId = pId;
     }
 }

ltkg-service/src/main/java/com/diagbot/facade/KgFacade.java

@@ -124,7 +124,7 @@ public class KgFacade extends KgServiceImpl {
     private TreeDTO getTreeDTO(String tree) {
         TreeDTO treeDTO = new TreeDTO();
         List<TreeNodeDTO> nodes = new ArrayList<>();
-        Tree treeObj = FastJsonUtils.getJsonToBean(tree, Tree.class);
+        Tree treeObj = FastJsonUtils.getJsonToBean(jsonStr, Tree.class);
         if (null != treeObj) {
             Map<String, Integer> treeMap = new HashMap<>();
             treeMap.put("根节点", 0);
@@ -140,14 +140,19 @@ public class KgFacade extends KgServiceImpl {
             treeMap.put(treeObj.getName(), treeMap.size());
         }
 
-        TreeNodeDTO node = new TreeNodeDTO("M", treeMap.get(treeObj.getName()), treeObj.getName(), treeMap.get(pName));
+        TreeNodeDTO node = new TreeNodeDTO("/images/icon.png",
+                "/images/iconOpen.png",
+                "/images/iconClose.png",
+                treeMap.get(treeObj.getName()), treeObj.getName(), treeMap.get(pName));
         nodes.add(node);
         if (ListUtil.isNotEmpty(treeObj.getSNode())) {
             for (Tree tree : treeObj.getSNode()) {
                 addTree(nodes, tree, treeMap, treeObj.getName());
             }
         } else {
-            node.setIcon("E");
+            node.setIcon("/images/node.png");
+            node.setIconOpen("");
+            node.setIconClose("");
         }
     }