49RabbitMQ 消息队列基础入门--发布与订阅

发布与订阅

在本篇教程中,我们要做的跟之前完全不一样 —— 分发一个消息给多个消费者(consumers)。这种模式被称为“发布/订阅”。
为了描述这种模式,我们将会构建一个简单的日志系统。它包括两个程序——第一个程序负责发送日志消息,第二个程序负责获取消息并输出内容。
在我们的这个日志系统中,所有正在运行的接收方程序都会接受消息。我们用其中一个接收者(receiver)把日志写入硬盘中,另外一个接受者(receiver)把日志打印到屏幕上。
最终,日志消息被广播给所有的接受者(receivers)。
参考代码

cd /home/shiyanlou/

# 下载
wget https://labfile.oss.aliyuncs.com/courses/630/Code.zip

# 解压
unzip Code.zip

下载并解压后,本小节的参考代码在 /home/shiyanlou/Code/4/ 中,你编写的代码需要存放在 /home/shiyanlou/Code/ 中。

交换机

让我们简单的概括一下之前的教程:
发布者(producer):发布消息的应用程序
队列(queue):用于消息存储的缓冲
消费者(consumer):接收消息的应用程序

RabbitMQ 消息模型的核心理念是:发布者(producer)不会直接发送任何消息给队列。事实上,发布者(producer)甚至不知道消息是否已经被投递到队列。
发布者(producer)只需把消息发送给一个交换机(exchange)。交换机非常简单,它一边从发布者接收消息,一边把消息消息推送到队列。交换机必须知道如何处理它接收的消息,是应该推送到指定的队列还是多个队列,或者是直接忽略消息。这些规则是通过交换机类型(exchange type)来定义的。


image.png

有几个可供选择的交换机类型:直连交换机(direct),主题交换机(topic),头交换机(headers)和扇形交换机(fanout)。
我们在这里主要说明最后一个 -- 扇形交换机。先创建一个 fanout 类型的交换机,命名为 logs :

channel.exchange_declare(exchange='logs',
                         type='fanout')

完整的代码为:

#!/usr/bin/env python3
import pika

connection = pika.BlockingConnection(
    pika.ConnectionParameters(host='localhost'))
channel = connection.channel()

channel.exchange_declare(exchange='logs', exchange_type='fanout')

connection.close()

这段代码在之后的 emit_log.py 和 receive.py 中都要使用。等下我们要列出交换机,因此可以保存为 create_exchange.py ,并使用 python3 create_exchange.py 执行它。
扇形交换机(fanout)很简单,你可能从名字上就能猜测出来,它把消息发送给所有的队列,这正是我们日志系统所需要的。
交换机列表
rabbitmqctl 能够列出服务器上所有的交换机:

sudo service rabbitmq-server start

sudo rabbitmqctl list_exchanges
image.png

由于这里 rabbitmq-server 已经启动,所以没有执行 sudo service rabbitmq-server start。

这个列表中有一些叫做 amq.* 的交换机。这些都是默认创建的,不过这时候你还不需要它们。
可以看到我们之前执行了 python3 create_exchange.py,已经创建了类型为 fanout 的 logs 交换机。
匿名的交换机
前面的教程中我们对交换机一无所知,但仍然能够发送消息到队列中。因为我们使用了命名为空字符串("")默认的交换机。
之前我们是这么发送消息的:

channel.basic_publish(exchange='',routing_key='hello',body=message)

exchange 参数就是交换机的名字。空字符串代表默认或者匿名交换机:消息将会根据指定的 routing_key 分发到指定的队列。
现在,我们就可以发送消息到一个具体名字的交换机了:

channel.basic_publish( exchange='logs', routing_key='hello',body=message)

临时队列

