2024/05/04_API文档接口修改

rs-website/rs-website-server/src/main/java/com/upyuns/platform/rs/website/biz/ApiDocBiz.java

@@ -26,12 +26,13 @@ import javax.annotation.Resource;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  *
  */
 @Service
-public class ApiDocBiz extends BaseBiz<ApiDocMapper, ApiDoc>{
+public class ApiDocBiz extends BaseBiz<ApiDocMapper, ApiDoc> {
 
 
     @Resource
@@ -47,35 +48,35 @@ public class ApiDocBiz extends BaseBiz<ApiDocMapper, ApiDoc>{
         Example exampleParams = new Example(ApiCustomParams.class);
         List<ApiDocTreeVo> docTreeVoList = mapper.selectList(query.getSuper());
         for (ApiDocTreeVo apiDocTreeVo : docTreeVoList) {
-            if (StringUtils.isNotBlank(apiDocDTO.getKeyWord())){
+            exampleNode.clear();
+            if (StringUtils.isNotBlank(apiDocDTO.getKeyWord())) {
                 exampleNode.createCriteria()
-                        .andEqualTo("isDel",0)
-                        .andLike("nodeName",'%'+apiDocDTO.getKeyWord()+'%')
-                        .andEqualTo("docId",apiDocTreeVo.getId());
-            }else {
+                        .andEqualTo("isDel", 0)
+                        .andLike("nodeName", '%' + apiDocDTO.getKeyWord() + '%')
+                        .andEqualTo("docId", apiDocTreeVo.getId());
+            } else {
                 exampleNode.createCriteria()
-                        .andEqualTo("isDel",0)
-                        .andEqualTo("docId",apiDocTreeVo.getId());
+                        .andEqualTo("isDel", 0)
+                        .andEqualTo("docId", apiDocTreeVo.getId());
             }
             exampleNode.excludeProperties("respondJson");
-            List<ApiCustomNodeVo> nodeVos = new ArrayList<>();
-            for (ApiCustomNode node : nodeMapper.selectByExample(exampleNode)) {
+            List<ApiCustomNodeVo> nodeVos = nodeMapper.selectByExample(exampleNode).stream().map(apiCustomNode -> {
+                ApiCustomNodeVo nodeVoss = new ApiCustomNodeVo();
+                BeanUtils.copyProperties(apiCustomNode, nodeVoss);
+                exampleParams.clear();
                 exampleParams.createCriteria()
-                        .andEqualTo("nodeId",node.getId())
-                        .andEqualTo("isDel",0)
-                        .andEqualTo("docId",apiDocTreeVo.getId());
+                        .andEqualTo("nodeId", apiCustomNode.getId())
+                        .andEqualTo("isDel", 0)
+                        .andEqualTo("docId", apiDocTreeVo.getId());
                 ApiCustomParams params = paramsMapper.selectOneByExample(exampleParams);
-                if (Objects.isNull(params)){
-                    continue;
+                if (Objects.isNull(params)) {
+                    return nodeVoss;
                 }
-                ApiCustomNodeVo vo = new ApiCustomNodeVo();
-                BeanUtils.copyProperties(node,vo);
-                vo.setMethod(params.getMethod());
-                nodeVos.add(vo);
-                exampleParams.clear();
-            }
+                nodeVoss.setMethod(params.getMethod());
+                return nodeVoss;
+            }).collect(Collectors.toList());
+            apiDocTreeVo.setApiCount(nodeVos.size());
             apiDocTreeVo.setNodeVoList(nodeVos);
-            exampleNode.clear();
         }
         return ObjectRestResponse.succ(docTreeVoList);
     }