Permit to choose gunicorn workers number

kernelci · Nov 17, 2023 · d34a316 · d34a316
1 parent ab549e7
commit d34a316
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lava-master/entrypoint.d/01_setup.sh b/lava-master/entrypoint.d/01_setup.sh
@@ -266,6 +266,12 @@ do
 	done
 done
 
+if [ ! -z "$GUNICORN_WORKERS" ];then
+	echo "DEBUG: set gunicorn workers to $GUNICORN_WORKERS"
+	grep -ri workers /etc
+	sed -i "s,.*WORKERS=.*,WORKERS=$GUNICORN_WORKERS," /etc/lava-server/lava-server-gunicorn || exit $?
+fi
+
 echo "DEBUG: fix owning rights on /etc/lava-server/dispatcher-config"
 chown -Rc lavaserver:lavaserver /etc/lava-server/dispatcher-config
 exit 0
diff --git a/lavalab-gen.py b/lavalab-gen.py
@@ -114,7 +114,7 @@ def main():
             "allowed_hosts",
             "build_args",
             "event_notifications",
-            "groups",
+            "groups", "gunicorn_workers",
             "healthcheck_url", "host", "http_fqdn",
             "loglevel", "lava-coordinator",
             "name",
@@ -199,6 +199,10 @@ def main():
             f_entrypoint.write("exit $?\n")
             f_entrypoint.close()
             os.chmod("%s/entrypoint.d/02_lava-coordinator.sh" % workerdir, 0o755)
+        if "gunicorn_workers" in worker:
+            dockcomp["services"][name]["environment"] = {}
+            dockcomp["services"][name]["environment"]["GUNICORN_WORKERS"] = worker["gunicorn_workers"]
+
         with open(dockcomposeymlpath, 'w') as f:
             yaml.dump(dockcomp, f)
         if "healthcheck_url" in master: