feat(cli): add a worker-thread options on standalone command

kestra-io · Apr 2, 2022 · bd3db0c · bd3db0c
1 parent f35e95f
commit bd3db0c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/cli/src/main/java/io/kestra/cli/commands/servers/StandAloneCommand.java b/cli/src/main/java/io/kestra/cli/commands/servers/StandAloneCommand.java
@@ -30,6 +30,10 @@ public class StandAloneCommand extends AbstractCommand {
     @CommandLine.Option(names = {"-f", "--flow-path"}, description = "the flow path containing flow to inject at startup (when running with a memory flow repository)")
     private File flowPath;
 
+
+    @CommandLine.Option(names = {"--worker-thread"}, description = "the number of worker thread")
+    private Integer workerThread;
+
     public StandAloneCommand() {
         super(true);
     }
@@ -55,6 +59,11 @@ public Integer call() throws Exception {
         }
 
         StandAloneRunner standAloneRunner = applicationContext.getBean(StandAloneRunner.class);
+
+        if (this.workerThread != null) {
+            standAloneRunner.setWorkerThread(this.workerThread);
+        }
+
         standAloneRunner.run();
 
         this.shutdownHook(standAloneRunner::close);

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -66,7 +66,7 @@ services:
         /usr/bin/wait-for-it -t 60 elasticsearch:9200 && \
         curl -s -o /tmp/examples.zip https://kestra.io/examples.zip && \
         unzip -o -d /tmp/examples /tmp/examples.zip > /dev/null && \
-        /app/kestra server standalone --flow-path=/tmp/examples
+        /app/kestra server standalone --flow-path=/tmp/examples --worker-thread=128
     volumes:
       - kestra-data:/app/storage
       - /var/run/docker.sock:/var/run/docker.sock