查询接口优化(线程池异步调用)

---

前言

在项目中遇到一个接口响应时间比较慢，查看代码发现是有几个sql查询加在一起比较慢，而且同时业务只需要这几个sql的结果进行拼接处理。
因此考虑使用线程池异步运行多个sql的方式来提高接口响应时间。

以下为相关代码实现

一、配置线程池

application.yml

thread-pool:config:core-size: 8max-size: 16queue-capacity: 64keep-alive-seconds: 180

参数配置类

@Component
@ConfigurationProperties(prefix = "thread-pool.config")
@Data
public class TestThreadPoolConfig {private Integer coreSize;private Integer maxSize;private Integer queueCapacity;private Integer keepAliveSeconds;
}

线程池配置类

import org.springframework.beans.factory.annotation.Autowired;
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.RejectedExecutionHandler;
@Configuration
@EnableAsync
public class ThreadPoolConfig {@Autowiredprivate TestThreadPoolConfig testThreadPoolConfig;/*** @return*/@Bean(name = "testExecutor")public ThreadPoolTaskExecutor testThreadPoolExecutor() {return getAsyncTaskExecutor("test-Executor-", testThreadPoolConfig.getCoreSize(),testThreadPoolConfig.getMaxSize(), testThreadPoolConfig.getQueueCapacity(),testThreadPoolConfig.getKeepAliveSeconds(), null);}/*** 统一异步线程池** @param threadNamePrefix* @param corePoolSize* @param maxPoolSize* @param queueCapacity* @param keepAliveSeconds* @param rejectedExecutionHandler 拒接策略 没有填null* @return*/private ThreadPoolTaskExecutor getAsyncTaskExecutor(String threadNamePrefix, int corePoolSize, int maxPoolSize, int queueCapacity, int keepAliveSeconds, RejectedExecutionHandler rejectedExecutionHandler) {ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();taskExecutor.setCorePoolSize(corePoolSize);taskExecutor.setMaxPoolSize(maxPoolSize);taskExecutor.setQueueCapacity(queueCapacity);taskExecutor.setThreadPriority(Thread.MAX_PRIORITY);//线程优先级taskExecutor.setDaemon(false);//是否为守护线程taskExecutor.setKeepAliveSeconds(keepAliveSeconds);taskExecutor.setThreadNamePrefix(threadNamePrefix);taskExecutor.setRejectedExecutionHandler(rejectedExecutionHandler);taskExecutor.initialize();//线程池初始化return taskExecutor;}
}

二、服务类

    @Autowiredprivate ThreadPoolTaskExecutor testExecutor;@Overridepublic List test() {CompletableFuture<List> future = CompletableFuture.supplyAsync(this::test01, testExecutor).exceptionally(this::handleException);CompletableFuture<List> future2 = CompletableFuture.supplyAsync(this::test02, testExecutor).exceptionally(this::handleException);CompletableFuture<Void> combinedFuture = CompletableFuture.allOf(future, future2);List vos = new ArrayList<>();combinedFuture.thenAccept(v -> {vos.addAll(future.join());vos.addAll(future2.join());System.out.println(vos);});combinedFuture.join();return vos;}private List handleException(Throwable t) {throw new RuntimeException("An exception occurred: " + t.getMessage(), t);}public List test01(){System.out.println("执行test01");try {Thread.sleep(2000);} catch (InterruptedException e) {throw new RuntimeException(e);}List list = new ArrayList();list.add("111");list.add("222");System.out.println("执行test01结束");return list;}public List test02(){System.out.println("执行test02");try {Thread.sleep(1000);} catch (InterruptedException e) {throw new RuntimeException(e);}List list = new ArrayList();list.add("333");list.add("444");System.out.println("执行test02结束");return list;}

---