SpringBoot自定义线程池ThreadPoolTaskExecutor

配置自定义线程池

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;

@Configuration
@EnableAsync
public class ThreadPoolConfig {

    @Bean("taskExecutor") //线程池名称
    public Executor asyncServiceExecutor() {
        //获取当前主机cpu核数
        final int cpuCount = Runtime.getRuntime().availableProcessors();
        //springboot基于ThreadPoolExecutor实现的一个线程池执行类
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        //设置核心线程数量
        executor.setCorePoolSize(cpuCount);
        //设置最大线程数量
        executor.setMaxPoolSize(cpuCount * 2);
        //设置队列长度
        executor.setQueueCapacity(100);
        //设置线程活跃时间（秒）
        executor.setKeepAliveSeconds(60);
        //设置默认线程名称前缀
        executor.setThreadNamePrefix("myBlog--");
        //等待所有任务结束后再关闭线程池
        executor.setWaitForTasksToCompleteOnShutdown(true);
        //除此之外，还有许多可以配置的，比如拒绝策略等等
        //初始化
        executor.initialize();
        return executor;
    }
}

@Async异步执行方法，使用线程池

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.yyf.blogapi.mapper.ArticleMapper;
import com.yyf.blogapi.po.Article;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

@Slf4j
@Service
public class UpdateViewCountService {

    @Async("taskExecutor")
    public void updateViewCount(ArticleMapper articleMapper, Article article) {
        Article updateViewArticle = new Article();
        updateViewArticle.setViewCounts(article.getViewCounts() + 1);
        /*LambdaUpdateWrapper<Article> updateWrapper = new LambdaUpdateWrapper<>();
        updateWrapper.eq(Article::getId, article.getId());
        updateWrapper.eq(Article::getViewCounts,article.getViewCounts());
        articleMapper.update(updateViewArticle,updateWrapper);*/
        updateViewArticle.setId(article.getId());
        articleMapper.updateById(updateViewArticle);
    }
}