快捷導(dǎo)航

springboot整合mybatis實(shí)現(xiàn)數(shù)據(jù)庫(kù)的更新批處理方式

更新時(shí)間：2023年03月12日 16:21:27 作者：qq_1757537040

這篇文章主要介紹了springboot整合mybatis實(shí)現(xiàn)數(shù)據(jù)庫(kù)的更新批處理方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

springboot整合mybatis實(shí)現(xiàn)數(shù)據(jù)庫(kù)更新批處理

1.在mapper接口中編寫(xiě)方法

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

2.在mapper.xml中編寫(xiě)對(duì)相應(yīng)的更新sql語(yǔ)句

<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>

3.這個(gè)配置文件的sql語(yǔ)句流程如下：

update t_book(表名)
set sales(這個(gè)是數(shù)據(jù)庫(kù)的銷量字段名) = case book_id(這個(gè)是數(shù)據(jù)庫(kù)的id字段名)
? ? when bookid(從list集合中取出來(lái)的) then sales+(從集合中取出的數(shù)據(jù))
? ? ...(這里可以一直進(jìn)行拼接)
? end,
? ? stock(這個(gè)是數(shù)據(jù)庫(kù)的庫(kù)存字段名) = CASE book_id(這個(gè)是數(shù)據(jù)庫(kù)的id字段名)
? ? when bookid(從list集合中取出來(lái)的) then stock-(從集合中取出數(shù)據(jù))
? ? ...(這里可以一直進(jìn)行拼接)
? end,
where `book_id`(這個(gè)是數(shù)據(jù)庫(kù)的id字段名) IN(bookid(從list集合中取出來(lái)),bookid(從list集合中取出來(lái))...)

4.這個(gè)sql語(yǔ)句的含義：

更新表里面的數(shù)據(jù)根據(jù)集合遍歷出來(lái)的id值，設(shè)置要更新的字段名，讓要更新的字段值跟這個(gè)表的主鍵id進(jìn)行綁定，當(dāng)這個(gè)主鍵id與list中取出來(lái)的id值一致時(shí)就讓這個(gè)要更新的字段名，取then后面的值

Mybatis批量更新數(shù)據(jù)庫(kù) MybatisBatchUtils batchInsertupdate spring boot

MybatisBatchUtils

    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        需要被處理的數(shù)據(jù)
     * @param mapperClass Mybatis的Mapper類
     * @param function    自定義處理邏輯
     * @return int 影響的總行數(shù)
     */
    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++;
            }
            // 非事務(wù)環(huán)境下強(qiáng)制commit，事務(wù)情況下該commit相當(dāng)于無(wú)效
            batchSqlSession.commit(true);
        } catch (Exception e) {
            batchSqlSession.rollback();
            // throw new BusinessException(e.getMessage());
            log.error("batchUpdateOrInsert", e);
        } finally {
            batchSqlSession.close();
        }
        return i - 1;
    }
}