ASJPush/Controller/Cron/CronTaskController.py

# @Author    : Rocky
# @File      : CronTaskController.py
# @Time      : 2023/12/11 15:47
import threading

from django.views import View

from Object.RedisObject import RedisObject
from Object.ResponseObject import ResponseObject
from Model.models import EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, \
    EquipmentInfo4, EquipmentInfo5, EquipmentInfo6, EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10, \
    EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16, \
    EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20
from Service.EquipmentInfoService import EQUIPMENT_INFO_DICT, EQUIPMENT_INFO_KEY_LIST

# 每次写入的数据条数
size = 1000


class CronTaskView(View):

    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.GET, request, operation)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.POST, request, operation)

    def validation(self, request_dict, request, operation):
        response = ResponseObject()
        if operation == 'equipment_info':
            return self.equipment_info(response)
        else:
            return response.json(414)

    @classmethod
    def equipment_info(cls, response):
        # 异步保存数据
        push_thread = threading.Thread(target=cls.save_equipment_info)
        push_thread.start()
        return response.json(0)

    @staticmethod
    def save_equipment_info():
        redis_obj_pipe = RedisObject().pipe
        for equipment_info_key in EQUIPMENT_INFO_KEY_LIST:
            equipment_info_model = EQUIPMENT_INFO_DICT[equipment_info_key]

            # 读取缓存的前n条数据批量写入
            redis_r_obj = RedisObject(mode='r')
            equipment_info_redis_list = redis_r_obj.lrange(equipment_info_key, 0, size - 1)

            redis_obj_pipe.ltrim(equipment_info_key, size, -1)

            equipment_info_list = []
            for equipment_info in equipment_info_redis_list:
                equipment_info_data = eval(equipment_info)
                # 设备昵称存在表情,解码utf-8
                if equipment_info_data.get('device_nick_name') is not None:
                    equipment_info_data['device_nick_name'] = equipment_info_data['device_nick_name']. \
                        encode('UTF-8', 'ignore').decode('UTF-8')
                equipment_info_list.append(equipment_info_model(**equipment_info_data))
            equipment_info_model.objects.bulk_create(equipment_info_list)
        redis_obj_pipe.execute()