快捷導(dǎo)航

mybatis-plus saveOrUpdateBatch踩坑記錄

更新時(shí)間：2023年12月08日 09:42:48 作者：Dango丶小七

這篇文章主要介紹了mybatis-plus saveOrUpdateBatch踩坑記錄,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

mybatis-plus saveOrUpdateBatch踩坑

mybatis-plus版本: 3.5.1

調(diào)用方法

@Transactional(rollbackFor = Exception.class)  
default boolean saveOrUpdateBatch(Collection<T> entityList) {  
    return saveOrUpdateBatch(entityList, DEFAULT_BATCH_SIZE);  
}

問(wèn)題說(shuō)明

當(dāng)對(duì)entityList進(jìn)行批量更新操作時(shí)，方法內(nèi)部會(huì)根據(jù)主鍵查詢(xún)?cè)撚涗?，?dǎo)致批量更新操作十分緩慢

具體代碼如下：

@Transactional(rollbackFor = Exception.class)
@Override
public boolean saveOrUpdateBatch(Collection<T> entityList, int batchSize) {
    TableInfo tableInfo = TableInfoHelper.getTableInfo(entityClass);
    Assert.notNull(tableInfo, "error: can not execute. because can not find cache of TableInfo for entity!");
    String keyProperty = tableInfo.getKeyProperty();
    Assert.notEmpty(keyProperty, "error: can not execute. because can not find column for id from entity!");
    return SqlHelper.saveOrUpdateBatch(this.entityClass, this.mapperClass, this.log, entityList, batchSize, (sqlSession, entity) -> {
        Object idVal = tableInfo.getPropertyValue(entity, keyProperty);
        // 當(dāng)主鍵存在時(shí)，會(huì)執(zhí)行sqlSession.selectList(getSqlStatement(SqlMethod.SELECT_BY_ID), entity)查詢(xún)?cè)摋l記錄
        return StringUtils.checkValNull(idVal) || CollectionUtils.isEmpty(sqlSession.selectList(getSqlStatement(SqlMethod.SELECT_BY_ID), entity));
    }, (sqlSession, entity) -> {
        MapperMethod.ParamMap<T> param = new MapperMethod.ParamMap<>();
        param.put(Constants.ENTITY, entity);
        sqlSession.update(getSqlStatement(SqlMethod.UPDATE_BY_ID), param);
    });
}

public static <E> boolean saveOrUpdateBatch(Class<?> entityClass, Class<?> mapper, Log log, Collection<E> list, int batchSize, BiPredicate<SqlSession, E> predicate, BiConsumer<SqlSession, E> consumer) {
	String sqlStatement = getSqlStatement(mapper, SqlMethod.INSERT_ONE);
	return executeBatch(entityClass, log, list, batchSize, (sqlSession, entity) -> {
		// 執(zhí)行predicate
		if (predicate.test(sqlSession, entity)) {
			sqlSession.insert(sqlStatement, entity);
		} else {
			consumer.accept(sqlSession, entity);
		}
	});
}

public static <E> boolean executeBatch(Class<?> entityClass, Log log, Collection<E> list, int batchSize, BiConsumer<SqlSession, E> consumer) {
	Assert.isFalse(batchSize < 1, "batchSize must not be less than one");
	return !CollectionUtils.isEmpty(list) && executeBatch(entityClass, log, sqlSession -> {
		int size = list.size();
		int idxLimit = Math.min(batchSize, size);
		int i = 1;
		for (E element : list) {
			// 循環(huán)每條記錄，執(zhí)行consumer，在此次調(diào)用中，consumer中會(huì)執(zhí)行predicate，當(dāng)該條記錄主鍵不為空時(shí)，會(huì)通過(guò)主鍵查詢(xún)?cè)撚涗?
			consumer.accept(sqlSession, element);
			if (i == idxLimit) {
				sqlSession.flushStatements();
				idxLimit = Math.min(idxLimit + batchSize, size);
			}
			i++;
		}
	});
}