话不多说,直接上代码。
python3.6 pika 实现rabbitmq 优先级队列
import pika
import sys
import time
# 远程rabbitmq服务的配置信息
username = '用户名' # 指定远程rabbitmq的用户名密码
pwd = '密码'
ip_addr = 'ip'
port_num = 5672
# 消息队列服务的连接和队列的创建
credentials = pika.PlainCredentials(username, pwd)
connection = pika.BlockingConnection(pika.ConnectionParameters(ip_addr, port_num, '/', credentials, heartbeat=0))
channel = connection.channel()
max_priority = 10
channel.queue_declare(queue='ws_interface_priority', arguments={"x-max-priority": max_priority}, durable=True)
for x in range(0,10000):
channel.basic_publish(
exchange='',
routing_key='ws_interface_priority', # 写明将消息发送给队列balance
body=pub_dict, # 要发送的消息
properties=pika.BasicProperties(priority=1, delivery_mode=2, ) #priority=1,代表消息优先级,数字越大级别越高,可插队 设置消息持久化(持久化第二步),将要发送的消息的属性标记为2,表示该消息要持久化
) # 向消息队列发送一条消息
print(" [%s] Sent 'message!'")
connection.close() # 关闭消息队列服务的连接
import pika
import json
import time
import os
# 远程rabbitmq服务的配置信息
username = '用户名' # 指定远程rabbitmq的用户名密码
pwd = '密码'
ip_addr = 'ip'
port_num = 5672
credentials = pika.PlainCredentials(username, pwd)
connection = pika.BlockingConnection(pika.ConnectionParameters(ip_addr, port_num, '/', credentials, heartbeat=0))
connection.process_data_events()
channel = connection.channel()
max_priority = 10
channel.queue_declare(queue='ws_interface_priority', arguments={"x-max-priority": max_priority}, durable=True)
# 消费成功的回调函数
def callback(ch, method, properties, body):
body = json.loads(body)
id = (body["id"])
。。。
print(" [%s] Received %r" % (time.time(), body))
# 具体逻辑
# 当工作者完成任务后,会反馈给rabbitmq
channel.basic_ack(delivery_tag=method.delivery_tag)
# 开始依次消费balance队列中的消息
channel.basic_qos(prefetch_count=1)
channel.basic_consume(on_message_callback=callback, queue='ws_interface_priority', auto_ack=False)
print(' [*] Waiting for messages. To exit press CTRL+C')
channel.start_consuming() # 启动消费