From 9def4d4d390e4a86f9588dea5db4557a03b032a4 Mon Sep 17 00:00:00 2001
From: Arun Suresh <asuresh@apache.org>
Date: Fri, 1 Apr 2016 14:57:06 -0700
Subject: [PATCH] YARN-4895. Add subtractFrom method to ResourceUtilization
 class. Contributed by Konstantinos Karanasos.

(cherry picked from commit 82621e38a0445832998bc00693279e23a98605c1)
---
 .../yarn/api/records/ResourceUtilization.java | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java
index 5f52f85f0c6..2ae4872fbcd 100644
--- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java
+++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceUtilization.java
@@ -44,6 +44,14 @@ public abstract class ResourceUtilization implements
     return utilization;
   }
 
+  @Public
+  @Unstable
+  public static ResourceUtilization newInstance(
+      ResourceUtilization resourceUtil) {
+    return newInstance(resourceUtil.getPhysicalMemory(),
+        resourceUtil.getVirtualMemory(), resourceUtil.getCPU());
+  }
+
   /**
    * Get used <em>virtual memory</em>.
    *
@@ -147,4 +155,18 @@ public abstract class ResourceUtilization implements
     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