扩展tk.mybatis的流式查询功能实现
mybatis查询默认是一次获取全部, 有时候需要查询上万上百万数据时,如果一次性读取到内存中,会容易导致OOM问题。这时候需要采用流式查询。以下扩展了tk.mybatis的流式查询功能。 直接上干货:
@Options注解是关键
import org.apache.ibatis.annotations.Options; import org.apache.ibatis.annotations.SelectProvider; import org.apache.ibatis.mapping.ResultSetType; import org.apache.ibatis.session.ResultHandler; /** * 通用Mapper接口,流式查询 * * @param <T> 不能为空 * @author sunchangtan */ @tk.mybatis.mapper.annotation.RegisterMapper public interface SelectStreamByExampleMapper<T> { /** * 根据example条件和RowBounds进行流式查询 * * @param example * @return */ @Options(resultSetType = ResultSetType.FORWARD_ONLY, fetchSize = 1000) @SelectProvider(type = StreamExampleProvider.class, method = "dynamicSQL") void selectStreamByExampleMapper(Object example, ResultHandler resultHandler); }
带RowBounds的流式查询
import org.apache.ibatis.annotations.Options; import org.apache.ibatis.annotations.SelectProvider; import org.apache.ibatis.mapping.ResultSetType; import org.apache.ibatis.session.ResultHandler; import org.apache.ibatis.session.RowBounds; /** * 通用Mapper接口,流式查询 * * @param <T> 不能为空 * @author sunchangtan */ @tk.mybatis.mapper.annotation.RegisterMapper public interface SelectStreamByExampleRowBoundsMapper<T> { /** * 根据example条件和RowBounds进行流式查询 * * @param example * @return */ @Options(resultSetType = ResultSetType.FORWARD_ONLY, fetchSize = 1000) @SelectProvider(type = StreamExampleProvider.class, method = "dynamicSQL") void selectStreamByExampleRowBoundsMapper(Object example, RowBounds rowBounds, ResultHandler resultHandler); }
流式ExampleProvider
import org.apache.ibatis.mapping.MappedStatement; import tk.mybatis.mapper.mapperhelper.MapperHelper; import tk.mybatis.mapper.provider.ExampleProvider; /** * 流式查询的SqlProvider * @author sunchangtan */ public class StreamExampleProvider extends ExampleProvider { public StreamExampleProvider(Class<?> mapperClass, MapperHelper mapperHelper) { super(mapperClass, mapperHelper); } /** * 根据Example流式查询 * * @param ms * @return */ public String selectStreamByExampleMapper(MappedStatement ms) { return this.selectByExample(ms); } /** * 根据Example和RowBounds流式查询 * @param ms * @return */ public String selectStreamByExampleRowBoundsMapper(MappedStatement ms) { return this.selectByExample(ms); } }
将SelectStreamByExampleMapper和SelectStreamByExampleRowBoundsMapper组合成一个接口
/** * 流式查询接口 * @param <T> * @author sunchangtan */ @tk.mybatis.mapper.annotation.RegisterMapper public interface StreamMapper<T> extends SelectStreamByExampleMapper<T>, SelectStreamByExampleRowBoundsMapper<T> { }
在BaseMapper中加入StreamMapper
/** * Mapper的基类 * @author sunchangtan * @param <T> */ public interface BaseMapper<T> extends Mapper<T>, MySqlMapper<T>, StreamMapper<T> { }
使用例子:
this.userMapper.selectStreamByExampleRowBoundsMapper(example, new RowBounds(0, 1), resultContext -> { User user= (User) resultContext.getResultObject(); System.out.println(user); }); this.userMapper.selectStreamByExampleMapper(example, resultContext -> { User user= (User) resultContext.getResultObject(); System.out.println(User); });
到此这篇关于扩展tk.mybatis的流式查询功能实现的文章就介绍到这了,更多相关tk.mybatis 流式查询内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
相关文章
Kafka中的producer拦截器与consumer拦截器详解
这篇文章主要介绍了Kafka中的producer拦截器与consumer拦截器详解,Producer 的Interceptor使得用户在消息发送前以及Producer回调逻辑前有机会对消息做 一些定制化需求,比如修改消息等,需要的朋友可以参考下2023-12-12使用Maven 搭建 Spring MVC 本地部署Tomcat的详细教程
这篇文章主要介绍了使用Maven 搭建 Spring MVC 本地部署Tomcat,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2021-08-08Java重写(Override)与重载(Overload)区别原理解析
这篇文章主要介绍了Java重写(Override)与重载(Overload)区别原理解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-02-02解决Spring Boot 在localhost域奇怪的404问题(Mac book pro)
这篇文章主要介绍了解决Spring Boot 在localhost域奇怪的404问题(Mac book pro),需要的朋友可以参考下2017-09-09
最新评论