RabbitMQ 使用(一)

RabbitMQ中的使用

这篇文章将会介绍关于RabbbitMQ的使用,并且使用的是kombo(客户端的Python实现)来实现;

安装

如果使用的是mac安装的话,可以先安装到指定的位置,接着配置命令访问路径:

  1. cd ~
  2. vi .bash_profile,输入下面两行

    RABBIT_HOME=/usr/local/Cellar/rabbitmq/3.6.9_1
    PATH=$PATH:$RABBIT_HOME/sbin
  3. esc,:wq保存并退出即可

启动和停止

开始:sudo rabbitmq-server start结束:sudo rabbitmq-server stop

Producer 和 Consumer

首先我们需要知道Producer和Consumer的初始化和其对应的publish和consumer方法。

Producer

class kombu.Producer(channel, exchange=None, routing_key=None,
serializer=None, auto_declare=None,
compression=None, on_return=None)
# 发布消息
.publish(body, routing_key=None, delivery_mode=None, mandatory=False,
immediate=False, priority=0, content_type=None, content_encoding=None,
serializer=None, headers=None, compression=None, exchange=None,
retry=False, retry_policy=None, declare=None, expiration=None, **properties)

Consumer

class kombu.Consumer(channel, queues=None, no_ack=None,
auto_declare=None, callbacks=None, on_decode_error=None,
on_message=None, accept=None, prefetch_count=None, tag_prefix=None)
# 消费
.consume(no_ack=None)

Hello world

当收到消息的时候,除非你已经对这个message进行了相关的操作,否则像是某个消费者的通道关闭等特殊情况下,RabbitMQ不会丢失掉这个信息,如果存在其它的消费者,则丢给其它消费者,没有就扔回队列中;当然你也可以通过no_ack=True来关闭消息确认机制。

from kombu import Exchange, Queue, Connection, Consumer, Producer
task_queue = Queue('tasks', exchange=Exchange('tasks', type='direct'), routing_key='tasks')
# 生产者
with Connection('amqp://guest@localhost:5672//') as conn:
with conn.channel() as channel:
producer = Producer(channel)
producer.publish({'hello': 'world'},
retry=True,
exchange=task_queue.exchange,
routing_key=task_queue.routing_key,
declare=[task_queue])
def get_message(body, message):
print(body)
# message.ack()
# 消费者
with Connection('amqp://guest@localhost:5672//') as conn:
with conn.channel() as channel:
consumer = Consumer(channel, queues=task_queue, callbacks=[get_message,], prefetch_count=10)
consumer.consume(no_ack=True)

生产者和消费者相互对应,这样一个简易的消息队列就可以使用了。

任务队列

我们将创建一个工作队列,专门用来处理分配耗时的任务。原理就是将任务封装成一个消息,由客户端发送到消息队列中,而后台运行的工作进程负责弹出任务并且分配给消费者来执行任务。这种方案在一些IO密集型的情况下很有用,比如在短时间内HTTP请求窗口中无法处理复杂的任务。

  1. 我们先创建相关的exchange和queue,queues.py文件如下:

    from kombu import Exchange, Queue
    task_exchange = Exchange('tasks', type='direct')
    task_queues = [Queue('high', exchange=task_exchange, routing_key='high'),
    Queue('middle', exchange=task_exchange, routing_key='middle'),
    Queue('low', exchange=task_exchange, routing_key='low')]
  2. 接下来再创建消费者,worker.py文件如下:

    from kombu.mixins import ConsumerMixin
    from queues import task_queues
    # 消费者
    class Worker(ConsumerMixin):
    def __init__(self, connection):
    self.connection = connection
    def get_consumers(self, Consumer, channel):
    consumer = Consumer(queues=task_queues, callbacks=[self.process_task], accept=['text/plain', 'json', 'pickle'])
    consumer.qos(prefetch_count=10) # 最多一下子获取10个任务
    return [consumer]
    def process_task(self, body, message):
    fun = body['fun']; args = body['args']; kwargs = body['kwargs']
    try:
    fun(*args, **kwargs)
    except Exception as exc:
    print(exc)
    message.requeue()
    else:
    message.ack()
    if __name__ == '__main__':
    from kombu import Connection
    with Connection('amqp://guest@localhost:5672//') as conn:
    try:
    worker = Worker(conn)
    worker.run()
    except KeyboardInterrupt:
    print('bye bye')
  3. 创建需要传递给消费者执行的任务,tasks.py如下:

    def hello_task(who='world'):
    import time
    print('wait one second')
    time.sleep(1)
    print('Hello {}'.format(who))
  4. 最后,创建生产者,client.py如下:

    from kombu.pools import producers
    from queues import task_exchange
    routing_keys = {
    'high': 'high',
    'middle': 'middle',
    'low': 'low'
    }
    # 将消息序列化后发送到队列中
    def send_as_task(connection, fun, key='middle', args=(), kwargs={}):
    payload = {'fun': fun, 'args': args, 'kwargs': kwargs}
    routing_key = routing_keys[key]
    with producers[connection].acquire(block=True) as producer:
    producer.publish(payload, serializer='pickle', exchange=task_exchange,
    routing_key=routing_key, declare=[task_exchange])
    if __name__ == '__main__':
    from kombu import Connection
    from tasks import hello_task
    with Connection('amqp://guest@localhost:5672//') as conn:
    send_as_task(conn, fun=hello_task, args=('wang',))

上面的代码主要实现的是,将hello_task这个任务经过pickle序列化以后发送到指定的middle消息队列中,接着消费者(可以开多个进程)从中取出消息后再执行任务。

上一篇:【转】TCP协议的无消息边界问题


下一篇:【Python自动化运维之路Day9】Socket