17 lines
526 B
Python
17 lines
526 B
Python
from pyinfra.queue.queue_manager.queue_manager import QueueManager
|
|
|
|
|
|
class Consumer:
|
|
def __init__(self, visitor, queue_manager: QueueManager):
|
|
self.queue_manager = queue_manager
|
|
self.visitor = visitor
|
|
|
|
def consume_and_publish(self, n=None):
|
|
self.queue_manager.consume_and_publish(self.visitor, n=n)
|
|
|
|
def basic_consume_and_publish(self):
|
|
self.queue_manager.basic_consume_and_publish(self.visitor)
|
|
|
|
def consume(self, **kwargs):
|
|
return self.queue_manager.consume(**kwargs)
|