你还记得我们使用的队列名吗:hello 和 task_queue。
给一个队列命名是很重要的。我们需要把工作者(workers)指向正确的队列。如果你打算在发布者(producers)和消费者(consumers)之间共享队列的话,给队列命名是十分重要的。
但是这并不适合我们的日志系统。我们打算接收所有的日志消息,而不仅仅是一小部分。我们关心的是最近的消息而不是久的。为了解决这个问题,我们需要做两件事情。
第一步,当我们连接上 RabbitMQ 的时候,我们需要一个全新的,空的队列。我们可以手动创建一个随机的队列名,或者让服务器为我们选择一个随机的队列名(推荐)。我们只需要在调用 queue_declare 方法的时候,不提供 queue 参数就可以了:

result = channel.queue_declare()

这时候我们可以通过 result.method.queue 获得已经生成的随机队列名。它可能是这样子的:amq.gen-U0srCoW8TsaXjNh73pnVAw==。
第二步,当与消费者(consumer)断开连接的时候,这个队列应当被立即删除。 exclusive 标识符即可达到此目的。

绑定(Bindings)

image.png

我们已经创建了一个扇形交换机( fanout )和一个队列。现在我们需要告诉交换机如何发送消息给我们的队列。
交换机和队列之间的联系我们称之为绑定( binding )。

channel.queue_bind(exchange='logs',queue=result.method.queue)

现在,logs 交换机将会把消息加到我们的队列中。
绑定列表
我们可以使用以下命令绑定所有现存的绑定。

sudo rabbitmqctl list_bindings
image.png

如果你使用之前保存的环境,可以看到 hello、task_queue 队列与匿名交换机的绑定。

代码整合

image.png

发布日志消息的程序看起来和之前的没有太大区别。最重要的改变就是我们把消息发送给 logs 交换机而不是匿名交换机。在发送的时候我们需要提供 routing_key 参数,但是它的值会被扇形交换机(fanout exchange)忽略。以下是 emit_log.py 脚本,编写源代码文件 /home/shiyanlou/Code/emit_log.py:

#!/usr/bin/env python3
import pika
import sys

connection = pika.BlockingConnection(
    pika.ConnectionParameters(host='localhost'))
channel = connection.channel()

channel.exchange_declare(exchange='logs', exchange_type='fanout')

message = ' '.join(sys.argv[1:]) or "info: Hello World!"
channel.basic_publish(exchange='logs', routing_key='', body=message)
print(" [x] Sent %r" % message)
connection.close()

正如你看到的那样,在连接成功以后,我们声明了一个交换机,这一个很重要的,因为不允许发布消息到不存在的交换机。
如果没有绑定队列到交换器,消息将会丢失。但这个没有所谓,如果没有消费者监听,那么消息就会被忽略。receive_logs.py 的代码,编写源代码文件 /home/shiyanlou/Code/receive_logs.py:

#!/usr/bin/env python3
import pika

connection = pika.BlockingConnection(
    pika.ConnectionParameters(host='localhost'))
channel = connection.channel()

channel.exchange_declare(exchange='logs', exchange_type='fanout')

result = channel.queue_declare(queue='', exclusive=True)
queue_name = result.method.queue

channel.queue_bind(exchange='logs', queue=queue_name)

print(' [*] Waiting for logs. To exit press CTRL+C')


def callback(ch, method, properties, body):
    print(" [x] %r" % body)


channel.basic_consume(
    queue=queue_name, on_message_callback=callback, auto_ack=True)

channel.start_consuming()

这样我们就完成了。
首先在终端中输入下面的命令:

# 确保服务已开启
sudo service rabbitmq-server start

打开三个终端,进入代码的目录,分别输入以下三个命令:

cd /home/shiyanlou/Code/
# 创建日志文件
touch logs_from_rabbit.log

# 把日志保存到 log 文件里
python3 receive_logs.py >> logs_from_rabbit.log
# 在屏幕上查看日志
python3 receive_logs.py
# 发送日志
python3 emit_log.py
image.png

打印到屏幕上:


image.png

重新打开一个终端,查看保存到文件里的内容:

cd /home/shiyanlou/Code/

tail -5 logs_from_rabbit.log
image.png

运行 sudo rabbitmqctl list_bindings 你可确认已经创建的队列绑定。你可以看到运行中的两个 receive_logs.py 程序:


image.png

你可能感兴趣的:(49RabbitMQ 消息队列基础入门--发布与订阅)