springboot整合mybatis實現資料庫的更新批次處理方式

/**
 * 修改book表中的銷量和庫存
 * 要使用批次處理
 */
Integer batchBookCountStork(@Param("bookList") List<CartItem> bookList);

<update id="batchBookCountStork" parameterType="java.util.List">
    UPDATE t_book
    <set>
        <foreach collection="bookList" item="book" index="index" open="`sales` = CASE `book_id`" close="END,">
            WHEN #{book.bookId} THEN sales+#{book.count}
        </foreach>
        <foreach collection="bookList" item="book" index="index" open="`stock` = CASE `book_id`" close="END,">
            WHEN #{book.bookId} THEN stock-#{book.count}
        </foreach>
    </set>
    <where>
        <foreach collection="bookList" item="book" index="index" open="`book_id` IN(" close=")" separator=",">
            #{book.bookId}
        </foreach>
    </where>
  </update>

update t_book(表名)
set sales(這個是資料庫的銷量欄位名) = case book_id(這個是資料庫的id欄位名)
    when bookid(從list集合中取出來的) then sales+(從集合中取出的資料)
    ...(這裡可以一直進行拼接)
  end,
    stock(這個是資料庫的庫存欄位名) = CASE book_id(這個是資料庫的id欄位名)
    when bookid(從list集合中取出來的) then stock-(從集合中取出資料)
    ...(這裡可以一直進行拼接)
  end,
where `book_id`(這個是資料庫的id欄位名) IN(bookid(從list集合中取出來),bookid(從list集合中取出來)...)

    int cnt = mybatisBatchUtils.batchUpdateOrInsert(addList, UiConfigDetailMapper.class,
                            (item, uiConfigDetailMapper) -> uiConfigDetailMapper.insertSelective(item));

package cn.XXX.dao.serivce.common;

import com.XXX.doctorusercenter.exception.BusinessException;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.List;
import java.util.function.BiFunction;


@Slf4j
@Component
public class MybatisBatchUtils {

    /**
     * 每次處理1000條
     */
    private static final int BATCH_SIZE = 1000;

    @Resource
    private SqlSessionFactory sqlSessionFactory;

    /**
     * 批次處理修改或者插入
     *
     * @param data        需要被處理的資料
     * @param mapperClass Mybatis的Mapper類
     * @param function    自定義處理邏輯
     * @return int 影響的總行數
     */
    public <T, U, R> int batchUpdateOrInsert(List<T> data, Class<U> mapperClass, BiFunction<T, U, R> function)  {
        int i = 1;
        SqlSession batchSqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH);
        try {
            U mapper = batchSqlSession.getMapper(mapperClass);
            int size = data.size();
            for (T element : data) {
              function.apply(element, mapper);
                if ((i % BATCH_SIZE == 0) || i == size) {
                    batchSqlSession.flushStatements();
                }
                i++;
            }
            // 非事務環境下強制commit，事務情況下該commit相當於無效
            batchSqlSession.commit(true);
        } catch (Exception e) {
            batchSqlSession.rollback();
            // throw new BusinessException(e.getMessage());
            log.error("batchUpdateOrInsert", e);
        } finally {
            batchSqlSession.close();
        }
        return i - 1;
    }
}