通过 http 方式判断 RabbitMQ 队列是否存在

最新推荐文章于 2024-07-20 15:03:20 发布

原创最新推荐文章于 2024-07-20 15:03:20 发布 · 7.7k 阅读

1 ·

CC 4.0 BY-SA版权

coding 专栏收录该内容

13 篇文章

订阅专栏

# summary: 通过 http 方式获取 RabbitMQ 队列状态

# import os, sys, time
import requests
import json
from nlp_property import NLP_FRAMEWORK_IP
from urllib import parse


class RabbitMQTool:
    def __init__(self, host, vhost, queue):
        self.host = host
        self.vhost = vhost
        self.queue = queue
        self.user = 'admin'
        self.passwd = 'admin@0'

    def get_queue_status(self):

        vhost = parse.quote_plus(self.vhost)
        queue = parse.quote_plus(self.queue)

        url = 'http://%s:15672/api/queues/%s/%s' % (self.host, vhost, queue)

        # url = url.encode(encoding='utf-8')

        # url = 'http://%s:15672/api/queues' % self.host
        print(url)
        # r = requests.get(url, auth=(self.user, self.passwd))
        r = requests.get(url, auth=(self.user, self.passwd))
        print(r)
        if r.status_code != 200:
            return False
        else:
            return True

        # dic = json.loads(r.text)
        # print(dic)
        #
        # return dic['messages_ready'], dic['messages_unacknowledged'], dic['messages']


if __name__ == '__main__':
    mqTool = RabbitMQTool(host=NLP_FRAMEWORK_IP,
                          vhost="/",
                          queue="nlp.classify.cnn.banktest.request.#")

    status = mqTool.get_queue_status()
    print(status)