libretificacaotjcore 0.1.37__py3-none-any.whl → 0.1.40__py3-none-any.whl

libretificacaotjcore/services/rabbitmq_consumer.py

@@ -1,15 +1,16 @@
-import pika
+import asyncio
+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 +19,41 @@ 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,
+
+    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,
-            blocked_connection_timeout=1200,
         )
-        
-        self.connection = pika.BlockingConnection(parameters)
-        self.channel = self.connection.channel()
-        self.channel.queue_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()
-            
+        self.channel = await self.connection.channel()
+        await self.channel.set_qos(prefetch_count=1)
+        await self.channel.declare_queue(self.queue, durable=True)
+
+    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 def on_message(message):
+            async with message.process():
+                try:
+                    mensagem = json.loads(message.body.decode())
+                    await callback(mensagem)  # aqui sim passa o DTO
+                except Exception as e:
+                    print(f"❌ Erro ao processar mensagem: {e}")
+
+        await queue.consume(on_message)  # registra callback
+
+        print(f'[*] Aguardando mensagens na fila "{self.queue}". Para sair pressione CTRL+C')
+
+        # Mantém o consumer rodando
+        await asyncio.Future()
+
 
-    def close(self):
+    async def close(self):
         if self.connection:
-            self.connection.close()
+            await self.connection.close()

{libretificacaotjcore-0.1.37.dist-info → libretificacaotjcore-0.1.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libretificacaotjcore
-Version: 0.1.37
+Version: 0.1.40
 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.37.dist-info → libretificacaotjcore-0.1.40.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=zk3DDjWZMDHcj32pTeu4zQTabRbDtvreypmr_loNRFg,1762
+libretificacaotjcore/services/rabbitmq_consumer.py,sha256=yRS3sAkaMIIn8Y06icwkO1bpddlLC7BmNSHJVhUalzY,1818
 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.37.dist-info/METADATA,sha256=Z9mls_R_1NA7XPbeJPvPRzWWeeNCjJ8euXjZqsIKuDc,2564
-libretificacaotjcore-0.1.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libretificacaotjcore-0.1.37.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
-libretificacaotjcore-0.1.37.dist-info/RECORD,,
+libretificacaotjcore-0.1.40.dist-info/METADATA,sha256=3dB_2pwlsN-Xgb2zFhn3N__WBIxHPPmeMfXuupIk-l0,2564
+libretificacaotjcore-0.1.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libretificacaotjcore-0.1.40.dist-info/top_level.txt,sha256=J9vnz_X9OUnxC-eXHiAzlc9xIrWBwZ5bgnIDQIIFY4c,21
+libretificacaotjcore-0.1.40.dist-info/RECORD,,