package com.glxp.api.task;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.glxp.api.common.res.BaseResponse;
import com.glxp.api.dao.basic.SysWorkplaceDocumentDao;
import com.glxp.api.dao.collect.IoCollectSetBustypeMapper;
import com.glxp.api.dao.schedule.ScheduledDao;
import com.glxp.api.entity.collect.IoCollectSet;
import com.glxp.api.entity.collect.IoCollectSetBustype;
import com.glxp.api.entity.system.ScheduledEntity;
import com.glxp.api.req.basic.WorkBindBusTypeRequest;
import com.glxp.api.req.collect.CollectOrderRequest;
import com.glxp.api.req.system.ScheduledRequest;
import com.glxp.api.res.basic.SysWorkplaceDocumentResponse;
import com.glxp.api.service.collect.IoCollectOriginService;
import com.glxp.api.service.collect.IoCollectSetService;
import com.glxp.api.util.MsDateUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;

@Slf4j
@Component
@EnableScheduling
public class AsyncIoCollectOrderDownloadTask implements SchedulingConfigurer {

    final Logger logger = LoggerFactory.getLogger(AsyncIoCollectOrderDownloadTask.class);

    @Resource
    private ScheduledDao scheduledDao;
    @Resource
    SysWorkplaceDocumentDao sysWorkplaceDocumentDao;
    @Resource
    IoCollectOriginService collectOriginService;
    @Resource
    IoCollectSetService collectSetService;
    @Resource
    IoCollectSetBustypeMapper ioCollectSetBustypeMapper;

    private static final String LOCK_KEY = "IO_COLLECT_ORDER_DOWNLOAD_LOCK";
    private static final long LOCK_TIMEOUT = 30 * 60 * 1000; // 30分钟超时时间


    // 使用原子布尔值作为简单的锁机制
    private final AtomicBoolean isRunning = new AtomicBoolean(false);


    @Override
    public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {
        scheduledTaskRegistrar.addTriggerTask(() -> processWithLock(),
                triggerContext -> {
                    ScheduledRequest scheduledRequest = new ScheduledRequest();
                    scheduledRequest.setCronName("ioCollectOrderDownloadTask");
                    ScheduledEntity scheduledEntity = scheduledDao.findScheduled(scheduledRequest);
                    String cron = scheduledEntity != null ? scheduledEntity.getCron() : "0 */10 * * * ?";
                    if (cron.isEmpty()) {
                        logger.error("cron is null");
                    }
                    return new CronTrigger(cron).nextExecutionTime(triggerContext);
                });
    }

    private void processWithLock() {
        // 如果已经在运行，则跳过本次执行
        if (!isRunning.compareAndSet(false, true)) {
            log.warn("上一次任务还在执行中，跳过本次执行");
            return;
        }
        try {
            log.info("开始执行订单下载任务");
            process();
        } finally {
            isRunning.set(false);
            log.info("订单下载任务执行完成");
        }
    }


    private void process() {
        //获取下载的起始时间
        IoCollectSet ioCollectSet = collectSetService.getSet();
        Date startDownloadTime = ioCollectSet.getStartDownloadTime();
        String startDownloadTimes = MsDateUtil.formatDateTime(startDownloadTime);
        Boolean autoDownload = ioCollectSet.getAutoDownload();

        if (StrUtil.isNotBlank(startDownloadTimes) && autoDownload) {

            //获取当前时间
            String paramValue = startDownloadTimes;
            String nowTime = MsDateUtil.getDateTime();
            CollectOrderRequest collectOrderRequest = new CollectOrderRequest();

            collectOrderRequest.setStartTime(paramValue);
            collectOrderRequest.setEndTime(nowTime);

            List<IoCollectSetBustype> sysWorkplaceDocumentResponses = ioCollectSetBustypeMapper.selectList(new LambdaQueryWrapper<IoCollectSetBustype>().eq(IoCollectSetBustype::getSetCode, "autoDownload"));
            if (CollUtil.isNotEmpty(sysWorkplaceDocumentResponses)) {
                for (IoCollectSetBustype item : sysWorkplaceDocumentResponses) {
                    String documentTypeCode = item.getBusType();
                    collectOrderRequest.setBusType(documentTypeCode);
                    try {
                        BaseResponse baseResponse = collectOriginService.downloadOrderV2(collectOrderRequest);
                        if (baseResponse.getCode() == 20000) {
                            ioCollectSet.setStartDownloadTime(MsDateUtil.parseDate(nowTime));
                        }
                    } catch (Exception e) {
                        log.error("自动下载原始单据出错:{}", ExceptionUtils.getStackTrace(e));
                    }
                }
            }
            collectSetService.updateById(ioCollectSet);
        }

    }

}