Python 消息队列rabbitmq使用之有选择的发布消息/接收消息-直连交换机direct

1、发布端代码

# new_direct_p.py
import pika
import sys

# 直连交换机(direct)的使用
# 用来指定某些消费者只收到,它该消费的消息
connection = pika.BlockingConnection(pika.ConnectionParameters(host='localhost'))
channel = connection.channel()

# 指定使用 直接交换机
channel.exchange_declare(exchange='direct_logs',exchange_type='direct')

# 获取用户输入,如果没有或长度小于1 那么就 发送info 给对应的消费者,这是只有收取info的消费者,才会收到消息
severity = sys.argv[1] if len(sys.argv) > 1 else 'info'

# 获取输入的消息 运行脚本时的格式为:python new_direact.py info This is info messge. 
message = ' '.join(sys.argv[2:]) or 'Hello World!'
# 发布消息
channel.basic_publish(exchange='direct_logs',
                      routing_key=severity,
                      body=message)
print(" [x] Sent %r:%r" % (severity, message))
connection.close()

2、消费端代码

# new_direct_c.py
import pika
import sys

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

channel.exchange_declare(exchange='direct_logs',exchange_type='direct')

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

# 获取运行脚本时传入的 参数
severities = sys.argv[1:]
if not severities:
    # 如果没有参数,进行提示 并退出脚本
    sys.stderr.write("Usage: %s [info] [warning] [error]\n" % sys.argv[0])
    sys.exit(1)

for severity in severities:
    # 循环传入的参数,来绑定routing_key
    # 例如:获取到的是info 那么就绑定接收info类型的消息,获取到的是error 那么就绑定接收error类型的错误
    channel.queue_bind(exchange='direct_logs',
                       queue=queue_name,
                       routing_key=severity)

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

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

channel.basic_consume(callback,
                      queue=queue_name,
                      no_ack=True)

# 开始消费
channel.start_consuming()

3、如何运行
消费端:

python new_direct_c.py info
python new_direct_c.py warning
python new_direct_c.py error warning

注意:在多个终端中运行接收端代码,可以更好的理解
发布端:

python new_direct_p.py info 这是一条info消息 
python new_direct_p.py warning 这是一条警告消息
python new_direct_p.py error 这是一条错误消息消息
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

haeasringnar

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值