package cn.iocoder.yudao.module.mes.framework.qrexport;

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

import java.util.concurrent.ThreadPoolExecutor;

/**
 * 二维码后台导出线程池
 * <p>
 * 单独建池而不是复用公共异步池：二维码生成是长时间、CPU 密集的任务，混进公共池会把其他异步业务一起堵住。
 * 池大小取并发闸门配置，队列留少量冗余以吸收「校验通过但线程尚未就绪」的瞬时窗口；真正超出的提交
 * 由 {@link MesProQrExportTaskRegistry} 的并发许可拦在更前面，不会堆到这里。
 */
@Configuration(proxyBeanMethods = false)
public class MesProQrExportExecutorConfig {

    /**
     * 线程名，便于线上定位导出线程
     */
    private static final String THREAD_NAME_PREFIX = "mes-qr-export-";

    @Resource
    private MesProQrExportProperties properties;

    @Bean(name = "mesQrExportExecutor")
    public ThreadPoolTaskExecutor mesQrExportExecutor() {
        int concurrency = Math.max(1, properties.getMaxConcurrentTasks());
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(concurrency);
        executor.setMaxPoolSize(concurrency);
        executor.setQueueCapacity(concurrency);
        executor.setThreadNamePrefix(THREAD_NAME_PREFIX);
        // 队列满即拒绝，由调用方把任务置为失败，避免任务无声地堆在队列里
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.AbortPolicy());
        executor.initialize();
        return executor;
    }

}