summaryrefslogtreecommitdiff
path: root/examples/simple_task_queue/client.py
blob: 489aa2e6beafd18b743a59ccc31bab6ea6bb95e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from __future__ import with_statement

from kombu.common import maybe_declare
from kombu.pools import producers

from .queues import task_exchange

priority_to_routing_key = {"high": "hipri",
                           "mid": "midpri",
                           "low": "lopri"}

def send_as_task(connection, fun, args, kwargs, priority="mid"):
    payload = {"fun": fun, "args": args, "kwargs": kwargs}
    routing_key = priority_to_routing_key[priority]

    with producers[connection].acquire(block=True) as producer:
        maybe_declare(task_exchange, producer.channel)
        producer.publish(payload, serializer="pickle",
                                  compression="zlib",
                                  routing_key=routing_key)

if __name__ == "__main__":
    from kombu import BrokerConnection
    from .tasks import hello_task

    conection = BrokerConnection("amqp://guest:guest@localhost:5672//")
    send_as_task(connection, fun=hello_task, args=("Kombu", ),
                 priority="high")