Переглянути джерело

Update AliyunRocketMQQueueBackend: add invisible_duration for consume

StrayWarrior 1 місяць тому
батько
коміт
08b46d1622
1 змінених файлів з 3 додано та 2 видалено
  1. 3 2
      message_queue_backend.py

+ 3 - 2
message_queue_backend.py

@@ -75,10 +75,11 @@ class AliyunRocketMQQueueBackend(MessageQueueBackend):
     def __del__(self):
         self.shutdown()
 
-    def consume(self) -> Optional[Message]:
+    def consume(self, invisible_duration=60) -> Optional[Message]:
         if not self.has_consumer:
             raise Exception("Consumer not initialized.")
-        messages = self.consumer.receive(1, 10)
+        # TODO(zhoutian): invisible_duration实际是不同消息类型不同的,有些消息预期的处理时间会更长
+        messages = self.consumer.receive(1, invisible_duration)
         if not messages:
             return None
         rmq_message = messages[0]