Only apply the "unstick" task pending workaround to V2 queues

OasisLMF · Jan 20, 2025 · 0d9733d · 0d9733d
1 parent 9e7e7c4
commit 0d9733d
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/kubernetes/worker-controller/src/autoscaler.py b/kubernetes/worker-controller/src/autoscaler.py
@@ -166,14 +166,15 @@ async def parse_queued_pending(self, msg) -> [RunningAnalysis]:
 
             # Check for pending analyses
             if (queue_name not in ['celery', 'celery-v2', 'task-controller']):
-                queued_task_count = entry.get('queue', {}).get('queued_count', 0)  # sub-task queued (API DB status)
-                queue_message_count = entry.get('queue', {}).get('queue_message_count', 0)  # queue has messages
-                queued_count = max(queued_task_count, queue_message_count)
-
-                if (queued_count > 0) and not analyses_list:
-                    # a task is queued, but no analyses are running.
-                    # worker-controller might have missed the analysis displatch
-                    pending_analyses[f'pending-task_{queue_name}'] = RunningAnalysis(id=None, tasks=1, queue_names=[queue_name], priority=4)
+                if queue_name.endswith('v2'):
+                    queued_task_count = entry.get('queue', {}).get('queued_count', 0)  # sub-task queued (API DB status)
+                    queue_message_count = entry.get('queue', {}).get('queue_message_count', 0)  # queue has messages
+                    queued_count = max(queued_task_count, queue_message_count)
+
+                    if (queued_count > 0) and not analyses_list:
+                        # a task is queued, but no analyses are running.
+                        # worker-controller might have missed the analysis displatch
+                        pending_analyses[f'pending-task_{queue_name}'] = RunningAnalysis(id=None, tasks=1, queue_names=[queue_name], priority=4)
 
         return pending_analyses