java
复制代码
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.baomidou.mybatisplus.extension.service.IService;
import com.eam.common.utils.spring.SpringUtils;
import lombok.SneakyThrows;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.Callable;
import java.util.concurrent.Future;
public class MdUtils {
/**
*
* @param clz 查询的isvc
* @param wrapper 查询的wrapper
* @param i 线程类数
* @param columns 需要查询的字段
*/
@SafeVarargs
public static <T>List<T> doPackageResCallables(Class<? extends IService<T>> clz, LambdaQueryWrapper<T> wrapper, Long i, SFunction<T, ?>... columns){
List<MdCallable<T>> mdCallables = packageCallables(clz, wrapper,i,columns);
return executeTask(mdCallables);
}
@SafeVarargs
public static <T>List<T> doPackageResCallables(Class<? extends IService<T>> clz,LambdaQueryWrapper<T> wrapper,SFunction<T, ?>... columns){
List<MdCallable<T>> mdCallables = packageCallables(clz, wrapper,columns);
return executeTask(mdCallables);
}
@SafeVarargs
public static <T>List<MdCallable<T>> packageCallables(Class<? extends IService<T>> clz,LambdaQueryWrapper<T> wrapper, SFunction<T, ?>... columns){
return packageCallables(clz,wrapper,200L,columns);
}
@SafeVarargs
public static <T>List<MdCallable<T>> packageCallables(Class<? extends IService<T>> clz,LambdaQueryWrapper<T> wrapper,Long i, SFunction<T, ?>... columns){
IService<T> bean = SpringUtils.getBean(clz);
long count = bean.count();
i = (Objects.isNull(i) || 0 == i) ? 10 : count>i?i:count;
Long pageSize = count / i;
List<MdCallable<T>> ts = new ArrayList<>();
for (Long l = 0L; l < i; l++) {
ts.add(new MdCallable<T>(pageSize*l,pageSize,clz,wrapper,columns));
}
return ts;
}
@SneakyThrows
@SuppressWarnings("ALL")
public static <T>List<T> executeTask(List<MdCallable<T>> c){
List<Future<T>> priceFutureList = new ArrayList<>();
ArrayList<T> res = new ArrayList<>();
c.forEach(tCallable -> priceFutureList.add(SpringUtils.getBean(ThreadPoolTaskExecutor.class).submit((Callable<T>) tCallable)));
for (Future<T> f : priceFutureList) {
res.addAll((List<T>) f.get());
}
return res;
}
//获取结果集
public static<T> List<T> getFotFutureList(Class<T> clz,List l){
for (Object o : l) {
if (o instanceof List && !((List<?>) o).isEmpty() && ((List<?>) o).get(0).getClass().equals(clz)){
return (List<T>) o;
}
}
return null;
}
//异步获取任务
@SafeVarargs
public static <U>List<U> doCompletableFuture(Supplier<U>... supplier){
List<CompletableFuture<U>> collect = new ArrayList<>(Arrays.asList(supplier)).stream().map(supplierSig -> CompletableFuture.supplyAsync(supplierSig, SpringUtils.getBean(ThreadPoolTaskExecutor.class))).collect(Collectors.toList());
CompletableFuture.allOf(collect.toArray(new CompletableFuture[0])).join();
return collect.stream().map(CompletableFuture::join).collect(Collectors.toList());
}
}