纵有疾风起
人生不言弃

Django自定义指令+mq消息队列的使用

import pikaimport jsonimport loggingimport base64from rest_framework.exceptions import ParseErrorfrom django.core.management.base import BaseCommandfrom device.access_device import parse_access_device_imagelogger = logging.getLogger('server.default')class Command(BaseCommand):    help = 'Get the value from the message queue and write to queue'    def handle(self, *args, **options):        obj = RabbitQueue()        queue_name = '_image'        obj.pop_queue(queue_name=queue_name)class RabbitQueue:    def __init__(self):        self.username = 'admin'        self.password = 'admin'        self.host = '127.0.0.1'        self.port = 5672        self.channel = None    def connect(self):        credit = pika.PlainCredentials(username=self.username, password=self.password)        self.channel = pika.BlockingConnection(pika.ConnectionParameters(host=self.host, port=self.port, credentials=credit)).channel()    @staticmethod    def callback(channel, method, properties, body):        receive = json.loads(body.decode())        try:            device_name = receive['device']['detail']            device_address = receive['device']['scene']            image = base64.b64decode(receive['image'])            username = receive['user']['name']            phone = receive['user']['username']            gender = '' if receive['user']['gender'] == 1 else ''            parse_access_device_image(device_name=device_name, device_address=device_address, image=image, username=username, phone=phone, gender=gender)        except ParseError:            logger.error(receive)            raise ParseError('ParseError...')    def pop_queue(self, queue_name, timeout=0):        """从消息队列获取数据"""        self.connect()        channel = self.channel        channel.queue_declare(queue=queue_name, durable=True)        channel.basic_consume(on_message_callback=self.callback, queue=queue_name, auto_ack=False)        channel.start_consuming()

 

文章转载于:https://www.cnblogs.com/52-qq/p/11596751.html

原著是一个有趣的人,若有侵权,请通知删除

未经允许不得转载:起风网 » Django自定义指令+mq消息队列的使用
分享到: 生成海报

评论 抢沙发

评论前必须登录!

立即登录