当前位置 知且行 其他 正文 下一篇:

RabbitMQ实战5.路由

继上篇 RabbitMQ实战4.发布与订阅

交换机的路由功能

注意!交换机为匿名交换机时,routing_key指的是队列名,这只是一个特例!这个参数的本意是路由键名!

本篇实现这样的功能:

  • 生产者投递不同类型的消息到交换机,消息类型用交换机的路由参数指定;
  • 消费者根据交换机不同的路由获取并处理消息

具体场景:生产者可以投递不同级别的日志,消费者根据自己的需求去捕获不同类型的日志。如error级别的日志才写入到文件,所有级别的日志都显示到终端。

路由

代码实现

新建 emit_log_direct.py

  1. import pika
  2. import sys
  3. connection = pika.BlockingConnection(pika.ConnectionParameters(
  4. host='localhost'))
  5. channel = connection.channel()
  6. channel.exchange_declare(exchange='direct_logs', # 声明交换机
  7. exchange_type='direct') # 交换机类型
  8. severity = sys.argv[1] if len(sys.argv) > 2 else 'info'
  9. message = ' '.join(sys.argv[2:]) or 'Hello World!'
  10. channel.basic_publish(exchange='direct_logs',
  11. routing_key=severity, # 声明路由
  12. body=message)
  13. print(" [x] Sent %r:%r" % (severity, message))
  14. connection.close()

新建 receive_logs_direct.py

  1. import pika
  2. import sys
  3. connection = pika.BlockingConnection(pika.ConnectionParameters(
  4. host='localhost'))
  5. channel = connection.channel()
  6. channel.exchange_declare(exchange='direct_logs',
  7. exchange_type='direct')
  8. result = channel.queue_declare(exclusive=True)
  9. queue_name = result.method.queue
  10. severities = sys.argv[1:]
  11. if not severities:
  12. sys.stderr.write("Usage: %s [info] [warning] [error]\n" % sys.argv[0])
  13. sys.exit(1)
  14. for severity in severities:
  15. # 将队列与交换机绑定,并指定交换机的路由, 一个队列可以绑定交换机的多个路由
  16. channel.queue_bind(exchange='direct_logs',
  17. queue=queue_name,
  18. routing_key=severity)
  19. print(' [*] Waiting for logs. To exit press CTRL+C')
  20. def callback(ch, method, properties, body):
  21. print(" [x] %r:%r" % (method.routing_key, body))
  22. channel.basic_consume(callback,
  23. queue=queue_name,
  24. no_ack=True)
  25. channel.start_consuming()

执行结果

开启消费者1

  1. python receive_logs_direct.py warning error > logs_from_rabbit.log # 将 warning 和 error 级别的日志写入文件

开启消费者2

  1. python receive_logs_direct.py info warning error # 将所有级别的日志输出到终端

开启生产者

  1. python emit_log_direct.py error "Run. Run. Or it will explode."

参数文档

转载必须注明出处:https://www.zhiqiexing.com/61.html

关于我

我希望能成为一个认真、有趣、创造更多价值的人
关注微信
微信扫一扫关注我

微信扫一扫关注我

返回顶部