如何使用SpringBoot-JPA進行自定義保存及批量保存功能

這篇文章主要介紹如何使用SpringBoot-JPA進行自定義保存及批量保存功能，文中介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們一定要看完！

說明

SpringBoot版本：2.1.4.RELEASE

java版本：1.8

文中所說JPA皆指spring-boot-starter-data-jpa

使用JPA保存一個Student對象

在JPA中保存一個對象，僅需要該對象，一個倉儲即可。StudentDO實體類：

@Getter@Setter@Entity@Table(name = "t_student")public class StudentDO {  @Id  @GeneratedValue(strategy = GenerationType.IDENTITY)  @Column  private Long id;  @Column  private String seq;  @Column  private String name;  @Column  private int sex;}

JPA倉儲：

@Repositorypublic interface StudentRepo extends JpaRepository<StudentDO, String> {}

一般的，我們只需要調用StudentRepo.save()方法即可完成對實體對象的保存操作。

@Test  public void testSave() {    StudentDO student = new StudentDO();    student.setName("張三");    student.setSex(1);    student.setSeq("123456");    studentRepo.save(student);    Assert.assertNotNull(student.getId());  }

在插入過程中使用mysql函數

如果我們希望student的seq值由系統自動生成，且生成規則為“yyMMdd + 8位自增序列”（例如19060310000000）又該如何實現呢？

首先想到的是該如何生成這一串序列，mysql不像oracle自身支持sequence，因此在這里可以借用函數以及額外的sequence表來實現這一操作，網上有很多實現方式，這里就不再贅述。

現在已經有了函數getseq('student_seq')可以獲取到該序列，該如何將其應用到保存對象的方法中？顯然的一個問題是，像上面那樣再直接調用save方法已經行不通了，應該得需要自定義插入的sql實現。

一個容易想到的辦法是，在StudentDO類上使用注解@SQLInsert來定義insert的實現，它寫起來應該會像這個樣子：

@SQLInsert(sql = "INSERT INTO t_student(seq, name, sex) VALUES (getseq('student_seq'), ?, ?")

這條sql語句本身并沒有什么問題，再次調用save()方法也確實能夠執行。但是很可惜，它確會拋出一個sql異常：

java.sql.SQLException: Parameter index out of range (3 > number of parameters, which is 2).

顯然是程序認為有多少個參數，就得有多少個“?”與之匹配，目前我并沒有找到解決這個問題的方案，所以這種方法宣告失敗。

既然@SQLInsert行不通，或許可以考慮使用@Query注解來自定義一個實現。我們可以在StudentRepo中定義這樣一個方法：

@Transactional  @Modifying  @Query(value = "INSERT INTO t_student(seq, name, sex) VALUES (getseq('student_seq'), :#{#student.name}, :#{#student.sex})", nativeQuery = true)  int insert(@Param("student") StudentDO student);

試著運行一下，結果很成功，對象被正常的存儲到數據庫中，并且seq的取值也正常。看上去我們的問題已經得到了解決，但事實真的如此么？

自定義的批量存儲

上面的例子中，我們成功通過JPA調用了mysql函數將對象存儲到數據庫中。但上面的例子只提供了單個保存的方法，如果我們想批量保存呢？@Query里面的sql能夠進行改造么？我并沒有找到@Query中使用List作為參數的insert方法，但是這并不代表這一操作不能執行。JPA仍舊提供給了使用者原始的使用方式：利用EntityManager來構造sql并執行。

@PersistenceContext  private EntityManager entityManager;  private int batchInsert(List<StudentDO> students) {    StringBuilder sb = new StringBuilder();    sb.append("INSERT INTO t_student(seq, name, sex) VALUES ");    for(StudentDO student : students) {      sb.append("(getseq('student_seq'), ?, ?),");    }    String sql = sb.toString().substring(0, sb.length() - 1);    Query query = entityManager.createNativeQuery(sql);    int paramIndex = 1;    for(StudentDO student : students) {      query.setParameter(paramIndex++, student.getName());      query.setParameter(paramIndex++, student.getSex());    }    return query.executeUpdate();  }

就像MyBatis一樣，使用者也可以自定義SQL來執行，試試看，同樣沒有問題，再多的數據也可以被保存到數據庫中！批量保存的效果達到了。

再仔細想一想，通過上面的過程，還有什么問題么？對比JPA自帶的save()方法，似乎我們的自定義保存返回的都是int結果，也就是操作影響的數據庫行數。使用過JPA的人都應該了解，JPA的save()方法（或者其他JPA方法）返回的對象是經過持久化的，得益于這一特性，使用者可以在調用save()方法之后獲取到對象的id等必須先插入到數據庫之后才會有的值。顯然這里的操作已經失去了這一特性，那如果我們把返回值對應的改為Object或者List可以做到么？答案是并不能，我們會得到如下異常：

org.springframework.dao.InvalidDataAccessApiUsageException: Modifying queries can only use void or int/Integer as return type!; nested exception is java.lang.IllegalArgumentException: Modifying queries can only use void or int/Integer as return type!

insert方法必須使用@Modifying進行注解，而@Modifying注解的方法又只能返回int類型的結果。這種情況下或許只能先利用查詢得到seq的值再進行操作。

以上是“如何使用SpringBoot-JPA進行自定義保存及批量保存功能”這篇文章的所有內容，感謝各位的閱讀！希望分享的內容對大家有幫助，更多相關知識，歡迎關注億速云行業資訊頻道！