lyhzzz 4 月之前
父节点
当前提交
4d02874985
共有 1 个文件被更改,包括 5 次插入2 次删除
  1. 5 2
      src/main/java/com/fdkankan/agent/controller/AgentNewController.java

+ 5 - 2
src/main/java/com/fdkankan/agent/controller/AgentNewController.java

@@ -123,14 +123,17 @@ public class AgentNewController extends BaseController {
         if(agentId == null){
              agentId = getAgent().getId();
         }
-        List<AgentNew> bySubAgent = agentNewService.getBySubAgent(agentId);
-        List<AgentNew> allList = new ArrayList<>(bySubAgent);
+        List<AgentNew> allList = new ArrayList<>();
         if(type == 1){
             List<AgentNew> agentNewList = agentNewService.getRecursionByAgentId(agentId);
             allList.addAll(agentNewList);
             AgentNew agentNew = agentNewService.getById(agentId);
             allList.add(agentNew);
+        }else {
+            List<AgentNew> bySubAgent = agentNewService.getBySubAgent(agentId);
+            allList.addAll(bySubAgent);
         }
+
         return ResultData.ok(allList);
     }