diff --git a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/core/mapper/BaseMapperPlus.java b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/core/mapper/BaseMapperPlus.java index 0d777f4e0224a6c33a1ed0e9ed2bd95e4836b798..9785b0832143393aa099fdbc37d41ae97b95706b 100644 --- a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/core/mapper/BaseMapperPlus.java +++ b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/core/mapper/BaseMapperPlus.java @@ -15,12 +15,15 @@ import org.apache.ibatis.logging.Log; import org.apache.ibatis.logging.LogFactory; import org.dromara.common.core.utils.MapstructUtils; import org.dromara.common.core.utils.StreamUtils; +import org.dromara.common.mybatis.handler.FilterResultHandler; import java.io.Serializable; +import java.util.ArrayList; import java.util.Collection; import java.util.List; import java.util.Map; import java.util.function.Function; +import java.util.function.Predicate; /** * 自定义 Mapper 接口, 实现 自定义扩展 @@ -355,4 +358,34 @@ public interface BaseMapperPlus extends BaseMapper { return StreamUtils.toList(this.selectObjs(wrapper), mapper); } + /** + * 根据查询条件和过滤条件查询数据,并将结果转换为指定的VO类型列表 + * + * @param wrapper 查询条件Wrapper + * @param filter 结果过滤条件 + * @return 查询到的符合条件的对象列表,经过转换为指定类型的对象后返回 + */ + default List selectFilterVoList(Wrapper wrapper, Predicate filter) { + return this.selectFilterVoList(wrapper, filter, this.currentVoClass()); + } + + /** + * 根据查询条件和过滤条件查询数据,并将结果转换为指定的VO类型列表 + * + * @param wrapper 查询条件Wrapper + * @param filter 结果过滤条件 + * @param voClass 要转换的VO类的Class对象 + * @param VO类的类型 + * @return 查询到的符合条件的对象列表,经过转换为指定类型的对象后返回 + */ + default List selectFilterVoList(Wrapper wrapper, Predicate filter, Class voClass) { + List list = new ArrayList<>(); + // 使用过滤处理器过滤数据,并将符合条件的结果加入 list + this.selectList(wrapper, new FilterResultHandler<>(filter, list::add)); + + if (CollUtil.isEmpty(list)) { + return CollUtil.newArrayList(); + } + return MapstructUtils.convert(list, voClass); + } } diff --git a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/BatchResultHandler.java b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/BatchResultHandler.java new file mode 100644 index 0000000000000000000000000000000000000000..cfa7a62338a432449c42934c1c4d63e2d47c1787 --- /dev/null +++ b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/BatchResultHandler.java @@ -0,0 +1,93 @@ +package org.dromara.common.mybatis.handler; + +import cn.hutool.core.collection.CollUtil; +import org.apache.ibatis.session.ResultContext; +import org.apache.ibatis.session.ResultHandler; + +import java.util.ArrayList; +import java.util.List; +import java.util.function.Consumer; + +/** + * 批量结果处理器 + * + *

将查询结果按指定批量大小缓存, + * 达到批量大小时调用批量消费函数进行批量处理,避免一次性加载全部数据导致内存压力过大 + * + *

适用于导出Excel、批量写库、定时上报等需要批量操作场景, + * 能有效降低内存使用并提升处理性能 + * + *

使用须知:
+ * 1. 构造时传入非空的批量消费函数和合法的批量大小(>0)
+ * 2. 结果回调时自动缓存并批量触发消费函数
+ * 3. 查询完成后务必调用 {@link #flush()},处理剩余不足批量的数据,避免数据遗漏 + * + * @param 查询结果类型 + * @author AprilWind + */ +public class BatchResultHandler implements ResultHandler { + + /** + * 批量数据消费函数 + * + *

注意:长时间的批量处理操作(如写Excel、写数据库或调用远程接口)可能导致数据库连接超时或断开, + * 以及MyBatis查询线程阻塞,影响系统性能和稳定性。 + * + *

因此推荐: + *

    + *
  • 结合合适的批量大小和限速策略,避免瞬时过载
  • + *
  • 采用异步写入机制,将批量处理操作与MyBatis查询线程解耦,减少阻塞
  • + *
  • 合理拆分查询任务,避免长时间持有数据库连接
  • + *
+ * + *

这样能有效保障系统的高性能和稳定运行 + */ + private final Consumer> batchConsumer; + + /** + * 批量大小,必须大于0 + */ + private final int batchSize; + + /** + * 当前缓存数据 + */ + private final List buffer; + + /** + * 构造批量结果处理器 + * + * @param batchConsumer 批量消费函数,不能为null + * @param batchSize 批量大小,必须大于0 + */ + public BatchResultHandler(Consumer> batchConsumer, int batchSize) { + this.batchConsumer = batchConsumer; + this.batchSize = batchSize; + this.buffer = new ArrayList<>(batchSize); + } + + /** + * MyBatis 查询结果回调,缓存单条结果, + * 达到批量大小时触发批量消费函数处理缓存数据 + * + * @param context 查询结果上下文,包含单条结果 + */ + @Override + public void handleResult(ResultContext context) { + buffer.add(context.getResultObject()); + if (buffer.size() >= batchSize) { + flush(); + } + } + + /** + * 刷新缓存,触发批量消费函数处理剩余数据, + * 通常查询结束后调用,确保所有数据均被处理 + */ + public void flush() { + if (CollUtil.isNotEmpty(buffer)) { + batchConsumer.accept(new ArrayList<>(buffer)); + buffer.clear(); + } + } +} diff --git a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/FilterResultHandler.java b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/FilterResultHandler.java new file mode 100644 index 0000000000000000000000000000000000000000..0c9096700da86047858ab9c3aeafba7c6f4abab7 --- /dev/null +++ b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/FilterResultHandler.java @@ -0,0 +1,57 @@ +package org.dromara.common.mybatis.handler; + +import org.apache.ibatis.session.ResultContext; +import org.apache.ibatis.session.ResultHandler; + +import java.util.function.Consumer; +import java.util.function.Predicate; + +/** + * 过滤处理器 + *

+ * 通过传入的过滤条件({@link Predicate})筛选查询结果, + * 仅对满足条件的结果调用消费函数进行处理 + *

+ * 适用于需要对查询结果做条件过滤再处理的场景 + * 避免无用数据进入后续处理逻辑,提高效率 + * + * @param 查询结果的数据类型 + * @author AprilWind + */ +public class FilterResultHandler implements ResultHandler { + + /** + * 过滤条件,用于判断是否处理该条结果 + */ + private final Predicate filter; + + /** + * 处理满足过滤条件的结果的消费函数 + */ + private final Consumer consumer; + + /** + * 构造过滤处理器 + * + * @param filter 过滤条件 + * @param consumer 满足过滤条件时的处理函数 + */ + public FilterResultHandler(Predicate filter, Consumer consumer) { + this.filter = filter; + this.consumer = consumer; + } + + /** + * 查询结果回调,先用过滤条件判断是否处理 + * 满足条件则调用消费函数,否则忽略该条结果 + * + * @param context 查询结果上下文,包含单条结果 + */ + @Override + public void handleResult(ResultContext context) { + T result = context.getResultObject(); + if (filter.test(result)) { + consumer.accept(result); + } + } +} diff --git a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/SimpleResultHandler.java b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/SimpleResultHandler.java new file mode 100644 index 0000000000000000000000000000000000000000..13dc06010905d28f3b340f8817aa83b5e1133c43 --- /dev/null +++ b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/handler/SimpleResultHandler.java @@ -0,0 +1,44 @@ +package org.dromara.common.mybatis.handler; + +import org.apache.ibatis.session.ResultContext; +import org.apache.ibatis.session.ResultHandler; + +import java.util.function.Consumer; + +/** + * 单条处理处理器 + * + *

适用于无需缓存或批量处理,直接对结果逐条消费的场景 + * 逻辑简单直接,实时处理每条数据 + * + * @param 查询结果的数据类型 + * @author AprilWind + */ +public class SimpleResultHandler implements ResultHandler { + + /** + * 消费单条查询结果的函数 + */ + private final Consumer consumer; + + /** + * 构造 SimpleResultHandler 实例 + * + * @param consumer 处理单条查询结果的消费函数,不能为空 + */ + public SimpleResultHandler(Consumer consumer) { + this.consumer = consumer; + } + + /** + * MyBatis 查询结果回调方法, + * 每接收到一条查询结果即调用消费函数进行处理 + * + * @param context 查询结果上下文,包含当前单条结果对象 + */ + @Override + public void handleResult(ResultContext context) { + T result = context.getResultObject(); + consumer.accept(result); + } +}