使用SpringBoot和JPA實現(xiàn)批量處理新增、修改

更新時間：2023年06月26日 10:37:09 作者：procedure?rabbit

最近項目需要在JPA中使用ID進行批量更新,所以下面這篇文章主要給大家介紹了關于使用SpringBoot和JPA實現(xiàn)批量處理新增、修改的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下

jpa的sava與saveAll

save()方法

 @Transactional
    public <S extends T> S save(S entity) {
        if (this.entityInformation.isNew(entity)) {
            this.em.persist(entity);
            return entity;
        } else {
            return this.em.merge(entity);
        }
    }

根據(jù)源碼我們可以看出來，save是先通過判斷這個對象是不是新的，新的便會新增，否則就是執(zhí)行的修改。整個是有分兩步進行的，先查詢再新增

saveAll()方法

  @Transactional
    public <S extends T> List<S> saveAll(Iterable<S> entities) {
        Assert.notNull(entities, "The given Iterable of entities not be null!");
        List<S> result = new ArrayList();
        Iterator var3 = entities.iterator();
        while(var3.hasNext()) {
            S entity = var3.next();
            result.add(this.save(entity));
        }
        return result;
    }

saveAll（）方法是一種更新多條的一種方式，里面?zhèn)鞯拇鎸ο蟮募?。分析源碼我們可以看出saveAll()底層還是調(diào)用的save()方法，也就是每次需要先查詢再做修改。在使用上方便，但是因每次涉及到查詢、新增，事務的關系，導致修改或者新增耗時得非常的久。

那么下面我們將結合EntityManager對批量新增，修改做出優(yōu)化。

jpa結合Batch

配置文件

spring:
  #配置 Jpa
  jpa:
    properties:
      hibernate:
        generate_statistics: false
        order_insert: true    //配置批量新增
        order_update: true    //配置批量修改
        dialect: org.hibernate.dialect.MySQL5InnoDBDialect
        jdbc:
          batch_size: 1000    //容器內(nèi)批量的大小
    open-in-view: true

EntityManager

EntityManager其實就是一個容器，通過注解引入EntityManager對象使用EntityManager對新增，修改進行批量處理

/**
 * @author 程序員panda
 * @date 
 * @desc 批量處理
 */
@Service
@Transactional
public class BatchService {
    @PersistenceContext
    private EntityManager entityManager;
	//配置文件中每次批量提交的數(shù)量
    @Value("${spring.jpa.properties.hibernate.jdbc.batch_size}")
    private long batchSize;
    /**
     * 批量插入
     *
     * @param list 實體類集合
     * @param <T>  表對應的實體類
     */
    public <T> void batchInsert(List<T> list) {
        if (!ObjectUtils.isEmpty(list)){
            for (int i = 0; i < list.size(); i++) {
                entityManager.persist(list.get(i));
                if (i % batchSize == 0) {
                    entityManager.flush();
                    entityManager.clear();
                }
            }
            entityManager.flush();
            entityManager.clear();
        }
    }
    /**
     * 批量更新
     *
     * @param list 實體類集合
     * @param <T>  表對應的實體類
     */
    public <T> void batchUpdate(List<T> list) {
        if (!ObjectUtils.isEmpty(list)){
            for (int i = 0; i < list.size(); i++) {
                entityManager.merge(list.get(i));
                if (i % batchSize == 0) {
                    entityManager.flush();
                    entityManager.clear();
                }
            }
            entityManager.flush();
            entityManager.clear();
        }
    }
}

實際運用

選擇一個需要新增的table，將原有的saveAll()，改為調(diào)用自定義的batchInsert（）方法

 public JSONObject sumbitPhone(MultipartFile phoneFile) {
        long timeIdStart = System.currentTimeMillis();
        JSONObject result=new JSONObject();
        List<CheckPhone> phoneList=getPhoneListEn(phoneFile);
        batchService.batchInsert(phoneList);
        result.put("code",200);
        result.put("msg","提交成功");
        logger.info("提交預校驗數(shù)據(jù)用時:"+(System.currentTimeMillis() - timeIdStart) / 1000.0+"秒");
        return result;
    }