增加元数据目录树形查询

spider-flow-web/src/main/java/org/spiderflow/controller/MetaDataCategoryController.java

 package org.spiderflow.controller;
 
 import cn.hutool.core.util.IdUtil;
-import lombok.RequiredArgsConstructor;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
 import org.spiderflow.entity.MetaDataCategory;
 import org.spiderflow.service.MetaDataCategoryService;
 import org.spiderflow.util.AssertUtils;
@@ -13,6 +14,7 @@ import org.springframework.web.bind.annotation.*;
 import javax.annotation.Resource;
 import java.util.List;
 
+@Api(tags = "元数据目录")
 @RestController
 @RequestMapping(value = "/spiderflow/metadataCategory")
 public class MetaDataCategoryController {
@@ -58,7 +60,7 @@ public class MetaDataCategoryController {
     public ResultVo add(@RequestBody MetaDataCategory metaDataCategory) {
         AssertUtils.notNull(metaDataCategory.getCategoryName(), "缺少类别目录名称");
         AssertUtils.notNull(metaDataCategory.getStandardLevel(), "缺少类别目录级别");
-        metaDataCategory.setCategoryCode(IdUtil.fastSimpleUUID());
+        metaDataCategory.setCategoryCode(IdUtil.simpleUUID());
         boolean b = metaDataCategoryService.save(metaDataCategory);
         return b ? ResultVo.success("操作成功！") : ResultVo.fail("操作失败");
     }
@@ -98,4 +100,11 @@ public class MetaDataCategoryController {
         boolean b = metaDataCategoryService.removeById(id);
         return b ? ResultVo.success("操作成功！") : ResultVo.fail("操作失败！");
     }
+
+    @ApiOperation("树形查询目录")
+    @GetMapping("/tree")
+    public ResultVo getCategoryTree(@RequestParam(value = "categoryCode", required = false) Integer categoryCode,
+                                    @RequestParam(value = "currentLevel", defaultValue = "0") Integer currentLevel){
+        return ResultVo.success(metaDataCategoryService.getCategoryTree(categoryCode, currentLevel));
+    }
 }
\ No newline at end of file

spider-flow-web/src/main/java/org/spiderflow/entity/MetaDataCategory.java

@@ -35,9 +35,14 @@ public class MetaDataCategory {
     private String categoryCode;
 
     /**
-     *  游戏品种名称
+     *  目录级别（顶级：0）
      */
-    private String standardLevel;
+    private Integer standardLevel;
+
+    /**
+     *  子目录数
+     */
+    private Integer subCount;
 
     /**
      *  createTime

spider-flow-web/src/main/java/org/spiderflow/service/MetaDataCategoryService.java

@@ -11,6 +11,8 @@ import org.spiderflow.vo.PageVo;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
+import java.util.ArrayList;
+import java.util.List;
 
 @Service
 public class MetaDataCategoryService extends ServiceImpl<MetaDataCategoryDao, MetaDataCategory> {
@@ -22,4 +24,14 @@ public class MetaDataCategoryService extends ServiceImpl<MetaDataCategoryDao, Me
         LambdaQueryWrapper<MetaDataCategory> wrapper = new LambdaQueryWrapper<>();
         return PagingUtils.paging(req, metaDataCategoryDao, wrapper, MetaDataCategory.class);
     }
+
+    public List<MetaDataCategory> getCategoryTree(Integer categoryCode, Integer currentLevel){
+        List<MetaDataCategory> list;
+        if(null == categoryCode){
+            list = lambdaQuery().eq(MetaDataCategory::getStandardLevel, 0).list();
+        }else {
+            list = lambdaQuery().eq(MetaDataCategory::getCategoryCode, categoryCode).eq(MetaDataCategory::getStandardLevel, ++currentLevel).orderByAsc(MetaDataCategory::getId).list();
+        }
+        return list;
+    }
 }
\ No newline at end of file