package cn.iocoder.yudao.module.iot.job.ota;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.quartz.core.handler.JobHandler;
import cn.iocoder.yudao.framework.tenant.core.job.TenantJob;
import cn.iocoder.yudao.module.iot.core.enums.device.IotDeviceStateEnum;
import cn.iocoder.yudao.module.iot.dal.dataobject.device.IotDeviceDO;
import cn.iocoder.yudao.module.iot.dal.dataobject.ota.IotOtaFirmwareDO;
import cn.iocoder.yudao.module.iot.dal.dataobject.ota.IotOtaTaskRecordDO;
import cn.iocoder.yudao.module.iot.enums.ota.IotOtaTaskRecordStatusEnum;
import cn.iocoder.yudao.module.iot.service.device.IotDeviceService;
import cn.iocoder.yudao.module.iot.service.ota.IotOtaFirmwareService;
import cn.iocoder.yudao.module.iot.service.ota.IotOtaTaskRecordService;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * IoT OTA 升级推送 Job：查询待推送的 OTA 升级记录，并推送给设备
 *
 * @author 芋道源码
 */
@Component
@Slf4j
public class IotOtaUpgradeJob implements JobHandler {

    @Resource
    private IotOtaTaskRecordService otaTaskRecordService;
    @Resource
    private IotOtaFirmwareService otaFirmwareService;
    @Resource
    private IotDeviceService deviceService;

    @Override
    @TenantJob
    public String execute(String param) throws Exception {
        // 1. 查询待推送的 OTA 升级记录
        List<IotOtaTaskRecordDO> records = otaTaskRecordService.getOtaRecordListByStatus(
                IotOtaTaskRecordStatusEnum.PENDING.getStatus());
        if (CollUtil.isEmpty(records)) {
            return null;
        }

        // TODO 芋艿：可以优化成批量获取 原因是：1. N+1 问题；2. offline 的设备无需查询
        // 2. 遍历推送记录
        int successCount = 0;
        int failureCount = 0;
        Map<Long, IotOtaFirmwareDO> otaFirmwares = new HashMap<>();
        for (IotOtaTaskRecordDO record : records) {
            try {
                // 2.1 设备如果不在线，直接跳过
                IotDeviceDO device = deviceService.getDeviceFromCache(record.getDeviceId());
                // TODO 芋艿：【优化】当前逻辑跳过了离线的设备，但未充分利用 MQTT 的离线消息能力。
                // 1. MQTT 协议本身支持持久化会话（Clean Session=false）和 QoS > 0 的消息，允许 broker 为离线设备缓存消息。
                // 2. 对于 OTA 升级这类非实时性强的任务，即使设备当前离线，也应该可以推送升级指令。设备在下次上线时即可收到。
                // 3. 后续可以考虑：增加一个“允许离线推送”的选项。如果开启，即使设备状态为 OFFLINE，也应尝试推送消息，依赖 MQTT Broker 的能力进行离线缓存。
                if (device == null || IotDeviceStateEnum.isNotOnline(device.getState())) {
                    continue;
                }
                // 2.2 获取 OTA 固件信息
                IotOtaFirmwareDO fireware = otaFirmwares.get(record.getFirmwareId());
                if (fireware == null) {
                    fireware = otaFirmwareService.getOtaFirmware(record.getFirmwareId());
                    otaFirmwares.put(record.getFirmwareId(), fireware);
                }
                // 2.3 推送 OTA 升级任务
                boolean result = otaTaskRecordService.pushOtaTaskRecord(record, fireware, device);
                if (result) {
                    successCount++;
                } else {
                    failureCount++;
                }
            } catch (Exception e) {
                failureCount++;
                log.error("[execute][推送 OTA 升级任务({})发生异常]", record.getId(), e);
            }
        }
        return StrUtil.format("升级任务推送成功：{} 条，送失败：{} 条", successCount, failureCount);
    }

}