YARN-7258. Add Node and Rack Hints to Opportunistic Scheduler. (Karth…

…eek Muthyala via asuresh).
apache · Oct 5, 2017 · b733348 · b733348
1 parent 9288206
commit b733348
Show file tree

Hide file tree

Showing 11 changed files with 998 additions and 734 deletions.
diff --git a/...arn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceRequest.java b/...arn/hadoop-yarn-api/src/main/java/org/apache/hadoop/yarn/api/records/ResourceRequest.java
@@ -239,6 +239,22 @@ public ResourceRequestBuilder executionTypeRequest(
       return this;
     }
 
+    /**
+     * Set the <code>executionTypeRequest</code> of the request with 'ensure
+     * execution type' flag set to true.
+     * @see ResourceRequest#setExecutionTypeRequest(
+     * ExecutionTypeRequest)
+     * @param executionType <code>executionType</code> of the request.
+     * @return {@link ResourceRequestBuilder}
+     */
+    @Public
+    @Evolving
+    public ResourceRequestBuilder executionType(ExecutionType executionType) {
+      resourceRequest.setExecutionTypeRequest(
+          ExecutionTypeRequest.newInstance(executionType, true));
+      return this;
+    }
+
     /**
      * Set the <code>allocationRequestId</code> of the request.
      * @see ResourceRequest#setAllocationRequestId(long)