def test_DjangoBrokerConnection(self):
try:
from django.conf import settings
except ImportError:
sys.stderr.write("Django is not installed. \
Not testing django specific features.\n")
return
configured_or_configure(settings,
CARROT_BACKEND=CARROT_BACKEND,
BROKER_HOST=BROKER_HOST,
BROKER_PORT=BROKER_PORT,
BROKER_VHOST=BROKER_VHOST,
BROKER_USER=BROKER_USER,
BROKER_PASSWORD=BROKER_PASSWORD)
expected_values = {
"backend_cls": CARROT_BACKEND,
"hostname": BROKER_HOST,
"port": BROKER_PORT,
"virtual_host": BROKER_VHOST,
"userid": BROKER_USER,
"password": BROKER_PASSWORD
}
conn = DjangoBrokerConnection()
self.assertTrue(isinstance(conn, BrokerConnection))
for val_name, val_value in expected_values.items():
self.assertEquals(getattr(conn, val_name, None), val_value)
def test_amqp(self):
from carrot.connection import DjangoBrokerConnection
from carrot.messaging import Publisher, Consumer
connection = DjangoBrokerConnection()
publisher = Publisher(connection=connection,
exchange="collector",
exchange_type='topic',
routing_key="collector.driver",
serializer='json')
publisher.send("test")
publisher.close()
connection.close()
def send_requests(requests, **options):
logger = logging
if 'logger' in options.keys():
logger = options['logger']
"""Send a import request message to be picked up by workers."""
connection = DjangoBrokerConnection()
publisher = Publisher(connection=connection,
exchange="collector",
exchange_type='topic',
routing_key="collector.driver",
serializer='json')
for req in requests:
routing_key = req['driver_routing_key']
publisher.send(req, routing_key=routing_key)
logger.debug("Sent request with routing_key %s:%s" % (
routing_key,
req,
))
publisher.close()
connection.close()
def __init__(self, exchange, queue, routing_key=None, debug=None):
"""
Initialise signal, conection, exchange, queue and run consumer
@return: Django Signal like object.
@param exchange: name of exchange for this signal
@type exchange: string
@param queue: name queue of this signal
@type queue: string
@param routing_key: name of routing_key betwen exchange and queue of this signal
@type routing_key: string
@param debug: debug flag
@type debug: bool
Example:
>>> amqp_signal_1 = SignalAMQP(exchange="test1", queue="q1")
>>> amqp_signal_2 = SignalAMQP(exchange="test2", queue="q2")
>>> amqp_signal_1.queue_bind(['q2'])
>>> def amqp_handler(sender, **kwargs):
print "AMPQ handler:", sender, kwargs
>>> amqp_signal_2.connect(amqp_handler, sender=None)
>>> amqp_signal_1.send("Hello world!")
"""
super(SignalAMQP, self).__init__(providing_args=["message"])
self.exchange = exchange
self.queue = queue
self.routing_key = routing_key
self.debug = debug is None and settings.DEBUG or debug
self.conn = DjangoBrokerConnection()
self.publisher = Publisher(connection=self.conn, exchange=self.exchange, exchange_type="fanout",\
routing_key=self.routing_key)
self.consumer = Consumer(connection=self.conn, queue=self.queue, exchange_type="fanout",\
exchange=self.exchange, routing_key=self.routing_key)
self.consumer.register_callback(self.callback)
self.cl = self.listen()
def get_responses(logger=logging):
connection = DjangoBrokerConnection()
consumer = Consumer(connection=connection,
exchange="collector.response",
queue="responses",
routing_key="response")
for message in consumer.iterqueue():
responses = message.payload
for resp in responses:
logger.debug("resp=%s" % resp)
try:
tag = Tag.objects.get(name=resp['name'])
tag.save_with_history(resp['current_value'])
except Exception as ex:
logger.error(ex)
#print "Could have saved '%s' for tag '%s'" % (resp['current_value'], tag.id,)
message.ack()
consumer.close()
connection.close()
def get_backend(self):
return self.conn.get_backend_cls()(DjangoBrokerConnection())
def _get_carrot_object(klass, **kwargs):
"Helper function to create Publisher and Consumer objects."
return klass(connection=DjangoBrokerConnection(),
exchange=settings.EXCHANGE,
routing_key=settings.ROUTING_KEY,
**kwargs)
