YARN-10456. RM PartitionQueueMetrics records are named QueueMetrics in Simon metrics registry. Contributed by Eric Payne.

(cherry picked from commit 632f64cadb)
This commit is contained in:
Jim Brennan 2021-07-15 14:23:31 +00:00
parent 450dae7383
commit b3481062e0
2 changed files with 13 additions and 1 deletions

View File

@ -167,7 +167,11 @@ public class QueueMetrics implements MetricsSource {
public QueueMetrics(MetricsSystem ms, String queueName, Queue parent,
boolean enableUserMetrics, Configuration conf) {
registry = new MetricsRegistry(RECORD_INFO);
if (this instanceof PartitionQueueMetrics) {
registry = new MetricsRegistry(P_RECORD_INFO);
} else {
registry = new MetricsRegistry(RECORD_INFO);
}
this.queueName = queueName;
this.parent = parent != null ? parent.getMetrics() : null;

View File

@ -37,6 +37,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CSQueueM
import org.apache.hadoop.yarn.server.utils.BuilderUtils;
import org.apache.hadoop.yarn.util.resource.Resources;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
@ -110,6 +111,13 @@ public class TestPartitionQueueMetrics {
checkResources(partitionSource, 0, 0, 0, 200 * GB, 200, 5 * GB, 5, 5);
checkResources(rootQueueSource, 0, 0, 0, 200 * GB, 200, 5 * GB, 5, 5);
checkResources(q2Source, 0, 0, 0, 0, 0, 3 * GB, 3, 3);
PartitionQueueMetrics pq1 =
new PartitionQueueMetrics(ms, "root.q1", parentQueue, true, CONF, "x");
Assert.assertTrue("Name of registry should be \""
+ PartitionQueueMetrics.P_RECORD_INFO.name() + "\", but was \""
+ pq1.registry.info().name() + "\".", pq1.registry.info().name()
.compareTo(PartitionQueueMetrics.P_RECORD_INFO.name()) == 0);
}
/**