RabbitMQ消息队列(六):使用主题进行消息分发

在上篇文章 RabbitMQ消息队列(五):Routing 消息路由 中,我们实现了一个简单的日志系统。Consumer可以监听不同severity的log。但是,这也是它之所以叫做简单日志系统的原因,因为是仅仅能够通过severity设定。不支持更多的标准。

        比如syslog unix的日志工具,它可以通过severity (info/warn/crit...) 和模块(auth/cron/kern...)。这可能更是我们想要的:我们可以仅仅需要cron模块的log。

        为了实现类似的功能,我们需要用到topic exchange。


1. Topic exchange

        对于Message的routing_key是有限制的,不能使任意的。格式是以点号“."分割的字符表。比如:"stock.usd.nyse", "nyse.vmw", "quick.orange.rabbit"。你可以放任意的key在routing_key中,当然最长不能超过255 bytes。

        对于routing_key,有两个特殊字符(在正则表达式里叫元字符):

  • * (星号) 代表任意 一个单词
  • # (hash) 0个或者多个单词

        请看下面一个例子:

     Producer发送消息时需要设置routing_key,routing_key包含三个单词和两个点号。第一个key是描述了celerity(灵巧,敏捷),第二个是colour(色彩),第三个是species(物种):"<celerity>.<colour>.<species>"。

     在这里我们创建了两个绑定: Q1 的binding key 是" *.orange.*"; Q2 是  " *.*.rabbit" 和 " lazy.#":
  • Q1 感兴趣所有orange颜色的动物
  • Q2 感兴趣所有的rabbits和所有的lazy的

     比如routing_key是 "quick.orange.rabbit"将会发送到Q1和Q2中。消息"lazy.orange.elephant" 也会发送到Q1和Q2。但是"quick.orange.fox" 会发送到Q1;"lazy.brown.fox"会发送到Q2。"lazy.pink.rabbit" 也会发送到Q2,但是尽管两个routing_key都匹配,它也只是发送一次。"quick.brown.fox" 会被丢弃。

     如果发送的单词不是3个呢? 答案要看情况,因为#是可以匹配0个或任意个单词。比如"orange" or "quick.orange.male.rabbit",它们会被丢弃。如果是lazy那么就会进入Q2。类似的还有 "lazy.orange.male.rabbit",尽管它包含四个单词。

Topic exchange和其他exchange

由于有"*" (star) and "#" (hash), Topic exchange 非常强大并且可以转化为其他的exchange:

    如果binding_key 是 "#" - 它会接收所有的Message,不管routing_key是什么,就像是fanout exchange。

    如果 "*" (star) and "#" (hash) 没有被使用,那么topic exchange就变成了direct exchange。

2. 代码实现

     现在我们要refine我们上篇的日志系统。routing keys 有两个部分: "<facility>.<severity>"。

The code for emit_log_topic.py:

[python] view plain copy
  1. #!/usr/bin/env python  
  2. import pika  
  3. import sys  
  4.   
  5. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  6.         host='localhost'))  
  7. channel = connection.channel()  
  8.   
  9. channel.exchange_declare(exchange='topic_logs',  
  10.                          type='topic')  
  11.   
  12. routing_key = sys.argv[1if len(sys.argv) > 1 else 'anonymous.info'  
  13. message = ' '.join(sys.argv[2:]) or 'Hello World!'  
  14. channel.basic_publish(exchange='topic_logs',  
  15.                       routing_key=routing_key,  
  16.                       body=message)  
  17. print " [x] Sent %r:%r" % (routing_key, message)  
  18. connection.close()  

The code for receive_logs_topic.py:
[python] view plain copy
  1. #!/usr/bin/env python  
  2. import pika  
  3. import sys  
  4.   
  5. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  6.         host='localhost'))  
  7. channel = connection.channel()  
  8.   
  9. channel.exchange_declare(exchange='topic_logs',  
  10.                          type='topic')  
  11.   
  12. result = channel.queue_declare(exclusive=True)  
  13. queue_name = result.method.queue  
  14.   
  15. binding_keys = sys.argv[1:]  
  16. if not binding_keys:  
  17.     print >> sys.stderr, "Usage: %s [binding_key]..." % (sys.argv[0],)  
  18.     sys.exit(1)  
  19.   
  20. for binding_key in binding_keys:  
  21.     channel.queue_bind(exchange='topic_logs',  
  22.                        queue=queue_name,  
  23.                        routing_key=binding_key)  
  24.   
  25. print ' [*] Waiting for logs. To exit press CTRL+C'  
  26.   
  27. def callback(ch, method, properties, body):  
  28.     print " [x] %r:%r" % (method.routing_key, body,)  
  29.   
  30. channel.basic_consume(callback,  
  31.                       queue=queue_name,  
  32.                       no_ack=True)  
  33.   
  34. channel.start_consuming()  

3. 运行和结果

接收所有的log:

[python] view plain copy
  1. python receive_logs_topic.py "#"  
接收所有kern facility的log:
[python] view plain copy
  1. python receive_logs_topic.py "kern.*"  
仅仅接收critical的log:
[python] view plain copy
  1. python receive_logs_topic.py "*.critical"  
可以创建多个绑定:
[python] view plain copy
  1. python receive_logs_topic.py "kern.*" "*.critical"  
Producer产生一个log:" kern.critical" type:
[python] view plain copy
  1. python emit_log_topic.py "kern.critical" "A critical kernel error"  

课后思考题:
  • Will "*" binding catch a message sent with an empty routing key?
  • Will "#.*" catch a message with a string ".." as a key? Will it catch a message with a single word key?
  • How different is "a.*.#" from "a.#"?


参考资料:

1. http://www.rabbitmq.com/tutorials/tutorial-five-python.html

你可能感兴趣的:(rabbitmq)