封装的多线程查询工具,需要依赖mybatis-plus开启线程池

文章目录


一、MdUtils

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());
    }
}

二、MdCallable

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.AllArgsConstructor;

import java.util.List;
import java.util.concurrent.Callable;

@AllArgsConstructor
public class  MdCallable<T> implements Callable<List<T>> {
    Long limit0;
    Long limit1;
    Class<? extends IService<T>> clz;
    LambdaQueryWrapper<T> wrapper;
    SFunction<T, ?>[] columns;
    @Override
    public List<T> call() throws Exception {
        return SpringUtils.getBean(clz).list(wrapper.select(columns).last(" LIMIT " +limit0 + "," + limit1 + ";"));
    }
}

三、调用

java 复制代码
例如:List<ActEamRepairOrder> actEamRepairOrders = doPackageResCallables(ActEamRepairOrderService.class, new LambdaQueryWrapper<ActEamRepairOrder>(), threadNumber,ActEamRepairOrder::getOrderId,ActEamRepairOrder::getOrderName);
相关推荐
Boilermaker199242 分钟前
【Java EE】Mybatis-Plus
java·开发语言·java-ee
xdscode1 小时前
SpringBoot ThreadLocal 全局动态变量设置
java·spring boot·threadlocal
lifallen1 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法
丶小鱼丶1 小时前
链表算法之【合并两个有序链表】
java·算法·链表
张先shen2 小时前
Elasticsearch RESTful API入门:全文搜索实战(Java版)
java·大数据·elasticsearch·搜索引擎·全文检索·restful
天河归来3 小时前
springboot框架redis开启管道批量写入数据
java·spring boot·redis
张先shen3 小时前
Elasticsearch RESTful API入门:全文搜索实战
java·大数据·elasticsearch·搜索引擎·全文检索·restful
codervibe3 小时前
如何用 Spring Security 构建无状态权限控制系统(含角色菜单控制)
java·后端
codervibe3 小时前
项目中如何用策略模式实现多角色登录解耦?(附实战代码)
java·后端
TCChzp3 小时前
synchronized全链路解析:从字节码到JVM内核的锁实现与升级策略
java·jvm