From b732d54812479746cfdea2a9da4cc7ba2bee470a Mon Sep 17 00:00:00 2001
From: sabersolooki <saber.solooki@adjust.com>
Date: Wed, 7 May 2025 14:52:33 +0200
Subject: [PATCH] Fix(Distro): bring back distro logic

---
 arq/worker.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/arq/worker.py b/arq/worker.py
index f0fcbfc0..90b9b5dc 100644
--- a/arq/worker.py
+++ b/arq/worker.py
@@ -200,6 +200,7 @@ def __init__(
         self,
         functions: Sequence[Union[Function, 'WorkerCoroutine']] = (),
         *,
+        distribution_index: Optional[int] = None,
         queue_name: Optional[str] = default_queue_name,
         use_stream: bool = False,
         consumer_group_name: str = default_consumer_group,
@@ -241,6 +242,10 @@ def __init__(
                 queue_name = redis_pool.default_queue_name
             else:
                 raise ValueError('If queue_name is absent, redis_pool must be present.')
+
+        if distribution_index is not None:
+            queue_name = f'{queue_name}_{distribution_index}'
+
         self.queue_name = queue_name
         self.use_stream = use_stream
         self.consumer_group_name = consumer_group_name