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

复制代码
import pika
import json
import logging
import base64

from rest_framework.exceptions import ParseError

from django.core.management.base import BaseCommand

from device.access_device import parse_access_device_image

logger = 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()
复制代码

 

posted @   一石数字欠我15w!!!  阅读(1618)  评论(0编辑  收藏  举报
编辑推荐:
· 一个奇形怪状的面试题:Bean中的CHM要不要加volatile?
· [.NET]调用本地 Deepseek 模型
· 一个费力不讨好的项目,让我损失了近一半的绩效!
· .NET Core 托管堆内存泄露/CPU异常的常见思路
· PostgreSQL 和 SQL Server 在统计信息维护中的关键差异
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 如何使用 Uni-app 实现视频聊天(源码,支持安卓、iOS)
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
点击右上角即可分享
微信分享提示