最近开发过程中遇到了请求接口耗时较长且需多次请求,故使用多线程处理。
多线程采用的是实现Callable接口+线程池ThreadPoolExecutor,实现Callable接口主要是处理的方法有返回值,若无返回值,可以采用继承Thread或者实现Runnable接口。下面是部分处理过后的代码,避免忘记故记录下来。
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.*;
/**
* @Author xxx
* @data 2020/12/xx 11:39
*/
public class RunnableSVImpl {
// 创建一个线程池
ThreadPoolExecutor pool = new ThreadPoolExecutor(3,5,8, TimeUnit.SECONDS, new LinkedBlockingDeque<Runnable>(10));//ThreadPoolExecutor为自定义线程池,读者可以更换成ExecutorService提供的几种线程池
List<Future> listFuture = new ArrayList();
// 创建多个有返回值的任务
for(int i = 0;i<5;i++) {
String index = "xxx" , specJsonStr = "xxx";
Callable callable = new MyCallable(index,specJsonStr);//创建一个callable线程
// 执行任务并获取Future对象
Future future = pool.submit(callable);
listFuture.add(future);//将执行的多线程加入list,方便后面拿结果
}
// 关闭线程池
pool.shutdown();
// 获取所有并发任务的运行结果
List resultList = new ArrayList();
for (Future f : listFuture) {
List<Map> esSpecIdList = (List<Map>)f.get();//拿到多线程执行的结果
resultList.add(esSpecIdList);
}
//内部类 多线程实现Callable接口
class MyCallable implements Callable {
private String index;
private String specStr;
MyCallable(String index, String specStr) {
this.index = index;
this.specStr = specStr;
}
@Override
public Object call() throws Exception { //重写call方法
List<Map> esSpecIdList = new ArrayList<>();
try {
Date date1 = new Date();
esSpecIdList = esUtilController.queryEsData(index, specJsonStr);//需要多线程操作的方法,读者自行修改为自己的即可
Date date2 = new Date();
logger.info("多线程查询耗时:"+String.valueOf(date2.getTime()-date1.getTime()));
if (esSpecIdList != null && !esSpecIdList.isEmpty()) {
return esSpecIdList;
}
} catch (Exception e) {
logger.error("全量查询ES数据失败:", e);
}
return esSpecIdList;
}
}
}