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

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>
 * 单独建池而不是复用公共异步池：大批量落库是长时间、数据库密集的任务，
 * 混进公共池会把其他异步业务一起堵住。
 */
@Configuration(proxyBeanMethods = false)
public class MesProBagCodeGenerateExecutorConfig {

    private static final String THREAD_NAME_PREFIX = "mes-bag-code-generate-";

    @Resource
    private MesProBagCodeGenerateProperties properties;

    @Bean(name = "mesBagCodeGenerateExecutor")
    public ThreadPoolTaskExecutor mesBagCodeGenerateExecutor() {
        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;
    }

}