pika python rabbitmq 優先級隊列

話不多說，直接上代碼。
python3.6 pika 實現rabbitmq 優先級隊列

Send

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()  # 關閉消息隊列服務的連接

consume

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()  # 啓動消費