[REEF-1714] Add an option for Unmanaged Driver to YARN Client configuration
JIRA:
[REEF-1714](https://issues.apache.org/jira/browse/REEF-1714)
Pull request:
This closes #1228
diff --git a/lang/java/reef-runtime-yarn/src/main/java/org/apache/reef/runtime/yarn/client/YarnClientConfiguration.java b/lang/java/reef-runtime-yarn/src/main/java/org/apache/reef/runtime/yarn/client/YarnClientConfiguration.java
index 33fb177..0e6733b 100644
--- a/lang/java/reef-runtime-yarn/src/main/java/org/apache/reef/runtime/yarn/client/YarnClientConfiguration.java
+++ b/lang/java/reef-runtime-yarn/src/main/java/org/apache/reef/runtime/yarn/client/YarnClientConfiguration.java
@@ -21,6 +21,7 @@
import org.apache.reef.annotations.audience.ClientSide;
import org.apache.reef.annotations.audience.Public;
import org.apache.reef.client.parameters.DriverConfigurationProviders;
+import org.apache.reef.driver.parameters.DriverIsUnmanaged;
import org.apache.reef.runtime.common.client.CommonRuntimeConfiguration;
import org.apache.reef.runtime.common.client.DriverConfigurationProvider;
import org.apache.reef.runtime.common.client.api.JobSubmissionHandler;
@@ -35,19 +36,20 @@
import org.apache.reef.util.logging.LoggingSetup;
/**
- * A ConfigurationModule for the YARN resourcemanager.
+ * A ConfigurationModule for the YARN resource manager.
*/
@Public
@ClientSide
public class YarnClientConfiguration extends ConfigurationModuleBuilder {
+
static {
LoggingSetup.setupCommonsLogging();
}
public static final OptionalParameter<String> YARN_QUEUE_NAME = new OptionalParameter<>();
public static final OptionalParameter<Integer> YARN_PRIORITY = new OptionalParameter<>();
-
public static final OptionalParameter<Double> JVM_HEAP_SLACK = new OptionalParameter<>();
+ public static final OptionalParameter<Boolean> UNMANAGED_DRIVER = new OptionalParameter<>();
/**
* Configuration provides whose Configuration will be merged into all Driver Configuration.
@@ -63,6 +65,7 @@
.bindNamedParameter(JobQueue.class, YARN_QUEUE_NAME)
.bindNamedParameter(JobPriority.class, YARN_PRIORITY)
.bindNamedParameter(JVMHeapSlack.class, JVM_HEAP_SLACK)
+ .bindNamedParameter(DriverIsUnmanaged.class, UNMANAGED_DRIVER)
.bindImplementation(RuntimeClasspathProvider.class, YarnClasspathProvider.class)
// Bind external constructors. Taken from YarnExternalConstructors.registerClientConstructors
.bindConstructor(org.apache.hadoop.yarn.conf.YarnConfiguration.class, YarnConfigurationConstructor.class)