您的位置: 首页  >  文章  >  Springboot学习笔记（一）-线程池的简化及使用

Springboot学习笔记（一）-线程池的简化及使用

分类: 文章 • 2024-05-21 14:04:28

" rel="nofollow,noindex" target="_blank">https://www.cnblogs.com/yw0219/p/8810956.html

工作中经常涉及异步任务，通常是使用多线程技术，比如线程池ThreadPoolExecutor，它的执行规则如下：

Springboot学习笔记（一）-线程池的简化及使用

在Springboot中对其进行了简化处理，只需要配置一个类型为java.util.concurrent.TaskExecutor或其子类的bean，并在配置类或直接在程序入口类上声明注解@EnableAsync。

调用也简单，在由Spring管理的对象的方法上标注注解@Async，显式调用即可生效。

一般使用Spring提供的ThreadPoolTaskExecutor类。

声明

@Configuration
@EnableAsync
public class BeanConfig {

<span class="hljs-meta">@Bean</span>
<span class="hljs-function"><span class="hljs-keyword">public</span> TaskExecutor <span class="hljs-title">taskExecutor</span><span class="hljs-params">()</span> </span>{
    ThreadPoolTaskExecutor executor = <span class="hljs-keyword">new</span> ThreadPoolTaskExecutor();
    <span class="hljs-comment">// 设置核心线程数</span>
    executor.setCorePoolSize(<span class="hljs-number">5</span>);
    <span class="hljs-comment">// 设置最大线程数</span>
    executor.setMaxPoolSize(<span class="hljs-number">10</span>);
    <span class="hljs-comment">// 设置队列容量</span>
    executor.setQueueCapacity(<span class="hljs-number">20</span>);
    <span class="hljs-comment">// 设置线程活跃时间（秒）</span>
    executor.setKeepAliveSeconds(<span class="hljs-number">60</span>);
    <span class="hljs-comment">// 设置默认线程名称</span>
    executor.setThreadNamePrefix(<span class="hljs-string">"hello-"</span>);
    <span class="hljs-comment">// 设置拒绝策略</span>
    executor.setRejectedExecutionHandler(<span class="hljs-keyword">new</span> ThreadPoolExecutor.CallerRunsPolicy());
    <span class="hljs-comment">// 等待所有任务结束后再关闭线程池</span>
    executor.setWaitForTasksToCompleteOnShutdown(<span class="hljs-keyword">true</span>);
    <span class="hljs-keyword">return</span> executor;
}

}

调用

@Component
public class Hello {
    @Async
    public void sayHello(String name) {
        LoggerFactory.getLogger(Hello.class).info(name + ":Hello World!");
    }
}

测试

Springboot学习笔记（一）-线程池的简化及使用
从打印日志中可以看出线程池已经正常工作了。

进阶

有时候我们不止希望异步执行任务，还希望任务执行完成后会有一个返回值，在java中提供了Future泛型接口，用来接收任务执行结果，springboot也提供了此类支持，使用实现了ListenableFuture接口的类如AsyncResult来作为返回值的载体。比如上例中，我们希望返回一个类型为String类型的值，可以将返回值改造为:

    @Async
    public ListenableFuture<String> sayHello(String name) {
        String res = name + ":Hello World!";
        LoggerFactory.getLogger(Hello.class).info(res);
        return new AsyncResult<>(res);
    }

调用返回值：

    @Autowired
    private Hello hello;

<span class="hljs-comment">// 阻塞调用</span>
<span class="hljs-selector-tag">hello</span><span class="hljs-selector-class">.sayHello</span>(<span class="hljs-string">"yan"</span>)<span class="hljs-selector-class">.get</span>();
<span class="hljs-comment">// 限时调用</span>
<span class="hljs-selector-tag">hello</span><span class="hljs-selector-class">.sayHello</span>(<span class="hljs-string">"yan"</span>)<span class="hljs-selector-class">.get</span>(<span class="hljs-number">1</span>, TimeUnit.SECONDS)</code></pre>

补充

实际上，@Async还有一个参数，通过Bean名称来指定调用的线程池-比如上例中设置的线程池参数不满足业务需求，可以另外定义合适的线程池，调用时指明使用这个线程池-缺省时springboot会优先使用名称为'taskExecutor'的线程池，如果没有找到，才会使用其他类型为TaskExecutor或其子类的线程池。

相关推荐