diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/main/java/org/apache/hadoop/yarn/applications/distributedshell/Client.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/main/java/org/apache/hadoop/yarn/applications/distributedshell/Client.java index 61879d0..4866747 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/main/java/org/apache/hadoop/yarn/applications/distributedshell/Client.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/main/java/org/apache/hadoop/yarn/applications/distributedshell/Client.java @@ -645,7 +645,19 @@ public boolean run() throws IOException, YarnException { + ", nodeNumContainers=" + node.getNumContainers()); } - QueueInfo queueInfo = yarnClient.getQueueInfo(this.amQueue); + QueueInfo queueInfo = null; + List allQueuesInfo = yarnClient.getAllQueues(); + for (QueueInfo qInfo : allQueuesInfo) { + if (qInfo.getQueueName().equals(this.amQueue)) { + queueInfo = qInfo; + break; + } + } + if (queueInfo == null) { + String message = "Unknown queueName '" + this.amQueue + "'"; + LOG.error(message); + throw new IOException(message); + } LOG.info("Queue info" + ", queueName=" + queueInfo.getQueueName() + ", queueCurrentCapacity=" + queueInfo.getCurrentCapacity()