需求
项目使用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();
}
}
至此问题解决。