package com.gs.xky.task;


import cn.hutool.core.collection.ListUtil;
import com.gs.xky.config.DingTalkParam;
import com.gs.xky.config.DingTalkResponse;
import com.gs.xky.dto.EmployeeInfo;
import com.gs.xky.entity.MesInvItemArn;
import com.gs.xky.service.*;
import lombok.RequiredArgsConstructor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;

@Component
@RequiredArgsConstructor
public class ScheduledTasks {
    private static final Logger log = LoggerFactory.getLogger(ScheduledTasks.class);

    private final XkyService xkyService;
    private final ApiService apiService;
    private final MesStaffService staffService;
    private final DeliveryNoticeService deliveryNoticeService;
    private final MesInvItemArnService invItemArnService;
    private final Executor taskExecutor; // 注入通用线程池

    // 用于标记各任务是否正在执行
    private final AtomicBoolean isDeviceDataRunning = new AtomicBoolean(false);
    private final AtomicBoolean isCompensateRunning = new AtomicBoolean(false);
    private final AtomicBoolean isDingTalkRunning = new AtomicBoolean(false);

    /**
     * 每五分钟执行一次
     * 获取设备最近的一条记录
     */
    @Scheduled(cron = "0 0/5 * * * ?")
    public void getDeviceRealTimeData() {
        // 如果任务已经在运行，则跳过本次执行
        if (!isDeviceDataRunning.compareAndSet(false, true)) {
            log.info("【getDeviceRealTimeData】上一次任务还在执行中，跳过本次执行");
            return;
        }

        log.info("【getDeviceRealTimeData】开始获取设备实时数据");

        // 使用异步执行，避免阻塞调度线程
        CompletableFuture.runAsync(() -> {
            try {
                xkyService.GetSaveDetail();
                log.info("【getDeviceRealTimeData】获取设备实时数据成功");
            } catch (IOException e) {
                log.error("【getDeviceRealTimeData】获取设备实时数据异常: {}", e.getMessage(), e);
            } finally {
                isDeviceDataRunning.set(false);
            }
        }, taskExecutor);
    }

    /**
     * 定时执行补偿逻辑
     */
    @Scheduled(cron = "10 3,8,13,18,23,28,33,38,43,48,53,58 * * * ?")
    public void compensateMethod() {
        // 如果任务已经在运行，则跳过本次执行
        if (!isCompensateRunning.compareAndSet(false, true)) {
            log.info("【compensateMethod】上一次任务还在执行中，跳过本次执行");
            return;
        }

        log.info("【compensateMethod】开始执行补偿逻辑");

        // 使用异步执行，避免阻塞调度线程
        CompletableFuture.runAsync(() -> {
            try {
                // 补偿逻辑
                List<MesInvItemArn> itemArnMinus = invItemArnService.getItemArnMinus();
                deliveryNoticeService.processMesInvItemArnStatusAsync(itemArnMinus);
                log.info("【compensateMethod】补偿逻辑执行成功");
            } catch (Exception e) {
                log.error("【compensateMethod】补偿逻辑执行异常: {}", e.getMessage(), e);
            } finally {
                isCompensateRunning.set(false);
            }
        }, taskExecutor);
    }

    /**
     * 定时获取钉钉数据
     */
    @Scheduled(cron = "0 0/53 * * * ?")
    public void getDinTalkData() {
        // 如果任务已经在运行，则跳过本次执行
        if (!isDingTalkRunning.compareAndSet(false, true)) {
            log.info("【getDinTalkData】上一次任务还在执行中，跳过本次执行");
            return;
        }

        log.info("【getDinTalkData】开始获取钉钉数据");

        // 使用异步执行，避免阻塞调度线程
        CompletableFuture.runAsync(() -> {
            try {
                DingTalkParam dingTalkParam = new DingTalkParam(1);
                DingTalkResponse<EmployeeInfo> employeeInfoDingTalkResponse =
                        apiService.sendListRequest(dingTalkParam, EmployeeInfo.class,
                                "http://192.168.1.64/eHR/eHRExternalService/Service.ashx");

                List<EmployeeInfo> collect = employeeInfoDingTalkResponse.getData().stream()
                        .filter(s -> "造梦者（浙江）科技有限公司".equals(s.getCUnitName()))
                        .collect(Collectors.toList());

                log.info("【getDinTalkData】获取到{}条员工数据", collect.size());

                List<List<EmployeeInfo>> partition = ListUtil.partition(collect, 100);
                partition.forEach(staffService::UpdateStaff);

                log.info("【getDinTalkData】钉钉数据处理完成");
            } catch (IOException e) {
                log.error("【getDinTalkData】获取钉钉数据异常: {}", e.getMessage(), e);
            } finally {
                isDingTalkRunning.set(false);
            }
        }, taskExecutor);
    }
}