From 15eb84e993d6724e605cc570180023a69bccf8ab Mon Sep 17 00:00:00 2001
From: lanyuanxiaoyao <lanyuanxiaoyao@gmail.com>
Date: Thu, 1 Jun 2023 10:29:33 +0800
Subject: [PATCH] =?UTF-8?q?refactor(yarn-query):=20=E4=BF=AE=E5=A4=8D=20ya?=
 =?UTF-8?q?rn=20=E6=9F=A5=E8=AF=A2=E5=AD=90=E9=98=9F=E5=88=97=E4=B8=8D?=
 =?UTF-8?q?=E5=85=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/yarn/service/QueueService.java    | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/service-yarn-query/src/main/java/com/lanyuanxiaoyao/service/yarn/service/QueueService.java b/service-yarn-query/src/main/java/com/lanyuanxiaoyao/service/yarn/service/QueueService.java
index 666b5dc..6c9aad1 100644
--- a/service-yarn-query/src/main/java/com/lanyuanxiaoyao/service/yarn/service/QueueService.java
+++ b/service-yarn-query/src/main/java/com/lanyuanxiaoyao/service/yarn/service/QueueService.java
@@ -4,6 +4,8 @@ import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.lanyuanxiaoyao.service.configuration.entity.yarn.YarnQueue;
+import java.util.LinkedList;
+import java.util.Queue;
 import org.eclipse.collections.api.block.function.Function;
 import org.eclipse.collections.api.factory.Lists;
 import org.eclipse.collections.api.list.ImmutableList;
@@ -23,10 +25,22 @@ public interface QueueService {
     default Function<YarnQueue, Iterable<YarnQueue>> flatChildren() {
         return queue -> {
             MutableList<YarnQueue> children = Lists.mutable.of(queue);
-            if (ObjectUtil.isNotNull(queue.getQueues())) {
-                queue.getQueues().getQueue()
-                        .tap(child -> child.setQueueName(StrUtil.format("{}.{}", queue.getQueueName(), child.getQueueName())))
-                        .forEach(children::add);
+            Queue<YarnQueue> handleQueue = new LinkedList<>();
+            handleQueue.add(queue);
+
+            YarnQueue parent = handleQueue.poll();
+            while (parent != null) {
+                try {
+                    if (ObjectUtil.isNotNull(parent.getQueues())) {
+                        for (YarnQueue child : queue.getQueues().getQueue()) {
+                            child.setQueueName(StrUtil.format("{}.{}", parent.getQueueName(), child.getQueueName()));
+                            children.add(child);
+                            handleQueue.add(child);
+                        }
+                    }
+                } finally {
+                    parent = handleQueue.poll();
+                }
             }
             return children;
         };