This is an automated email from the ASF dual-hosted git repository. bharathkk pushed a commit to branch SAMZA-2798 in repository https://gitbox.apache.org/repos/asf/samza.git
commit 4ae386e48d4c679eafb0eb046ec70fc6c16bb97f Author: Bharath Kumarasubramanian <[email protected]> AuthorDate: Wed Nov 22 12:00:40 2023 -0800 SAMZA-2798: Populate worker.opts in environment variable only if available --- samza-core/src/main/java/org/apache/samza/job/ShellCommandBuilder.java | 3 ++- .../src/test/java/org/apache/samza/job/TestShellCommandBuilder.java | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/samza-core/src/main/java/org/apache/samza/job/ShellCommandBuilder.java b/samza-core/src/main/java/org/apache/samza/job/ShellCommandBuilder.java index 504d9a26f..4262145ee 100644 --- a/samza-core/src/main/java/org/apache/samza/job/ShellCommandBuilder.java +++ b/samza-core/src/main/java/org/apache/samza/job/ShellCommandBuilder.java @@ -44,9 +44,10 @@ public class ShellCommandBuilder extends CommandBuilder { envBuilder.put(ShellCommandConfig.ENV_CONTAINER_ID, this.id); envBuilder.put(ShellCommandConfig.ENV_COORDINATOR_URL, this.url.toString()); envBuilder.put(ShellCommandConfig.ENV_JAVA_OPTS, shellCommandConfig.getTaskOpts().orElse("")); - envBuilder.put(ShellCommandConfig.WORKER_JVM_OPTS, shellCommandConfig.getWorkerOpts().orElse("")); envBuilder.put(ShellCommandConfig.ENV_ADDITIONAL_CLASSPATH_DIR, shellCommandConfig.getAdditionalClasspathDir().orElse("")); + shellCommandConfig.getWorkerOpts() + .ifPresent(workerOpts -> envBuilder.put(ShellCommandConfig.WORKER_JVM_OPTS, workerOpts)); shellCommandConfig.getJavaHome().ifPresent(javaHome -> envBuilder.put(ShellCommandConfig.ENV_JAVA_HOME, javaHome)); return envBuilder.build(); } diff --git a/samza-core/src/test/java/org/apache/samza/job/TestShellCommandBuilder.java b/samza-core/src/test/java/org/apache/samza/job/TestShellCommandBuilder.java index afb6bfeec..4acba9487 100644 --- a/samza-core/src/test/java/org/apache/samza/job/TestShellCommandBuilder.java +++ b/samza-core/src/test/java/org/apache/samza/job/TestShellCommandBuilder.java @@ -45,7 +45,6 @@ public class TestShellCommandBuilder { ShellCommandConfig.ENV_CONTAINER_ID, "1", ShellCommandConfig.ENV_COORDINATOR_URL, URL_STRING, ShellCommandConfig.ENV_JAVA_OPTS, "", - ShellCommandConfig.WORKER_JVM_OPTS, "", ShellCommandConfig.ENV_ADDITIONAL_CLASSPATH_DIR, ""); // assertions when command path is not set assertEquals("foo", shellCommandBuilder.buildCommand());
