MyBatis-Plus 雪花算法生成 ID 怎么使用

1. 什么是雪花算法？

雪花算法（Snowflake Algorithm）是 Twitter 开源的分布式 ID 生成算法。它可以在分布式系统中生成全局唯一且有序的 ID。雪花算法生成的 ID 是一个 64 位的长整型数字，结构如下：

• 1 位：符号位，固定为 0，表示正数。
• 41 位：时间戳，表示从某个起始时间到当前时间的毫秒数。
• 10 位：机器 ID，表示生成 ID 的机器或节点。
• 12 位：序列号，表示同一毫秒内生成的 ID 序号。

通过这种结构，雪花算法可以在分布式系统中生成全局唯一且有序的 ID。

2. MyBatis-Plus 中的雪花算法

MyBatis-Plus 是一个 MyBatis 的增强工具，它提供了许多便捷的功能，其中包括自动生成主键 ID 的功能。MyBatis-Plus 默认使用雪花算法来生成主键 ID。

2.1 配置雪花算法

在 MyBatis-Plus 中，默认情况下已经启用了雪花算法生成 ID。如果你需要自定义雪花算法的配置，可以通过以下方式进行配置：

mybatis-plus:
  global-config:
    db-config:
      id-type: assign_id  # 使用雪花算法生成 ID
      worker-id: 1       # 机器 ID，默认为 1
      datacenter-id: 1    # 数据中心 ID，默认为 1

• id-type: 指定 ID 生成策略，assign_id 表示使用雪花算法生成 ID。
• worker-id: 机器 ID，用于区分不同的机器或节点。
• datacenter-id: 数据中心 ID，用于区分不同的数据中心。

2.2 使用雪花算法生成 ID

在实体类中，你可以通过 @TableId 注解来指定主键字段，并设置 type 为 IdType.ASSIGN_ID，这样 MyBatis-Plus 就会自动使用雪花算法生成 ID。

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;

@TableName("user")
public class User {
    @TableId(type = IdType.ASSIGN_ID)
    private Long id;
    private String name;
    private Integer age;

    // getter and setter
}

在上面的代码中，id 字段被标记为主键，并且使用雪花算法生成 ID。当你插入一条新的记录时，MyBatis-Plus 会自动为 id 字段生成一个全局唯一的 ID。

2.3 手动生成 ID

如果你需要手动生成雪花算法的 ID，可以使用 IdWorker 工具类：

import com.baomidou.mybatisplus.core.toolkit.IdWorker;

public class SnowflakeIdGenerator {
    public static void main(String[] args) {
        long id = IdWorker.getId();
        System.out.println("Generated ID: " + id);
    }
}

IdWorker.getId() 方法会返回一个使用雪花算法生成的全局唯一 ID。

3. 总结

MyBatis-Plus 提供了便捷的雪花算法生成 ID 的功能，通过简单的配置和注解，你可以轻松地在分布式系统中生成全局唯一且有序的 ID。无论是自动生成还是手动生成，MyBatis-Plus 都能满足你的需求。

通过本文的介绍，你应该已经掌握了如何在 MyBatis-Plus 中使用雪花算法生成 ID。希望这对你在实际开发中有所帮助！