需求
项目使用springboot，有一个场景需要先从本地查询所有数据，然后再调用第三方接口填充其他字段，每次调用第三方接口都需要本地数据的字段作为查询条件。厂商提供的接口只能单个查询，所以只能遍历从本地查询的数据然后挨个调用接口，这样响应时间实在慢且效率太低。
那就想办法优化呗，第一想法肯定是启用多线程，让每一条从本地数据库查询的记录可以并行调用第三方接口。
首先新建一个线程任务类来调用第三方接口并填充数据，用spring的@Async注解实现异步调用，伪代码如下：
-
@Component
-
public
class AsyncTask {
-
-
-
/**
-
* 异步调用第三方接口查询
-
*/
-
@Async
-
public void queryTask(Entity entity){
-
Data data = doPost();
-
entity.setData(data);
-
}
-
-
}
Service层循环调用
-
public
class Service {
-
-
@Autowired
-
private AsyncTask asyncTask;
-
@Autowired
-
private EntityMapper mapper;
-
-
public List<Entity> queryData(){
-
-
List<Entity> list = mapper.selectAll();
-
for(Entity entity:list){
-
asyncTask.queryTask(entity);
-
}
-
-
return list;
-
}
-
}
这样就实现了异步请求接口，效率上提升了很多。但是，由于异步调用的原因，数据还没填充完就会返回，这显然不是我们想要的效果。我们必须等待AsyncTask中的所有线程结束后，再返回当前调用线程任务的方法。于是就想到了JDK1.5版本后提供的计数器CountDownLatch（至于CountDownLatch的用法本文不阐述，只是第一次使用记录一下，有错误的地方还希望大家给建议）。
思路：
在Service层的方法中实例化CountDownLatch并且制定线程个数，线程个数就是从本地数据库查询的list的长度，并且传入线程任务中，每个线程执行完毕就调用countDown()方法。最后在Service层中调用await()方法。这样在线程计数为零之前，Service的线程就会一直等待。
Service:
-
public
class Service {
-
-
@Autowired
-
private AsyncTask asyncTask;
-
@Autowired
-
private EntityMapper mapper;
-
-
public List<Entity> queryData(){
-
-
List<Entity> list = mapper.selectAll();
-
CountDownLatch latch =
new CountDownLatch(list.size());
-
for(Entity entity:list){
-
asyncTask.queryTask(entity,latch);
-
}
-
latch.await();
-
return list;
-
}
-
}
AsyncTask:
-
@Component
-
public
class AsyncTask {
-
-
-
/**
-
* 异步调用第三方接口查询
-
*/
-
@Async
-
public void queryTask(Entity entity,CountDownLatch latch){
-
Data data = doPost();
-
entity.setData(data);
-
latch.countDown();
-
}
-
-
}
至此问题解决。