Mybatis新增數據返回id詳解

一、Mybatis如何返回新增數據的主鍵id

Mybatis是一種基於Java語言的持久層框架，它提供了一種方便、快捷的方式來與資料庫進行交互。在進行資料庫操作時，我們經常會遇到一種需求：新增數據後需要獲取該數據的自增id。Mybatis如何滿足這一需求呢？

在Mybatis中，當執行insert語句後，我們可以通過標籤來獲取新插入記錄的id值，它有以下兩種配置方式：

1、useGeneratedKeys

使用useGeneratedKeys配置可以使用資料庫廠商的自增欄位返回策略來產生主鍵。在statement標籤中，添加useGeneratedKeys=”true”和keyProperty屬性即可返回主鍵。

// mapper.xml配置文件
<insert id="addUser" parameterType="User" useGeneratedKeys="true" keyProperty="id">
  INSERT INTO user(name) VALUES(#{name})
</insert>

// 使用方式
User user = new User("Tom");
userMapper.addUser(user);
Long id = user.getId();

2、selectKey

使用selectKey標籤，需要設置keyProperty、resultType、order（在什麼時候執行）、pre、post等屬性。通過配置產生一個獨立的sql語句來返回新的主鍵。

// mapper.xml配置文件
<insert id="addUser" parameterType="User">
  <selectKey keyProperty="id" resultType="int" order="AFTER">
    SELECT LAST_INSERT_ID()
  </selectKey>
  INSERT INTO user(name) VALUES(#{name})
</insert>

// 使用方式
User user = new User("Tom");
userMapper.addUser(user);
Long id = user.getId();

二、Mybatis如何支持批量插入並返回主鍵id

Mybatis提供了batch插入數據的方式，這種方式可以一次性向資料庫插入多條數據，提高了資料庫操作的效率。但是，在批量插入數據時，返回主鍵id的需求並沒有得到完美的解決。Mybatis官方文檔給出的建議是：使用SELECT LAST_INSERT_ID()或者SELECT SCOPE_IDENTITY()（具體使用方法同上）。

// mapper.xml配置文件
<insert id="batchInsertUser" parameterType="java.util.List">
  <selectKey keyProperty="id" resultType="int" order="AFTER">
    SELECT LAST_INSERT_ID()
  </selectKey>
  INSERT INTO user(name) VALUES
  <foreach collection="list" item="item" index="index" separator=",">
    (#{item.name})
  </foreach>
</insert>

// 使用方式
List<User> list = new ArrayList<>();
list.add(new User("Tom"));
list.add(new User("Jerry"));
userMapper.batchInsertUser(list);
Long id1 = list.get(0).getId();
Long id2 = list.get(1).getId();

三、Mybatis插入數據返回非自增主鍵id的方法

有些時候，我們需要插入的數據主鍵並不是使用資料庫自增值。這時我們可以在Java代碼中手動設置主鍵值，然後在insert語句中將主鍵值確定即可。

// mapper.xml配置文件
<insert id="addBook" parameterType="Book">
    INSERT INTO book(id, title, author) VALUES(#{id}, #{title}, #{author})
</insert>

// 使用方式
Book book = new Book(123456L, "Mybatis詳解", "Tom");
bookMapper.addBook(book);
Long id = book.getId();

在這種情況下，我們必須手動給實體類中的主鍵id屬性設置值。如果想要跟mybatis類似，自動生成對應id的話，我們可以使用UUID.randomUUID().toString()方法來生成一個隨機的字元串：

Book book = new Book(UUID.randomUUID().toString(), "Mybatis詳解", "Tom");
bookMapper.addBook(book);
String id = book.getId();

四、總結

通過本文的介紹，我們對Mybatis如何返回新增數據的主鍵id有了清晰的認識。在具體使用時，根據不同的需求選擇不同的方式，實現新增數據後的主鍵id獲取。同時，也要注意批量新增數據時返回主鍵id的問題，需要使用上述特殊的方案來解決。最後，對於需要插入非自增主鍵id的情況，我們需要在Java代碼中手動設置主鍵值，並在mapper.xml文件中加入對應的配置。