add method to get message count

2024-01-20 11:34:53 +01:00
parent 47cf77e049
commit f7ac820aad
2 changed files with 14 additions and 6 deletions
@@ -10,12 +10,17 @@ class CookiesPublisher:
    def __init__(self):
        self.connection = None
        self.channel = None
+        self.queue_method = None

    def set_up_connection(self):
        self.connection = pika.BlockingConnection(
            pika.ConnectionParameters(host=QUEUE_HOST, port=5672, credentials=credentials))
        self.channel = self.connection.channel()
-        # self.channel.queue_declare(queue=REQUEST_DATA_QUEUE, durable=True)
+        self.queue_method = self.channel.queue_declare(queue=REQUEST_DATA_QUEUE, durable=True)

    def publish_body(self, body: str):
        self.channel.basic_publish(exchange='', routing_key=REQUEST_DATA_QUEUE, body=body)
+
+    def message_count(self):
+        return self.channel.queue_declare(queue=REQUEST_DATA_QUEUE, durable=True).method.message_count
+        # return self.queue_method.method.message_count