fix: use containerImage istead of DEFAULT_IMAGE when no image in options

kestra-io · Aug 8, 2024 · 7d71f06 · 7d71f06
1 parent ac7fd0e
commit 7d71f06
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/java/io/kestra/plugin/malloy/CLI.java b/src/main/java/io/kestra/plugin/malloy/CLI.java
@@ -7,6 +7,7 @@
 import io.kestra.core.models.tasks.runners.TaskRunner;
 import io.kestra.core.runners.RunContext;
 import io.kestra.plugin.scripts.exec.AbstractExecScript;
+import io.kestra.plugin.scripts.exec.scripts.models.DockerOptions;
 import io.kestra.plugin.scripts.exec.scripts.models.ScriptOutput;
 import io.kestra.plugin.scripts.runner.docker.Docker;
 import io.swagger.v3.oas.annotations.media.Schema;
@@ -76,6 +77,16 @@ public class CLI extends AbstractExecScript {
     @Builder.Default
     private String containerImage = DEFAULT_IMAGE;
 
+    @Override
+    protected DockerOptions injectDefaults(DockerOptions original) {
+        var builder = original.toBuilder();
+        if (original.getImage() == null) {
+            builder.image(this.getContainerImage());
+        }
+
+        return builder.build();
+    }
+
     @Override
     public ScriptOutput run(RunContext runContext) throws Exception {
         List<String> commandsArgs = ScriptService.scriptCommands(