feat: support for custom queues and workers

press/docker/config/supervisor.conf

@@ -163,6 +163,27 @@ process_name=%(program_name)s-%(process_num)d
 
 {% endif %}
 
+{% if custom_workers_config %}
+
+{% for worker_name, worker_details in custom_workers_config.items() %}
+[program:-frappe-bench-{{ worker_name }}-worker]
+command=bench worker --queue {{ worker_name }}
+priority=4
+autostart=true
+autorestart=true
+stdout_logfile=/home/frappe/frappe-bench/logs/{{ worker_name }}.log
+stderr_logfile=/home/frappe/frappe-bench/logs/{{ worker_name }}.error.log
+user=frappe
+stopwaitsecs={{ worker_details["timeout"] }}
+directory=/home/frappe/frappe-bench
+killasgroup=true
+numprocs=1
+process_name=%(program_name)s-%(process_num)d
+{% endfor %}
+
+{% endif %}
+
+
 [program:frappe-bench-redis-cache]
 command=redis-server /home/frappe/frappe-bench/config/redis-cache.conf
 priority=1
@@ -213,6 +234,13 @@ programs=frappe-bench-frappe-schedule,frappe-bench-frappe-default-worker,frappe-
 
 {% endif %}
 
+{% if custom_workers_config %}
+
+[group:frappe-bench-custom_workers]
+programs={%- for worker_name in workers -%}frappe-bench-{{ worker_name }}-worker,{%- endfor %}
+
+{% endif %}
+
 [group:frappe-bench-redis]
 programs=frappe-bench-redis-cache,frappe-bench-redis-queue
 

press/press/doctype/deploy_candidate/deploy_candidate.py

@@ -882,6 +882,8 @@ def _update_app_releases(self) -> None:
 			app.pullable_release = release_pair["new"]["name"]
 
 	def _prepare_build_context(self, no_push: bool):
+		rg: ReleaseGroup = frappe.get_cache_doc("Release Group", self.group)
+
 		repo_path_map = self._clone_repos()
 		pmf = get_package_manager_files(repo_path_map)
 		self._run_prebuild_validations_and_update_step(pmf)
@@ -904,7 +906,7 @@ def _prepare_build_context(self, no_push: bool):
 
 		self._copy_config_files()
 		self._generate_redis_cache_config()
-		self._generate_supervisor_config()
+		self._generate_supervisor_config(rg)
 		self._generate_apps_txt()
 		self.generate_ssh_keys()
 
@@ -1190,15 +1192,25 @@ def _generate_redis_cache_config(self):
 			)
 			f.write(content)
 
-	def _generate_supervisor_config(self):
+	def _generate_supervisor_config(self, rg):
 		supervisor_conf = os.path.join(self.build_directory, "config", "supervisor.conf")
 		with open(supervisor_conf, "w") as f:
 			supervisor_conf_template = "press/docker/config/supervisor.conf"
+			custom_workers_config = self._get_custom_workers(rg)
 			content = frappe.render_template(
-				supervisor_conf_template, {"doc": self}, is_path=True
+				supervisor_conf_template,
+				{"doc": self, "custom_workers_config": custom_workers_config},
+				is_path=True,
 			)
 			f.write(content)
 
+	def _get_custom_workers(self, rg):
+		if rg.common_site_config:
+			common_site_config = json.loads(self.common_site_config) or frappe._dict()
+
+			if "workers" in common_site_config:
+				return common_site_config["workers"]
+
 	def _generate_apps_txt(self):
 		apps_txt = os.path.join(self.build_directory, "apps.txt")
 		with open(apps_txt, "w") as f: