diff --git hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java
index 5f52f85..2ae4872 100644
--- hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java
+++ hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java
@@ -44,6 +44,14 @@ public static ResourceUtilization newInstance(int pmem, int vmem, float cpu) {
return utilization;
}
+ @Public
+ @Unstable
+ public static ResourceUtilization newInstance(
+ ResourceUtilization resourceUtil) {
+ return newInstance(resourceUtil.getPhysicalMemory(),
+ resourceUtil.getVirtualMemory(), resourceUtil.getCPU());
+ }
+
/**
* Get used virtual memory.
*
@@ -147,4 +155,18 @@ public void addTo(int pmem, int vmem, float cpu) {
this.setVirtualMemory(this.getVirtualMemory() + vmem);
this.setCPU(this.getCPU() + cpu);
}
+
+ /**
+ * Subtract utilization from the current one.
+ * @param pmem Physical memory to be subtracted.
+ * @param vmem Virtual memory to be subtracted.
+ * @param cpu CPU utilization to be subtracted.
+ */
+ @Public
+ @Unstable
+ public void subtractFrom(int pmem, int vmem, float cpu) {
+ this.setPhysicalMemory(this.getPhysicalMemory() - pmem);
+ this.setVirtualMemory(this.getVirtualMemory() - vmem);
+ this.setCPU(this.getCPU() - cpu);
+ }
}
\ No newline at end of file
diff --git hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/api/records/TestResourceUtilization.java hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/api/records/TestResourceUtilization.java
new file mode 100644
index 0000000..5934846
--- /dev/null
+++ hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/api/records/TestResourceUtilization.java
@@ -0,0 +1,63 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.hadoop.yarn.api.records;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+public class TestResourceUtilization {
+
+ @Test
+ public void testResourceUtilization() {
+ ResourceUtilization u1 = ResourceUtilization.newInstance(10, 20, 0.5f);
+ ResourceUtilization u2 = ResourceUtilization.newInstance(u1);
+ ResourceUtilization u3 = ResourceUtilization.newInstance(10, 20, 0.5f);
+ ResourceUtilization u4 = ResourceUtilization.newInstance(20, 20, 0.5f);
+ ResourceUtilization u5 = ResourceUtilization.newInstance(30, 40, 0.8f);
+
+ Assert.assertEquals(u1, u2);
+ Assert.assertEquals(u1, u3);
+ Assert.assertNotEquals(u1, u4);
+ Assert.assertNotEquals(u2, u5);
+ Assert.assertNotEquals(u4, u5);
+
+ Assert.assertTrue(u1.hashCode() == u2.hashCode());
+ Assert.assertTrue(u1.hashCode() == u3.hashCode());
+ Assert.assertFalse(u1.hashCode() == u4.hashCode());
+ Assert.assertFalse(u2.hashCode() == u5.hashCode());
+ Assert.assertFalse(u4.hashCode() == u5.hashCode());
+
+ Assert.assertTrue(u1.getPhysicalMemory() == 10);
+ Assert.assertFalse(u1.getVirtualMemory() == 10);
+ Assert.assertTrue(u1.getCPU() == 0.5f);
+
+ Assert.assertEquals("", u1.toString());
+
+ u1.addTo(10, 0, 0.0f);
+ Assert.assertNotEquals(u1, u2);
+ Assert.assertEquals(u1, u4);
+ u1.addTo(10, 20, 0.3f);
+ Assert.assertEquals(u1, u5);
+ u1.subtractFrom(10, 20, 0.3f);
+ Assert.assertEquals(u1, u4);
+ u1.subtractFrom(10, 0, 0.0f);
+ Assert.assertEquals(u1, u3);
+ }
+}