libretificacaotjcore 0.1.36__py3-none-any.whl → 0.1.39__py3-none-any.whl

libretificacaotjcore/services/rabbitmq_consumer.py

@@ -1,15 +1,15 @@
-import pika
+import aio_pika
 import json
 
-
 class RabbitMQConsumer:
     def __init__(
         self,
+        *,
         host: str,
         queue: str,
         username: str,
         password: str,
-        vhost: str ="/",
+        vhost: str = "/",
     ):
         self.host = host
         self.queue = queue
@@ -18,39 +18,36 @@ class RabbitMQConsumer:
         self.vhost = vhost
         self.connection = None
         self.channel = None
-        self.connect()
 
-    def connect(self):
-        credentials = pika.PlainCredentials(self.username, self.password)
-        parameters = pika.ConnectionParameters(
-            host=self.host, 
-            credentials=credentials, 
-            virtual_host=self.vhost,
-            heartbeat=60,
-            blocked_connection_timeout=300,
+    async def connect(self):
+        self.connection = await aio_pika.connect_robust(
+            host=self.host,
+            login=self.username,
+            password=self.password,
+            virtualhost=self.vhost,
+            heartbeat=600,
         )
-        
-        self.connection = pika.BlockingConnection(parameters)
-        self.channel = self.connection.channel()
-        self.channel.queue_declare(queue=self.queue, durable=True)
+        self.channel = await self.connection.channel()
+        await self.channel.set_qos(prefetch_count=1)
+        await self.channel.declare_queue(self.queue, durable=True)
 
-    def start_consuming(self, callback):
-        def on_message(ch, method, properties, body):
-            mensagem = json.loads(body)
-            callback(mensagem)
-            ch.basic_ack(delivery_tag=method.delivery_tag)
-            self.close()
-            
+    async def start_consuming(self, *, callback):
         if not self.channel:
             raise RuntimeError("❌ Canal RabbitMQ não conectado. Chame connect() antes.")
 
-        self.channel.basic_qos(prefetch_count=1)
-        self.channel.basic_consume(queue=self.queue, on_message_callback=on_message)
-        print(
-            f'[*] Aguardando mensagens na fila "{self.queue}". Para sair pressione CTRL+C'
-        )
-        self.channel.start_consuming()
+        queue = await self.channel.get_queue(self.queue)
+
+        async with queue.iterator() as queue_iter:
+            async for message in queue_iter:
+                async with message.process():
+                    try:
+                        mensagem = json.loads(message.body.decode())
+                        await callback(mensagem)
+                    except Exception as e:
+                        print(f"Erro ao processar mensagem: {e}")\
+                    
+                    await self.close()
 
-    def close(self):
+    async def close(self):
         if self.connection:
-            self.connection.close()
+            await self.connection.close()

{libretificacaotjcore-0.1.36.dist-info → libretificacaotjcore-0.1.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libretificacaotjcore
-Version: 0.1.36
+Version: 0.1.39
 Summary: Biblioteca para centralizar conexao com filas no rabbit e banco de dados no mongodb para os servicos de retificacao da TJ
 Author-email: Jhonatan Azevedo <dev.azevedo@outlook.com>
 Project-URL: Homepage, https://github.com/seu-usuario/libretificacaotjcore

{libretificacaotjcore-0.1.36.dist-info → libretificacaotjcore-0.1.39.dist-info}/RECORD

@@ -13,11 +13,11 @@ libretificacaotjcore/enums/e_fase_retificacao.py,sha256=g7xVlIMKpo9QUDnOdwYzsx8o
 libretificacaotjcore/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libretificacaotjcore/services/crypto_pass_service.py,sha256=9D0vyjan6f_8AfNxGkLpGdvyMpojsJq_AAySpv_zKMc,740
 libretificacaotjcore/services/file_service.py,sha256=14CJokBbrsryQGmL0_unH2QKZpnteEAfxf5CPFdv6cE,2075
-libretificacaotjcore/services/rabbitmq_consumer.py,sha256=XPbOvFR3HySfA1IROLPrsb4AmeHeq1Unthxt44vaqik,1760
+libretificacaotjcore/services/rabbitmq_consumer.py,sha256=aw2NaPkuF1AewJh2dQetTsZVvFlimjw96o5r-jnu86A,1687
 libretificacaotjcore/services/rabbitmq_publisher.py,sha256=xsb9LmIuZizzq5lxnEEZKMK3U952KSQ2oqlTBZi2Lt8,1787
 libretificacaotjcore/services/request_servico_api.py,sha256=G7vnmvOfwrCGL-Jy_5tCKG-l5E00OCkJfkuoa5Y6sHo,613
 libretificacaotjcore/services/s3_service.py,sha256=HKR_jt2H3XdV1PCzo5R5bnhmoQ3I46Yn5IqAvVPhsjs,2946
-libretificacaotjcore-0.1.36.dist-info/METADATA,sha256=f3LhO06GgojKkclG48eJgtN9ME3FNXpwDJDrcCyLowY,2564
-libretificacaotjcore-0.1.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libretificacaotjcore-0.1.36.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
-libretificacaotjcore-0.1.36.dist-info/RECORD,,
+libretificacaotjcore-0.1.39.dist-info/METADATA,sha256=DkNnpyxrvUU7Co7tsQGd149VPHS9sv3M3C3UtOckiHE,2564
+libretificacaotjcore-0.1.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libretificacaotjcore-0.1.39.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
+libretificacaotjcore-0.1.39.dist-info/RECORD,,