SpringBoot2 Jpa 批量刪除功能的實現

Posted on by

前臺處理

首先前臺先要獲取所有的要刪除數據的ID，並將ID拼接成字符串 例如： 2,3,4,5，然後通過GET請求返送到後臺。

後臺處理

控制器接收

/**
 * @function 批量刪除
 * @param stu_id
 * @return
*/
@GetMapping("/del_stu")
@ResponseBody
public Msg batch_del_stu(@RequestParam("stu_id") String stu_id){
 // 接收包含stuId的字符串，並將它分割成字符串數組
 String[] stuList = stu_id.split(",");
 // 將字符串數組轉為List<Intger> 類型
 List<Integer> LString = new ArrayList<Integer>();
 for(String str : stuList){
  LString.add(new Integer(str));
 }
 // 調用service層的批量刪除函數
 studentsService.deleteBatch(LString);
 return Msg.success().add("數組", LString);
}

service層

@Override
 public void deleteBatch(List<Integer> stuList) {
  // 第一種批量刪除方法，是通過spring data中繼承JpaRepository接口後，通過關鍵字拼接的方法進行刪除，刪除時候是通過ID一條一條刪除的
//  studentsRepository.deleteStudentsByStuIdIn(stuList);
  // 第二種批量刪除方法， 是通過自定義JPQL語句進行刪除，使用的是 where stuId in （）的操作
  studentsRepository.deleteBatch(stuList);
 }

Repository接口層

public interface StudentsRepository extends Repository<Students, Integer>, JpaRepository<Students, Integer> {
 /**
  * @function 自定義JPQL
  * @param ids
  */
 // 註意更新和刪除是需要加事務的， 並且要加上 @Modify的註解
 @Modifying
 @Transactional
 @Query("delete from Students s where s.stuId in (?1)")
 void deleteBatch(List<Integer> ids);
 // 這個是通過spring data拼接關鍵字進行的操作
 void deleteStudentsByStuIdIn(List<Integer> ids);
}

附加

@Modifying註解

1、在@Query註解中編寫JPQL實現DELETE和UPDATE操作時候必須加上@Modifying註解，通知Spring Data這是一個delete或者updata操作

2、 updata和delete操作需要使用事務，此時需要定義service層，在service方法上添加事務操作

3、 註意JPQL不支持insert操作

@Query 如果在註解中添加 nativeQuery=true 是支持原生SQL查詢

補充：SpringJPA 批量刪除引起的StackOverFlow

項目裡有一處根據Id，批量刪除一些歷史數據的代碼(xxxRepository.deleteInBatch(list);)，發現傳入list過大時，出現棧溢出(StackOverFlowError) 。

解決方法：

list切分成多份，循環批量刪除。

下面是簡單的瞭解一下執行流程。

deleteInBatch(Iterable<T> entities)
 
/**
 點進源碼 看看。
 org.springframework.data.jpa.repository.support.SimpleJpaRepository#deleteInBatch
*/
 @Transactional
 public void deleteInBatch(Iterable<T> entities) { 
 Assert.notNull(entities, "The given Iterable of entities not be null!"); 
 if (!entities.iterator().hasNext()) {
 return;
 }
 
// 繼續跟蹤
 applyAndBind(getQueryString(DELETE_ALL_QUERY_STRING, entityInformation.getEntityName()), entities, em)
 .executeUpdate();
 }
 
/**
 org.springframework.data.jpa.repository.query.QueryUtils#applyAndBind
*/
 
public static <T> Query applyAndBind(String queryString, Iterable<T> entities, EntityManager entityManager) {
 // ... 省略一些code
 // 最後會形成 delete from xx表 x(表別名) where x.id =? or x.id=?... 一條sql語句
 
  String alias = detectAlias(queryString);
 StringBuilder builder = new StringBuilder(queryString);
 builder.append(" where"); 
 int i = 0; 
 while (iterator.hasNext()) { 
 iterator.next(); 
 builder.append(String.format(" %s = ?%d", alias, ++i)); 
 if (iterator.hasNext()) {
 builder.append(" or");
 }
 }
 
 Query query = entityManager.createQuery(builder.toString()); 
 iterator = entities.iterator();
 i = 0; 
 while (iterator.hasNext()) {
 query.setParameter(++i, iterator.next());
 } 
}

結合日志記錄的錯誤信息，進入到org.hibernate.hql.internal.antlr.HqlSqlBaseWalker#logicalExpr 方法

下面貼一下調用棧

org.hibernate.hql.internal.antlr.HqlSqlBaseWalker#deleteStatement 方法中 whereClause()調用到瞭logicalExpr 方法。

由下圖可知，該方法在①處遞歸調用自身，會不斷的創建棧幀，當超出棧深度或者超出棧的大小後，會爆出 棧溢出。

至於① 處怎麼跳出繼續執行後面的代碼，還沒研究，有知道的小夥伴請指教，不正確的地方也請指正。

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：