获取树形结构接口

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

@@ -0,0 +1,18 @@
+package com.diagbot.dto;
+
+import lombok.Getter;
+import lombok.Setter;
+
+import java.util.List;
+
+/**
+ * @Description:
+ * @author: gaodm
+ * @time: 2020/3/17 9:04
+ */
+@Getter
+@Setter
+public class TreeDTO extends BaseNodeDTO {
+    private String icdCode;
+    private List<TreeDTO> sNode;
+}

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

@@ -3,11 +3,13 @@ package com.diagbot.facade;
 import com.diagbot.dto.BaseNodeRSDTO;
 import com.diagbot.dto.GraphDTO;
 import com.diagbot.dto.GraphLabelDTO;
+import com.diagbot.dto.TreeDTO;
 import com.diagbot.exception.CommonErrorCode;
 import com.diagbot.exception.CommonException;
 import com.diagbot.service.impl.KgServiceImpl;
 import com.diagbot.util.ListUtil;
 import com.diagbot.vo.KgQueryVO;
+import com.diagbot.vo.KgTreeVO;
 import org.springframework.stereotype.Component;
 
 import java.util.ArrayList;
@@ -42,4 +44,18 @@ public class KgFacade extends KgServiceImpl {
         }
         return graphLabelDTO;
     }
+
+    public TreeDTO getTreeFac(KgTreeVO kgTreeVO) {
+        TreeDTO treeDTO = new TreeDTO();
+        if (kgTreeVO.getType() == 1 && kgTreeVO.getSubType() == 1) {
+            addNode(treeDTO);
+        } else {
+            throw new CommonException(CommonErrorCode.PARAM_IS_ERROR);
+        }
+        return treeDTO;
+    }
+
+    private void addNode(TreeDTO treeDTO) {
+
+    }
 }

ltkg-service/src/main/java/com/diagbot/vo/KgTreeVO.java

@@ -0,0 +1,23 @@
+package com.diagbot.vo;
+
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Getter;
+import lombok.Setter;
+
+import javax.validation.constraints.NotNull;
+
+/**
+ * @Description:
+ * @author: gaodm
+ * @time: 2020/3/17 9:00
+ */
+@Getter
+@Setter
+public class KgTreeVO {
+    @ApiModelProperty(value = "类型 疾病:1")
+    @NotNull(message = "类型不能为空")
+    private Integer type;
+    @ApiModelProperty(value = "子类型 默认:0 ICD10:1")
+    @NotNull(message = "子类型不能为空")
+    private Integer subType;
+}

ltkg-service/src/main/java/com/diagbot/web/KgController.java

@@ -6,8 +6,10 @@ import com.diagbot.dto.GraphLabelDTO;
 import com.diagbot.dto.NodeDTO;
 import com.diagbot.dto.RespDTO;
 import com.diagbot.dto.SchemaDTO;
+import com.diagbot.dto.TreeDTO;
 import com.diagbot.facade.KgFacade;
 import com.diagbot.vo.KgQueryVO;
+import com.diagbot.vo.KgTreeVO;
 import com.diagbot.vo.SchemaVO;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -52,8 +54,8 @@ public class KgController {
     @ApiOperation(value = "获取分类", notes = "获取分类")
     @PostMapping("/getTree")
     @SysLogger("getTree")
-    public RespDTO<List<Map>> getTree() {
-        return RespDTO.onSuc(kgFacade.getTree());
+    public RespDTO<TreeDTO> getTree(@RequestBody @Valid KgTreeVO kgTreeVO) {
+        return RespDTO.onSuc(kgFacade.getTreeFac(kgTreeVO));
     }
 
     @ApiOperation(value = "获取描述框架", notes = "获取描述框架")