From 19d1e97075b24788daaf2895da3b46762b461ab3 Mon Sep 17 00:00:00 2001 From: Rory& Date: Tue, 28 Oct 2025 00:01:45 +0100 Subject: [PATCH 19/19] Allow overriding max background task count Signed-off-by: Rory& --- get_sched_tasks.sql | 1 + synapse/config/ratelimiting.py | 1 + synapse/util/task_scheduler.py | 6 ++++++ 3 files changed, 8 insertions(+) create mode 100644 get_sched_tasks.sql diff --git a/get_sched_tasks.sql b/get_sched_tasks.sql new file mode 100644 index 0000000000..680342cfa2 --- /dev/null +++ b/get_sched_tasks.sql @@ -0,0 +1 @@ +select * from scheduled_tasks where status != 'complete'; diff --git a/synapse/config/ratelimiting.py b/synapse/config/ratelimiting.py index 13c9c4dba0..a8add434d2 100644 --- a/synapse/config/ratelimiting.py +++ b/synapse/config/ratelimiting.py @@ -84,6 +84,7 @@ class RatelimitConfig(Config): section = "ratelimiting" def read_config(self, config: JsonDict, **kwargs: Any) -> None: + self.override_max_concurrent_running_tasks = config.get("ratelimiting", {}).get("override_max_concurrent_running_tasks", None) # Load the new-style messages config if it exists. Otherwise fall back # to the old method. if "rc_message" in config: diff --git a/synapse/util/task_scheduler.py b/synapse/util/task_scheduler.py index c1790fd3ae..a083e5b694 100644 --- a/synapse/util/task_scheduler.py +++ b/synapse/util/task_scheduler.py @@ -142,6 +142,12 @@ class TaskScheduler: hook=lambda: {(self.server_name,): len(self._running_tasks)}, ) + if hs.config.ratelimiting.override_max_concurrent_running_tasks is not None: + TaskScheduler.MAX_CONCURRENT_RUNNING_TASKS = ( + hs.config.ratelimiting.override_max_concurrent_running_tasks + ) + logger.warning("Max concurrent running tasks: %s, override: %s", TaskScheduler.MAX_CONCURRENT_RUNNING_TASKS, hs.config.ratelimiting.override_max_concurrent_running_tasks) + def register_action( self, function: Callable[ -- 2.53.0