MyBatisPlus之id生成策略怎么使用

MyBatis-Plus（简称MP）是MyBatis的增强工具，在MyBatis的基础上只做增强不做改变，简化开发、提高效率。在实际开发中，数据库表的主键生成策略是一个非常重要的部分。MyBatis-Plus提供了多种主键生成策略，本文将详细介绍如何使用这些策略。

1. 主键生成策略概述

在MyBatis-Plus中，主键生成策略主要通过@TableId注解来指定。@TableId注解有两个重要的属性：

• value：用于指定数据库表中的主键字段名，如果字段名与实体类属性名一致，可以省略。
• type：用于指定主键生成策略，MyBatis-Plus提供了多种内置的主键生成策略。

2. 内置主键生成策略

MyBatis-Plus内置了以下几种主键生成策略：

• IdType.AUTO：数据库ID自增
• IdType.NONE：未设置主键类型（默认）
• IdType.INPUT：用户输入ID
• IdType.ASSIGN_ID：分配ID（主键类型为Number或String）
• IdType.ASSIGN_UUID：分配UUID（主键类型为String）
• IdType.ID_WORKER：分布式全局唯一ID（Long类型）
• IdType.UUID：32位UUID字符串
• IdType.ID_WORKER_STR：分布式全局唯一ID（String类型）

下面我们将逐一介绍这些策略的使用方法。

2.1 IdType.AUTO

IdType.AUTO表示数据库ID自增，适用于MySQL等支持自增主键的数据库。使用该策略时，数据库表的主键字段需要设置为自增。

@TableId(type = IdType.AUTO)
private Long id;

2.2 IdType.NONE

IdType.NONE表示未设置主键类型，默认情况下MyBatis-Plus会使用该策略。此时，主键的生成需要由用户自行处理。

@TableId(type = IdType.NONE)
private Long id;

2.3 IdType.INPUT

IdType.INPUT表示用户输入ID，适用于用户自定义主键的场景。使用该策略时，用户需要在插入数据时手动设置主键值。

@TableId(type = IdType.INPUT)
private Long id;

2.4 IdType.ASSIGN_ID

IdType.ASSIGN_ID表示分配ID，适用于主键类型为Number或String的场景。MyBatis-Plus会自动生成一个全局唯一的ID。

@TableId(type = IdType.ASSIGN_ID)
private Long id;

2.5 IdType.ASSIGN_UUID

IdType.ASSIGN_UUID表示分配UUID，适用于主键类型为String的场景。MyBatis-Plus会自动生成一个32位的UUID字符串。

@TableId(type = IdType.ASSIGN_UUID)
private String id;

2.6 IdType.ID_WORKER

IdType.ID_WORKER表示分布式全局唯一ID，适用于主键类型为Long的场景。MyBatis-Plus会使用雪花算法生成一个全局唯一的ID。

@TableId(type = IdType.ID_WORKER)
private Long id;

2.7 IdType.UUID

IdType.UUID表示32位UUID字符串，适用于主键类型为String的场景。MyBatis-Plus会自动生成一个32位的UUID字符串。

@TableId(type = IdType.UUID)
private String id;

2.8 IdType.ID_WORKER_STR

IdType.ID_WORKER_STR表示分布式全局唯一ID，适用于主键类型为String的场景。MyBatis-Plus会使用雪花算法生成一个全局唯一的ID，并将其转换为字符串。

@TableId(type = IdType.ID_WORKER_STR)
private String id;

3. 自定义主键生成策略

除了使用MyBatis-Plus内置的主键生成策略外，我们还可以自定义主键生成策略。自定义主键生成策略需要实现IdentifierGenerator接口，并在@TableId注解中指定自定义的生成器。

3.1 实现IdentifierGenerator接口

首先，我们需要实现IdentifierGenerator接口，并重写nextId方法。

import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
import org.springframework.stereotype.Component;

@Component
public class CustomIdGenerator implements IdentifierGenerator {

    @Override
    public Number nextId(Object entity) {
        // 自定义ID生成逻辑
        return System.currentTimeMillis();
    }
}

3.2 在@TableId注解中指定自定义生成器

然后，在实体类中使用@TableId注解，并指定type为IdType.ASSIGN_ID，同时指定generator为自定义的生成器。

@TableId(type = IdType.ASSIGN_ID, generator = "customIdGenerator")
private Long id;

4. 总结

MyBatis-Plus提供了多种主键生成策略，可以满足不同场景下的需求。通过@TableId注解，我们可以轻松地指定主键生成策略。此外，MyBatis-Plus还支持自定义主键生成策略，为开发者提供了更大的灵活性。

在实际开发中，选择合适的主键生成策略非常重要。对于单机应用，可以使用IdType.AUTO或IdType.INPUT；对于分布式系统，推荐使用IdType.ASSIGN_ID或IdType.ID_WORKER。希望本文能帮助你更好地理解和使用MyBatis-Plus的主键生成策略。