feat(task): allow docker configuration of network mode

Co-authored-by: tdenimal <thomas.denimal@gmail.com>
kestra-io · Apr 5, 2022 · ddec825 · ddec825
1 parent 3a08b1d
commit ddec825
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/core/src/main/java/io/kestra/core/tasks/scripts/AbstractBash.java b/core/src/main/java/io/kestra/core/tasks/scripts/AbstractBash.java
@@ -351,11 +351,16 @@ public static class DockerOptions {
         @PluginProperty(dynamic = true)
         protected List<String> entryPoint;
 
-
         @Schema(
             title = "Docker extra host to use"
         )
         @PluginProperty(dynamic = true)
         protected List<String> extraHosts;
+
+        @Schema(
+            title = "Docker network mode to use"
+        )
+        @PluginProperty(dynamic = true)
+        protected String networkMode;
     }
 }
diff --git a/core/src/main/java/io/kestra/core/tasks/scripts/runners/DockerScriptRunner.java b/core/src/main/java/io/kestra/core/tasks/scripts/runners/DockerScriptRunner.java
@@ -172,6 +172,10 @@ public RunResult run(
                 hostConfig.withExtraHosts(runContext.render(abstractBash.getDockerOptions().getExtraHosts(), additionalVars).toArray(String[]::new));
             }
 
+            if (abstractBash.getDockerOptions().getNetworkMode() != null) {
+                hostConfig.withNetworkMode(runContext.render(abstractBash.getDockerOptions().getNetworkMode(), additionalVars));
+            }
+
             container
                 .withHostConfig(hostConfig)
                 .withCmd(commandsWithInterpreter)