spring boot任务调度线程池 spring boot设置线程数

转载

goody 2023-07-05 23:15:59

文章标签 spring boot任务调度线程池多线程 springboot 线程池 Async 文章分类 spring boot 后端开发

一.配置线程池参数

@EnableAsync
@Configuration
public class TaskExecutorConfig {
    @Bean
    public TaskExecutor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        // 设置核心线程数
        executor.setCorePoolSize(60);
        // 设置最大线程数
        executor.setMaxPoolSize(100);
        // 设置队列容量
        executor.setQueueCapacity(300);
        // 设置线程活跃时间（秒）
        executor.setKeepAliveSeconds(60);
        // 设置默认线程名称
        executor.setThreadNamePrefix("hiksion-");
        // 设置拒绝策略
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        // 等待所有任务结束后再关闭线程池
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.initialize();
        return executor;
    }
}

二.调用方法上添加@Async注解

@Async
    @Transactional
    public void storePic(File file, byte[] bytes, ResourceRecordDto resouceRecordDto) throws IOException {
        // 将数据插入数据库(唯一键索引)
        ResourceRecordVo vo = null;
        try {
            vo = resourceRecordService.insertRecord(resouceRecordDto);
        } catch (DuplicateKeyException e) {
            log.error("数据文件已经存在,文件名称:{}", resouceRecordDto.getFileName());
            throw e;
        }

        @Cleanup
        BufferedOutputStream stream = new BufferedOutputStream(new FileOutputStream(file));
        stream.write(bytes);
        // 数据处理打包好
        ResourceRecordVo finalVo = vo;
        TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronizationAdapter() {
            @Override
            public void afterCommit() {
                if (finalVo != null) {
                    ResourceRecordMessage msg = new ResourceRecordMessage();
                    msg.setResourceCompositeRecordId(finalVo.getResourceCompositeRecordId());
                    highDeviceSender.send(msg);
                }
            }
        });
    }