python实现生产者和消费者

1 基础说明

python版本：Python 3.6.8

rabbitmq版本: rabbitmq-server-3.6.12-1.el7.noarch.rpm

 

2 python代码示例

(1)生产者

import pika
import json
import datetime

"""
/usr/bin/pip3 install -Iv pika -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com;
/usr/bin/python3 ./rabbimq_pro_v1.py > ./num_pro.txt
"""

today_time = datetime.datetime.now()
credentials = pika.PlainCredentials('usermq', 'xxxxxxx')  #mq用户名和密码

#虚拟队列需要指定参数 virtual_host，如果是默认的可以不填
connection = pika.BlockingConnection(pika.ConnectionParameters(host = 'x.x.x.x' ,port = 5672 ,virtual_host = '/' ,credentials = credentials))
channel=connection.channel()

#声明exchange，由exchange指定消息在哪个队列传递，如不存在，则创建; durable = True 代表exchange持久化存储，False 非持久化存储
channel.exchange_declare(exchange = 'python-celnumber_v1', durable = True, exchange_type='direct')
for i in range(200000): #写入20w的消息
    message=json.dumps({"number":"num-%s"%i,"key00000":1234})

    #指定 routing_key, delivery_mode = 2 声明消息在队列中持久化，delivery_mod = 1 消息非持久化
    channel.basic_publish(exchange = 'python-celnumber_v1', routing_key = 'OrderId', body = message, properties=pika.BasicProperties(delivery_mode = 2))
    print(message)
connection.close()
print("耗时为 {0} 秒".format((datetime.datetime.now() - today_time).total_seconds()))

 

(2)消费者

import pika

"""
/usr/bin/pip3 install -Iv pika -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com;
/usr/bin/python3 ./rabbitmq_cus_v1.py >> ./rabbitmq_cus.txt
"""

credentials = pika.PlainCredentials('usermq', 'xxxxxx')
connection = pika.BlockingConnection(pika.ConnectionParameters(host = 'x.x.x.x', port = 5672, virtual_host = '/', credentials = credentials))
channel = connection.channel()
# 创建临时队列，队列名传空字符，consumer关闭后，队列自动删除
result = channel.queue_declare('python_cus_v1', exclusive=True)
# 声明exchange，由exchange指定消息在哪个队列传递，如不存在，则创建。durable = True 代表exchange持久化存储，False 非持久化存储
channel.exchange_declare(exchange = 'python-celnumber_v1', durable = True, exchange_type='direct')
# 绑定exchange和队列  exchange 使我们能够确切地指定消息应该到哪个队列去
channel.queue_bind(exchange = 'python-celnumber_v1', queue = result.method.queue, routing_key='OrderId')
# 定义一个回调函数来处理消息队列中的消息，这里是打印出来
def callback(ch, method, properties, body):
    ch.basic_ack(delivery_tag = method.delivery_tag)
    print(body.decode())


#channel.basic_qos(prefetch_count=1)
# 告诉rabbitmq，用callback来接受消息
channel.basic_consume(result.method.queue, callback, auto_ack = False)
# 设置成 False，在调用callback函数时，未收到确认标识，消息会重回队列。True，无论调用callback成功与否，消息都被消费掉
channel.start_consuming()

 

(3)操作说明

先运行消费者：/usr/bin/python3 ./rabbitmq_cus_v1.py >> ./rabbitmq_cus.txt

再运行生产者：/usr/bin/python3 ./rabbimq_pro_v1.py > ./num_pro.txt