pika python rabbitmq 优先级队列、查看队列剩余数目

谁践踏了优雅 2023-02-11 03:22 5阅读 0赞

话不多说，直接上代码。  
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()  # 启动消费

#### 查看队列剩余数目 ####

credentials = pika.PlainCredentials(username, pwd)
    connection = pika.BlockingConnection(pika.ConnectionParameters(ip_addr_in, port_num, '/', credentials, heartbeat=0))
    connection.process_data_events()
    channel = connection.channel()
    max_priority = 10
    declear_queue_result = channel.queue_declare(queue=queue_name, arguments={ "x-max-priority": max_priority}, durable=True)
    
    queue_message_count = declear_queue_result.method.message_count
    
    print(queue_message_count)
    if queue_message_count == 0:
        # insert daily
        # do something. 
        # 这个代码略掉了一些 config 变量，自己定义一下