项目需要,数据库串行查询导致接口调用超时,使用多线程和反射改为异步查询。
方式一:建议通用的查询参数配置的线程,再用反射调用
利用反射机制创建查询数据库的线程类(此处为无参数的反射):
public class SelectTaskNoArgsCallable<T> implements Callable {
private Object object;
private Object[] args;
private String methodName;
public SelectTaskNoArgsCallable(Object object, String methodName, Object[] args) {
this.object = object;
this.args = args;
this.methodName = methodName;
}
@Override
public T call() throws Exception {
Method method = object.getClass().getMethod(methodName); //此处应用反射机制,根据实际方法参数设置的
return (T) method.invoke(object, args);
}
}
Service层:
import java.util.*;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
public class TestService{
@Autiwired
private CardingAssetsDataResultMapper cardingAssetsDataResultMapper;
public List<Object> multipleQueriesTest throws ExecutionException, InterruptedException{
// 三个线程的线程池,核心线程=最大线程,没有临时线程,阻塞队列无界
ExecutorService executorService = Executors.newFixedThreadPool(3);
// 开启线程执行
// 注意,此处Future对象接收线程执行结果不会阻塞,只有future.get()时候才会阻塞(直到线程执行完返回结果)
Future future1 = executorService.submit(new SelectTaskNoArgsCallable<>(cardingAssetsDataResultMapper, "getSensitiveTypeIdAndColumnCountInfo", new Object[]{}));
Future future2 = executorService.submit(new SelectTaskNoArgsCallable<>(cardingAssetsDataResultMapper, "getSensitiveTypeIdAndTableCountInfo", new Object[]{}));
Future future3 = executorService.submit(new SelectTaskNoArgsCallable<>(cardingAssetsDataResultMapper, "getSensitiveTypeIdAndSchemaCountInfo", new Object[]{}));
Object result1=future1.get();//此处可强转
Object result2=future2.get();
Object result3=future3.get();
List<Object> results = new ArrayList<>();
results.add(result1);
results.add(result2);
results.add(result3);
executorService.shutdown();//关闭线程池
return results;
}
}
mapper方法自定义,此处不做记录。
方式二:java.util.concurrent.CompletableFuture使用,更简单;
CompletableFuture<Integer> integerCompletableFuture = CompletableFuture.supplyAsync(
() -> nmgModelConfigurationMapper.getListCount(NmgEventSourceVO.builder().excludeModelRuleIds(excludeModelRuleIdLists.stream().map(Integer::valueOf).collect(Collectors.toList())).tagStatus(0).build())
);
try {
Integer integer = integerCompletableFuture.get();
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
}