package org.jeecg.modules.business.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * @author lxy
 */
@Configuration
@EnableAsync  // 開啟 Spring 異步支持
public class AsyncConfig {

    /**
     * 生成講稿專用異步線程池。
     * AI 調(diào)用屬于 IO 密集型，核心線程數(shù)不需要很大。
     * 使用 CallerRunsPolicy 作為拒絕策略，保證任務(wù)不丟失。
     */
    @Bean("speechExecutor")
    public Executor speechExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(2);       // 核心線程數(shù)
        executor.setMaxPoolSize(5);        // 最大線程數(shù)
        executor.setQueueCapacity(20);     // 隊列容量
        executor.setThreadNamePrefix("speech-async-");
        executor.setKeepAliveSeconds(60);
        // 隊列滿且線程達(dá)到上限時，由調(diào)用方線程執(zhí)行，避免任務(wù)丟失
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.initialize();
        return executor;
    }
}

/**
 * 異步生成講稿。方法立即返回，由后臺線程池（speechExecutor）執(zhí)行耗時的 AI 調(diào)用。
 * 生成成功后自動將 cdGenStatus 更新為"講稿已生成"（2），
 * 生成失敗則回滾為"暫存"（0），無需前端二次調(diào)用更新狀態(tài)接口。
 *
 * @param speechDto 生成講稿所需參數(shù)，包含 courseDetailId、classId 等
 */
@Async("speechExecutor")  // 指定使用 speechExecutor 線程池
@Override
public void generateSpeech(SpeechDto speechDto) {
    try {
        // --- 耗時操作：調(diào)用 AI 大模型 ---
        String genWord = cozeUtil.pptCreate(cozeDto);

        // 其他業(yè)務(wù)邏輯...

        // 生成成功 → 后端直接更新狀態(tài)為"已生成"
        this.updateCdGenStatus("2", speechDto.getCourseDetailId(), speechDto.getClassId());

    } catch (Exception e) {
        // 生成失敗 → 回滾狀態(tài)為"暫存"，前端可重試
        this.updateCdGenStatus("0", speechDto.getCourseDetailId(), speechDto.getClassId());
        throw new RuntimeException(e);
    }
}

@PostMapping(value = "/generateSpeech")
public Result<?> generateSpeech(@RequestBody SpeechDto speechDto) {
    bizCourseDetailService.generateSpeech(speechDto);  // 立即返回
    return Result.OK();
}

// ? 錯誤：在同一個 Service 內(nèi)部調(diào)用，@Async 不生效
public void someMethod() {
    this.generateSpeech(dto);  // 直接調(diào)用，不走代理，不會異步
}

// ? 正確：從另一個 Spring Bean（如 Controller）注入后調(diào)用
@Autowired
private IBizCourseDetailService bizCourseDetailService;

public void someMethod() {
    bizCourseDetailService.generateSpeech(dto);  // 走代理，異步生效
}

// 調(diào)用生成講稿接口后，定時輪詢狀態(tài)
async function generateSpeech(params) {
  await api.generateSpeech(params);  // 立即返回

  // 輪詢，每 3 秒檢查一次
  const timer = setInterval(async () => {
    const res = await api.queryCourseDetail({ id: params.courseDetailId, classId: params.classId });
    const status = res.result.cdGenStatus;
    if (status === '2') {
      clearInterval(timer);
      // 生成成功，刷新頁面
    } else if (status === '0') {
      clearInterval(timer);
      // 生成失敗，提示用戶重試
    }
  }, 3000);
}