Mybatis集成MySQL使用游標查詢處理大批量數(shù)據(jù)方式

更新時間：2024年01月27日 09:50:12 作者：WannaRunning

這篇文章主要介紹了Mybatis集成MySQL使用游標查詢處理大批量數(shù)據(jù)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

背景

基于數(shù)據(jù)的時間范圍查詢，給符合條件的用戶推送積分即將到期的提醒。

初期用戶量小使用最普通簡單的分頁查詢掃描數(shù)據(jù)處理數(shù)據(jù)沒問題。隨著用戶量的上升表數(shù)據(jù)已經(jīng)上千萬，每天掃描處理的數(shù)量也超百萬，limit分頁出現(xiàn)了慢sql，任務執(zhí)行時間也達不到預期了。

上述方案出現(xiàn)瓶頸后考慮放棄limit方案，使用游標的方式進行全量數(shù)據(jù)的獲取，這樣一來SQL執(zhí)行快任務執(zhí)行也快。

MySQL游標查詢

useCursorFetch

使用游標查詢時，必須在jdbc url上設置連接屬性參數(shù)useCursorFetch=true

FetchSize

在設置了useCursorFetch=true后,需要在SQL中指定fetchSize，即一次獲取的數(shù)據(jù)量。

如果不設置fetchSize參數(shù)，則執(zhí)行時仍然是全量返回，可能會出現(xiàn)OOM。

Mybatis集成Cursor查詢

mapper接口方法返回值聲明為Cursor類型，下面是SQL和Mapper的示例。

Cursor<Long> selectExpireCouponMember(@Param("endTime") String endTime);
 
 
<select id="selectExpireCouponMember" resultType="java.lang.Long" fetchSize="5000">
	select
		distinct member_id
	from t_dj_coupon
	where
	   end_time > end_time < #{endStartTime} and end_time < #{endTime}
</select>

下面是基于上面的SQL做大量數(shù)據(jù)查詢后寫入文件的代碼示例。

String fileName = DateFormatUtils.format(System.currentTimeMillis(), DateUtils.PATTERN_YYYY_MM_DD) + "_status_push_member.txt";
File file = new File(fileName);
file.createNewFile();
 
fileWriter = new FileWriter(file);
bufferedWriter = new BufferedWriter(fileWriter);
try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
	DjCouponMapper mapper = sqlSession.getMapper(DjCouponMapper.class);
	try (Cursor<Long> cursor = mapper.selectStartCouponMember(startDate, startDateEnd, BrandContextHolder.getBrandMdCode())) {
		Iterator<Long> iterator = cursor.iterator();
		Set<Long> couponMemberSet = new HashSet<>(pageSize.intValue());
		while (iterator.hasNext()) {
			couponMemberSet.add(iterator.next());
			writeNum++;
			if (couponMemberSet.size() >= pageSize) {
				bufferedWriter.write(couponMemberSet.toString());
				bufferedWriter.newLine();
				bufferedWriter.flush();
				writeLine++;
				couponMemberSet.clear();
			}
		
		}
		if (CollectionUtils.isNotEmpty(couponMemberSet)) {
			bufferedWriter.write(couponMemberSet.toString());
			bufferedWriter.newLine();
			bufferedWriter.flush();
			writeLine++;
		}
	}
}