快捷導(dǎo)航

MyBatis輕松獲取自增長(zhǎng)主鍵的幾種方法小結(jié)

更新時(shí)間：2025年06月18日 09:41:47 作者：墨瑾軒

在日常開發(fā)中,獲取自增長(zhǎng)主鍵是一個(gè)非常常見的需求,尤其是在使用MyBatis進(jìn)行數(shù)據(jù)庫(kù)操作時(shí),然而,很多人在初次接觸這個(gè)功能時(shí)會(huì)感到困惑,不知道該如何下手,別擔(dān)心,本文將詳細(xì)介紹MyBatis獲取自增長(zhǎng)主鍵的幾種方法,需要的朋友可以參考下

引言：告別繁瑣，擁抱高效！

在日常開發(fā)中，獲取自增長(zhǎng)主鍵是一個(gè)非常常見的需求，尤其是在使用MyBatis進(jìn)行數(shù)據(jù)庫(kù)操作時(shí)。然而，很多人在初次接觸這個(gè)功能時(shí)會(huì)感到困惑，不知道該如何下手。別擔(dān)心，本文將詳細(xì)介紹MyBatis獲取自增長(zhǎng)主鍵的幾種方法，并附上詳細(xì)的代碼示例和注意事項(xiàng)，讓你輕松掌握這一技能！

第一步：使用 useGeneratedKeys 獲取自增長(zhǎng)主鍵

概述

useGeneratedKeys 是 MyBatis 提供的一個(gè)屬性，用于在插入數(shù)據(jù)時(shí)獲取數(shù)據(jù)庫(kù)自動(dòng)生成的主鍵值。這種方法簡(jiǎn)單直接，適用于大多數(shù)支持自增長(zhǎng)主鍵的數(shù)據(jù)庫(kù)。

代碼示例

假設(shè)我們有一個(gè) User 表，其結(jié)構(gòu)如下：

