Update rabbitmq test

catch-all/05_infra_test/02_rabbitmq/06_rpc/rpc_client.py

@@ -0,0 +1,113 @@
+#!/usr/bin/env python
+import pika
+import uuid
+import sys
+import argparse
+
+
+class FibonacciRpcClient:
+    def __init__(self, host='localhost'):
+        # Establece la conexión con RabbitMQ
+        try:
+            self.connection = pika.BlockingConnection(
+                pika.ConnectionParameters(host=host)
+            )
+        except pika.exceptions.AMQPConnectionError as e:
+            print(f"[!] Error al conectar con RabbitMQ: {e}")
+            sys.exit(1)
+
+        # Crea un canal de comunicación
+        self.channel = self.connection.channel()
+
+        # Declara una cola exclusiva para recibir las respuestas del servidor RPC
+        result = self.channel.queue_declare(queue='', exclusive=True)
+        self.callback_queue = result.method.queue
+
+        # Configura el canal para consumir mensajes de la cola de respuestas
+        self.channel.basic_consume(
+            queue=self.callback_queue,
+            on_message_callback=self.on_response,
+            auto_ack=True
+        )
+
+        # Inicializa las variables para manejar la respuesta RPC
+        self.response = None
+        self.corr_id = None
+
+    def on_response(self, ch, method, props, body):
+        """Callback ejecutado al recibir una respuesta del servidor RPC."""
+        # Verifica si el ID de correlación coincide con el esperado
+        if self.corr_id == props.correlation_id:
+            self.response = body
+
+    def call(self, n):
+        """
+        Envía una solicitud RPC para calcular el número de Fibonacci de n.
+
+        :param n: Número entero no negativo para calcular su Fibonacci.
+        :return: Resultado del cálculo de Fibonacci.
+        :raises ValueError: Si n no es un número entero no negativo.
+        """
+        # Verifica que la entrada sea un número entero no negativo
+        if not isinstance(n, int) or n < 0:
+            raise ValueError("[!] El argumento debe ser un entero no negativo.")
+
+        # Resetea la respuesta y genera un nuevo ID de correlación único
+        self.response = None
+        self.corr_id = str(uuid.uuid4())
+
+        # Publica un mensaje en la cola 'rpc_queue' con las propiedades necesarias
+        self.channel.basic_publish(
+            exchange='',
+            routing_key='rpc_queue',
+            properties=pika.BasicProperties(
+                reply_to=self.callback_queue,  # Cola de retorno para recibir la respuesta
+                correlation_id=self.corr_id,   # ID único para correlacionar la respuesta
+            ),
+            body=str(n)
+        )
+
+        # Espera hasta que se reciba la respuesta del servidor
+        while self.response is None:
+            self.connection.process_data_events(time_limit=None)
+
+        # Devuelve la respuesta convertida a un entero
+        return int(self.response)
+
+    def close(self):
+        """Cierra la conexión con el servidor de RabbitMQ."""
+        self.connection.close()
+
+
+if __name__ == "__main__":
+    # Configura el analizador de argumentos para recibir el número de Fibonacci
+    parser = argparse.ArgumentParser(
+        description="Calcula números de Fibonacci mediante RPC.")
+    parser.add_argument(
+        "number",
+        type=int,
+        help="Número entero no negativo para calcular su Fibonacci."
+    )
+    args = parser.parse_args()
+
+    # Inicializa el cliente RPC de Fibonacci
+    fibonacci_rpc = FibonacciRpcClient()
+
+    # Número para el cual se desea calcular el Fibonacci
+    n = args.number
+
+    try:
+        print(f" [+] Solicitando fib({n})")
+        # Realiza la llamada RPC y obtiene el resultado
+        response = fibonacci_rpc.call(n)
+        print(f" [+] Resultado fib({n}) = {response}")
+
+    except ValueError as e:
+        print(f"[!] Error de valor: {e}")
+
+    except Exception as e:
+        print(f"[!] Error inesperado: {e}")
+
+    finally:
+        # Cierra la conexión del cliente RPC
+        fibonacci_rpc.close()

catch-all/05_infra_test/02_rabbitmq/06_rpc/rpc_server.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+import pika
+from functools import lru_cache
+import time
+
+# Establecemos la conexión a RabbitMQ
+
+
+def create_connection():
+
+    try:
+        connection = pika.BlockingConnection(
+            pika.ConnectionParameters(host='localhost')
+        )
+
+        return connection
+
+    except pika.exceptions.AMQPConnectionError as e:
+        print(f"[!] Error al conectar a RabbitMQ: {e}")
+
+        return None
+
+
+# Conexión a RabbitMQ
+connection = create_connection()
+
+if not connection:
+    exit(1)
+
+# Canal de comunicación con RabbitMQ
+channel = connection.channel()
+
+# Declaración de la cola 'rpc_queue'
+channel.queue_declare(queue='rpc_queue')
+
+# Implementación mejorada de Fibonacci con memoización
+
+
+@lru_cache(maxsize=None)
+def fib(n):
+
+    if n < 0:
+        raise ValueError("\n[!] El número no puede ser negativo.")
+
+    elif n == 0:
+        return 0
+
+    elif n == 1:
+        return 1
+
+    else:
+        return fib(n - 1) + fib(n - 2)
+
+# Callback para procesar solicitudes RPC
+
+
+def on_request(ch, method, props, body):
+
+    try:
+        n = int(body)
+        print(f" [+] Calculando fib({n})")
+
+        # Simula un tiempo de procesamiento de 2 segundos
+        time.sleep(2)
+
+        response = fib(n)
+        print(f" [+] Resultado: {response}")
+
+    except ValueError as e:
+        response = f"\n[!] Error: {str(e)}"
+
+    except Exception as e:
+        response = f"\n[!] Error inesperado: {str(e)}"
+
+    # Publicar la respuesta al cliente
+    ch.basic_publish(
+        exchange='',
+        routing_key=props.reply_to,
+        properties=pika.BasicProperties(correlation_id=props.correlation_id),
+        body=str(response)
+    )
+
+    # Confirmar la recepción del mensaje
+    ch.basic_ack(delivery_tag=method.delivery_tag)
+
+
+# Configuración de calidad de servicio y consumo de mensajes
+channel.basic_qos(prefetch_count=1)
+channel.basic_consume(queue='rpc_queue', on_message_callback=on_request)
+
+print("\n[i] Esperando solicitudes RPC")
+channel.start_consuming()