YARN-8625. Aggregate Resource Allocation for each job is not present in ATS. Contributed by Prabhu Joseph.
This commit is contained in:
parent
500dc05924
commit
441339a716
|
@ -19,6 +19,7 @@
|
|||
package org.apache.hadoop.yarn.server.applicationhistoryservice.webapp;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
|
@ -408,6 +409,10 @@ public class TestAHSWebServices extends JerseyTestBase {
|
|||
assertEquals(FinalApplicationStatus.UNDEFINED.toString(),
|
||||
app.get("finalAppStatus"));
|
||||
assertEquals(YarnApplicationState.FINISHED.toString(), app.get("appState"));
|
||||
assertNotNull("Aggregate resource allocation is null",
|
||||
app.get("aggregateResourceAllocation"));
|
||||
assertNotNull("Aggregate Preempted Resource Allocation is null",
|
||||
app.get("aggregatePreemptedResourceAllocation"));
|
||||
}
|
||||
|
||||
@Test
|
||||
|
|
|
@ -28,6 +28,7 @@ import org.apache.hadoop.classification.InterfaceAudience.Public;
|
|||
import org.apache.hadoop.classification.InterfaceStability.Evolving;
|
||||
|
||||
import org.apache.hadoop.yarn.api.records.ApplicationReport;
|
||||
import org.apache.hadoop.yarn.api.records.ApplicationResourceUsageReport;
|
||||
import org.apache.hadoop.yarn.api.records.FinalApplicationStatus;
|
||||
import org.apache.hadoop.yarn.api.records.YarnApplicationState;
|
||||
import org.apache.hadoop.yarn.util.Times;
|
||||
|
@ -67,6 +68,8 @@ public class AppInfo {
|
|||
protected boolean unmanagedApplication;
|
||||
private String appNodeLabelExpression;
|
||||
private String amNodeLabelExpression;
|
||||
private String aggregateResourceAllocation;
|
||||
private String aggregatePreemptedResourceAllocation;
|
||||
|
||||
public AppInfo() {
|
||||
// JAXB needs this
|
||||
|
@ -97,19 +100,23 @@ public class AppInfo {
|
|||
if (app.getPriority() != null) {
|
||||
priority = app.getPriority().getPriority();
|
||||
}
|
||||
if (app.getApplicationResourceUsageReport() != null) {
|
||||
runningContainers = app.getApplicationResourceUsageReport()
|
||||
.getNumUsedContainers();
|
||||
if (app.getApplicationResourceUsageReport().getUsedResources() != null) {
|
||||
allocatedCpuVcores = app.getApplicationResourceUsageReport()
|
||||
.getUsedResources().getVirtualCores();
|
||||
allocatedMemoryMB = app.getApplicationResourceUsageReport()
|
||||
.getUsedResources().getMemorySize();
|
||||
reservedCpuVcores = app.getApplicationResourceUsageReport()
|
||||
.getReservedResources().getVirtualCores();
|
||||
reservedMemoryMB = app.getApplicationResourceUsageReport()
|
||||
.getReservedResources().getMemorySize();
|
||||
ApplicationResourceUsageReport usageReport =
|
||||
app.getApplicationResourceUsageReport();
|
||||
if (usageReport != null) {
|
||||
runningContainers = usageReport.getNumUsedContainers();
|
||||
if (usageReport.getUsedResources() != null) {
|
||||
allocatedCpuVcores = usageReport.getUsedResources().getVirtualCores();
|
||||
allocatedMemoryMB = usageReport.getUsedResources().getMemorySize();
|
||||
reservedCpuVcores = usageReport.getReservedResources().
|
||||
getVirtualCores();
|
||||
reservedMemoryMB = usageReport.getReservedResources().getMemorySize();
|
||||
}
|
||||
aggregateResourceAllocation = usageReport.getMemorySeconds()
|
||||
+ " MB-seconds, " + usageReport.getVcoreSeconds()
|
||||
+ " vcore-seconds";
|
||||
aggregatePreemptedResourceAllocation = usageReport.
|
||||
getPreemptedMemorySeconds() + " MB-seconds, " +
|
||||
usageReport.getPreemptedVcoreSeconds() + " vcore-seconds";
|
||||
}
|
||||
progress = app.getProgress() * 100; // in percent
|
||||
if (app.getApplicationTags() != null && !app.getApplicationTags().isEmpty()) {
|
||||
|
@ -235,4 +242,12 @@ public class AppInfo {
|
|||
public String getAmNodeLabelExpression() {
|
||||
return amNodeLabelExpression;
|
||||
}
|
||||
|
||||
public String getAggregateResourceAllocation() {
|
||||
return aggregateResourceAllocation;
|
||||
}
|
||||
|
||||
public String getAggregatePreemptedResourceAllocation() {
|
||||
return aggregatePreemptedResourceAllocation;
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue