Boto3：有没有办法优雅地中断 SQS 长轮询接收消息请求？

Question

我已经成功创建了一个使用长轮询从 SQS 队列中提取消息的方法，如下所示：

def dequeue_message(self, callback):
    result = self.queue.receive_messages(MaxNumberOfMessages=1)
    if len(result) != 0:
        body = result[0].body
        try:
            callback(body)
            result.delete()
        except Exception as e:
            print("message not dequeued because an error occurred"
                  "when running callback: " + str(e))

但我无法找到一种方法来停止轮询而不终止正在运行的 Python 进程（或者，显然，等待超时）。我能做什么？

Answer 1

您可以简单地使用标志来执行此操作。假设一个按钮应该中断轮询过程。按下按钮后立即更新标志。然后，当轮询返回一组消息时，您检查标志，并忽略处理消息。不用担心，消息仍会在队列中。见：link

Amazon SQS 在收到消息后不会自动删除它 为您，以防您未成功收到消息（对于 例如，消费者可能会失败或失去连接）。删除一个 消息，您必须发送一个单独的请求，确认您 不再需要该消息，因为您已成功收到并且 处理它。

示例代码：

# This is the flag to check for interruptions
is_interrupted = False

# This function will set the flag if an interruption is occurred
def interrupt_polling():
    is_interrupted = True


def dequeue_message(self, callback):
    result = self.queue.receive_messages(MaxNumberOfMessages=1)

    # this is the check that will bypass the Polling process
    # Handle this logic as required
    if is_interrupted:
        # assuming that this needs to be reset to False
        is_interrupted = False
        return

    if len(result) != 0:
        body = result[0].body
        try:
            callback(body)
            result.delete()
        except Exception as e:
            print("message not dequeued because an error occurred"
                  "when running callback: " + str(e))

希望这会有所帮助。