package com.vincent.rsf.schedule.schedules;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.framework.common.Cools;
import com.vincent.rsf.schedule.system.entity.ExportTask;
import com.vincent.rsf.schedule.system.service.ExportTaskService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.io.File;
import java.util.Date;
import java.util.List;
import java.util.Objects;

@Slf4j
@Component
public class ExportTaskCleanupSchedules {

    private final ExportTaskService exportTaskService;

    @Value("${rsf.export-task.cleanup-batch-size:200}")
    private int cleanupBatchSize;

    public ExportTaskCleanupSchedules(ExportTaskService exportTaskService) {
        this.exportTaskService = exportTaskService;
    }

    @Scheduled(
            initialDelayString = "${rsf.export-task.cleanup-initial-delay-ms:60000}",
            fixedDelayString = "${rsf.export-task.cleanup-fixed-delay-ms:3600000}"
    )
    public void cleanupExpiredExportTasks() {
        List<ExportTask> expiredTasks = exportTaskService.list(
                new LambdaQueryWrapper<ExportTask>()
                        .eq(ExportTask::getDeleted, 0)
                        .isNotNull(ExportTask::getExpireTime)
                        .lt(ExportTask::getExpireTime, new Date())
                        .last("LIMIT " + Math.max(cleanupBatchSize, 1))
        );
        if (Cools.isEmpty(expiredTasks)) {
            return;
        }

        expiredTasks.forEach(this::deleteTaskFile);
        exportTaskService.removeByIds(expiredTasks.stream().map(ExportTask::getId).toList());
    }

    private void deleteTaskFile(ExportTask task) {
        if (task == null || Cools.isEmpty(task.getFilePath())) {
            return;
        }
        try {
            File file = new File(task.getFilePath());
            if (file.exists() && file.isFile() && !file.delete()) {
                log.warn("导出任务文件删除失败, taskId={}, filePath={}", task.getId(), task.getFilePath());
            }
        } catch (Exception error) {
            log.warn(
                    "导出任务文件删除异常, taskId={}, filePath={}",
                    task.getId(),
                    Objects.toString(task.getFilePath(), ""),
                    error
            );
        }
    }
}