CREATE TABLE `user` (
  `id` INT(11) NOT NULL AUTO_INCREMENT,
  `name` VARCHAR(255) NOT NULL,
  `email` VARCHAR(255) NOT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

對(duì)應(yīng)的 User 實(shí)體類如下：

public class User {
    private Integer id;
    private String name;
    private String email;

    // Getters and Setters
}

Mapper 接口

public interface UserMapper {
    int insertUser(User user);
}

Mapper XML 文件

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.example.mapper.UserMapper">

    <!-- 使用 useGeneratedKeys 獲取自增長(zhǎng)主鍵 -->
    <insert id="insertUser" parameterType="com.example.entity.User" useGeneratedKeys="true" keyProperty="id">
        INSERT INTO user (name, email) VALUES (#{name}, #{email})
    </insert>

</mapper>

測(cè)試代碼

@SpringBootTest
public class MyBatisTest {

    @Autowired
    private UserMapper userMapper;

    @Test
    public void testInsertUser() {
        User user = new User();
        user.setName("John Doe");
        user.setEmail("john.doe@example.com");

        // 插入數(shù)據(jù)
        int result = userMapper.insertUser(user);

        // 檢查插入結(jié)果
        assertEquals(1, result);

        // 獲取自增長(zhǎng)主鍵
        assertNotNull(user.getId());
        System.out.println("Generated ID: " + user.getId());
    }
}

深度解析

定義實(shí)體類：創(chuàng)建一個(gè) User 類，包含 id、name 和 email 屬性。
創(chuàng)建 Mapper 接口：定義一個(gè) UserMapper 接口，包含一個(gè) insertUser 方法。
編寫 Mapper XML 文件：在 UserMapper.xml 文件中，使用 <insert> 標(biāo)簽定義插入語(yǔ)句，并設(shè)置 useGeneratedKeys="true" 和 keyProperty="id"。
測(cè)試代碼：編寫測(cè)試代碼，創(chuàng)建一個(gè) User 對(duì)象，調(diào)用 insertUser 方法插入數(shù)據(jù)，并檢查插入結(jié)果和生成的主鍵值。

第二步：使用 selectKey 獲取自增長(zhǎng)主鍵

概述

除了 useGeneratedKeys，MyBatis 還提供了一個(gè) <selectKey> 標(biāo)簽，可以在插入數(shù)據(jù)后獲取自增長(zhǎng)主鍵。這種方法適用于更復(fù)雜的場(chǎng)景，如需要在插入前生成主鍵值。

代碼示例

假設(shè)我們有一個(gè) Order 表，其結(jié)構(gòu)如下：

CREATE TABLE `order` (
  `id` INT(11) NOT NULL AUTO_INCREMENT,
  `user_id` INT(11) NOT NULL,
  `amount` DECIMAL(10, 2) NOT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

對(duì)應(yīng)的 Order 實(shí)體類如下：

public class Order {
    private Integer id;
    private Integer userId;
    private BigDecimal amount;

    // Getters and Setters
}

Mapper 接口

public interface OrderMapper {
    int insertOrder(Order order);
}

Mapper XML 文件

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.example.mapper.OrderMapper">

    <!-- 使用 selectKey 獲取自增長(zhǎng)主鍵 -->
    <insert id="insertOrder" parameterType="com.example.entity.Order">
        <selectKey keyProperty="id" resultType="int" order="AFTER">
            SELECT LAST_INSERT_ID()
        </selectKey>
        INSERT INTO `order` (user_id, amount) VALUES (#{userId}, #{amount})
    </insert>

</mapper>

測(cè)試代碼

@SpringBootTest
public class MyBatisTest {

    @Autowired
    private OrderMapper orderMapper;

    @Test
    public void testInsertOrder() {
        Order order = new Order();
        order.setUserId(1);
        order.setAmount(new BigDecimal("100.00"));

        // 插入數(shù)據(jù)
        int result = orderMapper.insertOrder(order);

        // 檢查插入結(jié)果
        assertEquals(1, result);

        // 獲取自增長(zhǎng)主鍵
        assertNotNull(order.getId());
        System.out.println("Generated ID: " + order.getId());
    }
}

深度解析

定義實(shí)體類：創(chuàng)建一個(gè) Order 類，包含 id、userId 和 amount 屬性。
創(chuàng)建 Mapper 接口：定義一個(gè) OrderMapper 接口，包含一個(gè) insertOrder 方法。
編寫 Mapper XML 文件：在 OrderMapper.xml 文件中，使用 <insert> 標(biāo)簽定義插入語(yǔ)句，并嵌套 <selectKey> 標(biāo)簽，設(shè)置 keyProperty="id" 和 resultType="int"，并在 <selectKey> 標(biāo)簽中編寫 SQL 語(yǔ)句 SELECT LAST_INSERT_ID()。
測(cè)試代碼：編寫測(cè)試代碼，創(chuàng)建一個(gè) Order 對(duì)象，調(diào)用 insertOrder 方法插入數(shù)據(jù)，并檢查插入結(jié)果和生成的主鍵值。

第三步：注意事項(xiàng)與常見面試題

注意事項(xiàng)

數(shù)據(jù)庫(kù)支持：確保你的數(shù)據(jù)庫(kù)支持自增長(zhǎng)主鍵，如 MySQL、PostgreSQL 等。
配置正確：確保 useGeneratedKeys 和 keyProperty 配置正確，否則無(wú)法獲取自增長(zhǎng)主鍵。
事務(wù)管理：在插入數(shù)據(jù)時(shí)，確保事務(wù)管理得當(dāng)，避免數(shù)據(jù)不一致問題。
并發(fā)處理：在高并發(fā)場(chǎng)景下，注意 LAST_INSERT_ID() 的線程安全性。

常見面試題

什么是 useGeneratedKeys？

useGeneratedKeys 是 MyBatis 中的一個(gè)屬性，用于在插入數(shù)據(jù)時(shí)獲取數(shù)據(jù)庫(kù)自動(dòng)生成的主鍵值。

如何使用 selectKey 獲取自增長(zhǎng)主鍵？

在 <insert> 標(biāo)簽中嵌套 <selectKey> 標(biāo)簽，設(shè)置 keyProperty 和 resultType，并在 <selectKey> 標(biāo)簽中編寫 SQL 語(yǔ)句 SELECT LAST_INSERT_ID()。

useGeneratedKeys 和 selectKey 的區(qū)別是什么？