Mybatis中流式查詢的實現(xiàn)示例
1.ResultHandler結(jié)果集處理
1.1 ResultHandler 簡介
Mybatis 中的 ResultHandler 相當(dāng)于數(shù)據(jù)結(jié)果集的處理器，它是一個回調(diào)函數(shù)（Callback），用來處理每一行數(shù)據(jù)的結(jié)果，這個回調(diào)函數(shù)可以在查詢結(jié)果處理到一定量時觸發(fā)，對結(jié)果集數(shù)據(jù)進行定制化處理。
ResultHandler 的使用可以大幅提升數(shù)據(jù)處理的效率，當(dāng)我們需要處理大量的數(shù)據(jù)時，一般會使用 ResultHandler 來進行結(jié)果的處理，避免一次查詢就全部返回結(jié)果，浪費內(nèi)存資源或造成 OOM。
ResultHandler 接口只有一個方法 handleResult，我們可以實現(xiàn)這個方法去處理每一條 SQL 查詢返回的數(shù)據(jù)。
1.2 ResultHandler 實現(xiàn)結(jié)果流式輸出（兩種寫法）
Mybatis：
<select id="selectByResultHandler" resultMap="BaseResultMap">
SELECT
<include refid="Base_Column_List" />
FROM tbl_test_user
</select>
Mapper:
package com.test.dao;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.session.ResultHandler;
public interface TblTestUserMapper {
// SQL 返回值為 void，所以我們并沒有接收這個返回值。所以不會產(chǎn)生大對象
// 方法中的第一個參數(shù)是查詢時的限制條件，第二個是結(jié)果處理器
void selectByResultHandler(@Param(value = "actNo") String actNo,
ResultHandler resultHandler);
}
實現(xiàn)ResultHandler：
package com.test.handler;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.ResultContext;
import org.apache.ibatis.session.ResultHandler;
import com.test.model.TblTestUser;
@Slf4j
public class TestResultHandler implements ResultHandler<TblTestUser> {
@Override
public void handleResult(ResultContext<? extends TblTestUser> resultContext) {
TblTestUser tblTestUser = resultContext.getResultObject();
// TODO 處理數(shù)據(jù)的邏輯
try {
// 打印每行日志
log.info("userNo : [{}]", tblTestUser.getUserNo());
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}Service:
package com.test.service;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.ResultContext;
import org.apache.ibatis.session.ResultHandler;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import com.test.dao.TblTestUserMapper;
import com.test.model.TblTestUser;
import javax.annotation.Resource;
@Service
@Slf4j
public class TestService {
@Resource
private TblTestUserMapper tblTestUserMapper;
/**
* 寫法 1
*
* @param actNo 篩選條件：活動號
*/
@Async
public void test1(String actNo) {
log.info("ResultHandler1 處理查詢結(jié)果");
TestResultHandler testResultHandler = new TestResultHandler();
tblTestUserMapper.selectByResultHandler(actNo, testResultHandler);
}
/**
* 寫法 2
*
* @param actNo 篩選條件：活動號
*/
@Async
public void test2(String actNo) {
log.info("ResultHandler2 處理查詢結(jié)果");
tblTestUserMapper.selectByResultHandler(actNo, new ResultHandler<TblTestUser>() {
@Override
public void handleResult(ResultContext<? extends TblTestUser> resultContext) {
TblTestUser tblTestUser = resultContext.getResultObject();
// 處理數(shù)據(jù)的邏輯
try {
// 打印每行日志
log.info("userNo : [{}]", tblTestUser.getUserNo());
} catch (Exception e) {
throw new RuntimeException(e);
}
}
});
}
}
2.游標（Cursor）
MyBatis 3.2 以上版本支持游標，允許逐行處理結(jié)果集。它提供了一個叫 org.apache.ibatis.cursor.Cursor 的接口類用于流式查詢，這個接口繼承了 java.io.Closeable 和 java.lang.Iterable 接口，由此可知：
Cursor 是可關(guān)閉的；
Cursor 是可遍歷的。
除此之外，Cursor 還提供了三個方法：
isOpen()：用于在取數(shù)據(jù)之前判斷 Cursor 對象是否是打開狀態(tài)。只有當(dāng)打開時 Cursor 才能取數(shù)據(jù)；isConsumed()：用于判斷查詢結(jié)果是否全部取完。getCurrentIndex()：返回已經(jīng)獲取了多少條數(shù)據(jù)
因為 Cursor 實現(xiàn)了迭代器接口，因此在實際使用當(dāng)中，從 Cursor 取數(shù)據(jù)非常簡單， 在 Mapper 接口中，通過指定 Mapper 方法的返回值為 Cursor 類型，MyBatis 就知道這個查詢方法一個流式查詢。
cursor.forEach(rowObject -> {...}); 2.1 構(gòu)建Cursor
Mapper：
@Mapper
public interface FooMapper {
@Select("select * from foo limit #{limit}")
Cursor scan(@Param("limit") int limit);
}2.1.1 錯誤示范
@GetMapping("foo/scan/0/{limit}")
public void scanFoo0(@PathVariable("limit") int limit) throws Exception {
try (Cursor cursor = fooMapper.scan(limit)) { // 1
cursor.forEach(foo -> {}); // 2
}
} 說明：
上面的代碼中，fooMapper 是 @Autowired 進來的。
注釋 1 處調(diào)用 scan 方法，得到 Cursor 對象并保證它能最后關(guān)閉；
注釋 2 處則是從 cursor 中取數(shù)據(jù)。
上面的代碼看上去沒什么問題，但是執(zhí)行 scanFoo0() 時會報錯：
java.lang.IllegalStateException: A Cursor is already closed.
這是因為我們前面說了在取數(shù)據(jù)的過程中需要保持數(shù)據(jù)庫連接，而 Mapper 方法通常在執(zhí)行完后連接就關(guān)閉了，因此 Cusor 也一并關(guān)閉了。
所以，解決這個問題的思路不復(fù)雜，保持數(shù)據(jù)庫連接打開即可。我們至少有三種方案可選。
2.1.2 正確示范
SqlSessionFactory
我們可以用 SqlSessionFactory 來手工打開數(shù)據(jù)庫連接：
@GetMapping("foo/scan/1/{limit}")
public void scanFoo1(@PathVariable("limit") int limit) throws Exception {
try (
SqlSession sqlSession = sqlSessionFactory.openSession(); // 1
Cursor cursor =
sqlSession.getMapper(FooMapper.class).scan(limit) // 2
) {
cursor.forEach(foo -> { });
}
}上面的代碼中，1 處我們開啟了一個 SqlSession （實際上也代表了一個數(shù)據(jù)庫連接），并保證它最后能關(guān)閉；2 處我們使用 SqlSession 來獲得 Mapper 對象。這樣才能保證得到的 Cursor 對象是打開狀態(tài)的。
TransactionTemplate
在 Spring 中，我們可以用 TransactionTemplate 來執(zhí)行一個數(shù)據(jù)庫事務(wù)，這個過程中數(shù)據(jù)庫連接同樣是打開的：
@GetMapping("foo/scan/2/{limit}")
public void scanFoo2(@PathVariable("limit") int limit) throws Exception {
TransactionTemplate transactionTemplate =
new TransactionTemplate(transactionManager); // 1
transactionTemplate.execute(status -> { // 2
try (Cursor cursor = fooMapper.scan(limit)) {
cursor.forEach(foo -> { });
} catch (IOException e) {
e.printStackTrace();
}
return null;
});
}上面的代碼中，1 處我們創(chuàng)建了一個 TransactionTemplate 對象（此處 transactionManager 是怎么來的不用多解釋，本文假設(shè)讀者對 Spring 數(shù)據(jù)庫事務(wù)的使用比較熟悉了），2 處執(zhí)行數(shù)據(jù)庫事務(wù)，而數(shù)據(jù)庫事務(wù)的內(nèi)容則是調(diào)用 Mapper 對象的流式查詢。注意這里的 Mapper 對象無需通過 SqlSession 創(chuàng)建。
@Transactional 注解
這個本質(zhì)上和方案二一樣，代碼如下：
@GetMapping("foo/scan/3/{limit}")
@Transactional
public void scanFoo3(@PathVariable("limit") int limit) throws Exception {
try (Cursor cursor = fooMapper.scan(limit)) {
cursor.forEach(foo -> { });
}
}它僅僅是在原來方法上面加了個 @Transactional 注解。這個方案看上去最簡潔，但請注意 Spring 框架當(dāng)中注解使用的坑：只在外部調(diào)用時生效。在當(dāng)前類中調(diào)用這個方法，依舊會報錯。
到此這篇關(guān)于Mybatis中流式查詢的實現(xiàn)示例的文章就介紹到這了,更多相關(guān)Mybatis 流式查詢內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Java數(shù)據(jù)結(jié)構(gòu)優(yōu)先隊列實練
通常都把隊列比喻成排隊買東西，大家都很守秩序，先排隊的人就先買東西。但是優(yōu)先隊列有所不同，它不遵循先進先出的規(guī)則，而是根據(jù)隊列中元素的優(yōu)先權(quán)，優(yōu)先權(quán)最大的先被取出，這篇文章主要介紹了java優(yōu)先隊列的真題，感興趣的朋友一起看看吧2022-07-07
SpringBoot整合Groovy腳本實現(xiàn)動態(tài)編程詳解
這篇文章主要為大家介紹了SpringBoot整合Groovy腳本實現(xiàn)動態(tài)編程詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-09-09
這篇文章主要介紹了記一次Feign中如何傳實體Bean的問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-03-03
這篇文章主要介紹了Java中判斷集合是否相等的幾種方法詳解,在平時的開發(fā)中,可能會遇到需要判斷兩個集合是否相等的需求,那么本文就來詳細講解一下幾種實現(xiàn)方法,需要的朋友可以參考下2023-08-08
Java封裝數(shù)組之動態(tài)數(shù)組實現(xiàn)方法詳解
這篇文章主要介紹了Java封裝數(shù)組之動態(tài)數(shù)組實現(xiàn)方法,結(jié)合實例形式詳細分析了java動態(tài)數(shù)組的實現(xiàn)原理、操作步驟與相關(guān)注意事項,需要的朋友可以參考下2020-03-03
本篇文章主要介紹了Spring Boot 配置多個RabbitMQ，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-06-06
本文主要介紹了Java實現(xiàn)非對稱加密的三種方法,主要包括非對稱加密算法--DH（密鑰交換）,非對稱加密算法--RSA,非對稱加密算法--EIGamal,具有一定的參考價值,感興趣的可以了解一下2023-12-12
MyBatis Log是IDEA一款下載量非常高的插件，該插件可以對控制臺打印的日志進行解析，然后將對應(yīng)的SQL語句整理并拼接好對應(yīng)的參數(shù)，非常方便。這篇文章給大家介紹MyBatis Log 插件無法顯示SQL語句的原因，感興趣的朋友跟隨小編一起看看吧2020-09-09
這篇文章主要為大家詳細介紹了java實現(xiàn)文件下載的兩種方式，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-11-11
Spring MVC傳遞接收參數(shù)方式小結(jié)
大家在開發(fā)中經(jīng)常會用到Spring MVC Controller來接收請求參數(shù)，主要常用的接收方式就是通過實體對象以及形參等方式、有些用于GET請求，有些用于POST請求，有些用于兩者，下面介紹幾種常見的Spring MVC傳遞接收參數(shù)的方式2021-11-11