From a833364accb827c15c2644403759b0f4627c807d Mon Sep 17 00:00:00 2001 From: Stefan Friedli Date: Mon, 20 Nov 2023 14:31:09 +0100 Subject: [PATCH] Fix empty response on sqs dequeue (#14) --- polytope_server/common/queue/sqs_queue.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/polytope_server/common/queue/sqs_queue.py b/polytope_server/common/queue/sqs_queue.py index 9f396a9..2892aad 100644 --- a/polytope_server/common/queue/sqs_queue.py +++ b/polytope_server/common/queue/sqs_queue.py @@ -1,8 +1,10 @@ import json import logging -from . import queue + import boto3 + from ..metric_collector import SQSQueueMetricCollector +from . import queue class SQSQueue(queue.Queue): @@ -30,7 +32,7 @@ def dequeue(self): VisibilityTimeout=self.visibility_timeout, # If processing takes more seconds, message will be read twice MaxNumberOfMessages=1, ) - if not response["Messages"]: + if "Messages" not in response: return None msg, *remainder = response["Messages"]