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

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

@@ -20,6 +20,7 @@ import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import java.util.ArrayList;
 import java.util.List;
 
 @RestController
@@ -123,19 +124,20 @@ public class AgentNewController extends BaseController {
              agentId = getAgent().getId();
         }
         List<AgentNew> bySubAgent = agentNewService.getBySubAgent(agentId);
+        List<AgentNew> allList = new ArrayList<>(bySubAgent);
         if(getAgent().getParentId() == null && !bySubAgent.isEmpty()){
             for (AgentNew agentNew : bySubAgent) {
                 List<AgentNew> bySubAgent1 = agentNewService.getBySubAgent(agentNew.getId());
                 if(!bySubAgent1.isEmpty()){
-                    bySubAgent.addAll(bySubAgent1);
+                    allList.addAll(bySubAgent1);
                 }
             }
         }
         if(type == 1){
             AgentNew agentNew = agentNewService.getById(agentId);
-            bySubAgent.add(agentNew);
+            allList.add(agentNew);
         }
-        return ResultData.ok(bySubAgent);
+        return ResultData.ok(allList);
     }
 
 }