pyrabbit, experimentando con RabbitMQ

Llevo ya un tiempo probando RabbitMQ, un sistema de mensajería para aplicaciones basado en colas, muy potente y escalable, desarrollado con Erlang. Para saber un poco más de este lenguaje de programación recomiendo leer Learn You Some Erlang for Great Good! Cuando conoces un poco este lenguaje comprendes por qué lo han usado para programar RabbitMQ, ya que da la impresión de que Erlang fue diseñado expresamente para crear este tipo de sistemas.

Hay una cosa que no me ha gustado demasiado de RabbitMQ, y es que es el propio sistema de mensajería el que decide para quién es un determinado mensaje. Me explico. Supongamos que tenemos una cola que recibe mensajes con trabajos a realizar y tenemos dos procesos suscritos a la cola para realizar esos trabajos. Para ello, los procesos se suscriben a la cola y es RabbitMQ el que manda los mensajes al proceso que crea conveniente. Si el proceso que recibe el mensaje no puede atenderlo en ese momento, por las razones que sean, debe cancelar la recepción para que RabbitMQ pueda reencolar el mensaje y otro proceso pueda recibirlo.

Para este tipo de cosas yo prefiero usar peticiones tipo "long polling" con posibilidad de especificar un timeout, de forma que sea el propio proceso el que pida mensajes al servidor. Por supuesto, igual puede ocurrir que el proceso pierda la conexión, no pueda atender al mensaje, etc, con lo que también deberíamos cancelar la recepción para el reencolado del mensaje, pero ya resulta más extraño, y el código de los procesos resulta más legible.

He desarrollado una pequeña librería en Python, pyrabbit, que por debajo usa Pika, una implementación del protocolo AMPQ que nos sirve para conectar con RabbitMQ y comunicarnos con las colas. Con pyrabbit es posible realizar estas operaciones de forma muy sencilla.

import pyrabbit
connection = pyrabbit.Connection('localhost')

# Enviar mensaje a mi_cola y leerlo después con un timeout de diez segundos
connection.mi_cola.send("hola")
m = connection.mi_cola.receive(timeout=10)
print m.body   # Esto imprime 'hola'

# Confirmamos que el mensaje ha sido procesado y no es necesario reencolarlo
m.ack()

# Tambien es posible enviar mensajes y esperar confirmación con un timeout
# Supongamos que tenemos un proceso que calcula sumas
r = connection.cola_de_sumas.send("[2,3]", wait_response=True, timeout=5)
print r.body   # Si el proceso ficticio que suma números funciona, obtendremos "5"

# Cerramos
connection.close()

Con pyrabbit puedo comunicar programas de forma sencilla y el código es muy legible, pero no me gusta la forma en que tengo implementada la recepción de mensajes, ya que estoy solicitando mensajes al servidor hasta que me devuelve alguno o hasta que salta el timeout. Para no penalizar demasiado, incluyo un retardo incremental con el que acabo pidiendo mensajes cada segundo si la espera es demasiado larga, pero aún así no me acaba. Si alguien sabe una forma de pedir mensajes a RabbitMQ de tipo similar a long polling, le estaría muy agradecido.