Repository: reef Updated Branches: refs/heads/master be68df345 -> f78355489
[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 Project: http://git-wip-us.apache.org/repos/asf/reef/repo Commit: http://git-wip-us.apache.org/repos/asf/reef/commit/f7835548 Tree: http://git-wip-us.apache.org/repos/asf/reef/tree/f7835548 Diff: http://git-wip-us.apache.org/repos/asf/reef/diff/f7835548 Branch: refs/heads/master Commit: f783554890b1fc1c8ee865a4f706b578f718b411 Parents: be68df3 Author: Sergiy Matusevych <[email protected]> Authored: Fri Jan 13 16:24:00 2017 -0800 Committer: Mariia Mykhailova <[email protected]> Committed: Tue Jan 17 14:39:34 2017 -0800 ---------------------------------------------------------------------- .../reef/runtime/yarn/client/YarnClientConfiguration.java | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/reef/blob/f7835548/lang/java/reef-runtime-yarn/src/main/java/org/apache/reef/runtime/yarn/client/YarnClientConfiguration.java ---------------------------------------------------------------------- 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 @@ package org.apache.reef.runtime.yarn.client; 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.tang.formats.*; 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 @@ public class YarnClientConfiguration extends ConfigurationModuleBuilder { .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)
