mybatis-plus怎么使用雪花算法ID生成策略

发布时间:2021-12-03 16:25:41 作者:iii
来源:亿速云 阅读:3632

本篇内容介绍了“mybatis-plus怎么使用雪花算法ID生成策略”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

mybatis-plus 可以通过@TableId注解指定主键生成策略

@TableId(value="id",type=IdType.ASSIGN_ID)
描述
AUTO数据库ID自增
NONE无状态,该类型为未设置主键类型(注解里等于跟随全局,全局里约等于 INPUT)
INPUTinsert前自行set主键值
ASSIGN_ID分配ID(主键类型为Number(Long和Integer)或String)(since 3.3.0),使用接口IdentifierGenerator的方法nextId(默认实现类为DefaultIdentifierGenerator雪花算法)
ASSIGN_UUID分配UUID,主键类型为String(since 3.3.0),使用接口IdentifierGenerator的方法nextUUID(默认default方法)
ID_WORKER分布式全局唯一ID 长整型类型(please use ASSIGN_ID)
UUID32位UUID字符串(please use ASSIGN_UUID)
ID_WORKER_STR分布式全局唯一ID 字符串类型(please use ASSIGN_ID)

上面的注解设置了值,对应在MybatisDefaultParameterHandler中有判断这个类型和设置ID的处理,判断idType.getKey() == IdType.ASSIGN_ID.getKey()时,使用identifierGenerator.nextId(entity)创建了一个雪花算法ID。

protected static void populateKeys(TableInfo tableInfo, MetaObject metaObject, Object entity) {
        IdType idType = tableInfo.getIdType();
        String keyProperty = tableInfo.getKeyProperty();
        if (StringUtils.isNotBlank(keyProperty) && null != idType && idType.getKey() >= 3) {
            IdentifierGenerator identifierGenerator = GlobalConfigUtils.getGlobalConfig(tableInfo.getConfiguration()).getIdentifierGenerator();
            Object idValue = metaObject.getValue(keyProperty);
            if (StringUtils.checkValNull(idValue)) {
                if (idType.getKey() == IdType.ASSIGN_ID.getKey()) {
                    if (Number.class.isAssignableFrom(tableInfo.getKeyType())) {
                        metaObject.setValue(keyProperty, identifierGenerator.nextId(entity));
                    } else {
                        metaObject.setValue(keyProperty, identifierGenerator.nextId(entity).toString());
                    }
                } else if (idType.getKey() == IdType.ASSIGN_UUID.getKey()) {
                    metaObject.setValue(keyProperty, identifierGenerator.nextUUID(entity));
                }
            }
        } 
    }

3.X版本后默认就是使用 IdType.ASSIGN_ID,即雪花算法,如果需要使用主键自增,则需要修改IdType.AUTO

看一下源码大概找一下雪花算法的实现方式

首先来看主键生成策略的接口是IdentifierGenerator。mp中默认有一个实现类就是DefaultIdentifierGenerator,这个就是雪花算法的实现类。这个类中使用的Sequence就是雪花算法的实现类了。

mybatis-plus怎么使用雪花算法ID生成策略

mybatis-plus怎么使用雪花算法ID生成策略

mybatis-plus怎么使用雪花算法ID生成策略

如果想自己创建雪花算法手动创建ID怎么办

com.baomidou.mybatisplus.core.toolkit包下有一个IdWork的类,这个类就是使用上面的DefaultIdentifierGenerator来获取雪花算法ID,我们可以直接使用IdWork同名方法getId或者getIdStr来手动获取一个雪花算法ID。

Long id=IdWorker.getIdStr()

自定义ID生成器实现

第一种,使用spring扫描注解创建bean使用

@Component
public class CustomIdGenerator implements IdentifierGenerator {
    @Override
    public Long nextId(Object entity) {
       //可以将当前传入的class全类名来作为bizKey,或者提取参数来生成bizKey进行分布式Id调用生成.
       String bizKey = entity.getClass().getName();
        //根据bizKey调用分布式ID生成
        long id = ....;
       //返回生成的id值即可.
        return id;
    }
}

第二种,直接把写好的CustomIdGenerator实现类配置为一个bean

@Bean
public IdentifierGenerator idGenerator() {
    return new CustomIdGenerator();
}

Mybatis-plus 中生成雪花算法id的工具类

public static void main(String[] args) {
    // 返回值  1385106677482582018
    System.out.println(IdWorker.getId());
    // 返回值 "1385106677482582019"
    System.out.println(IdWorker.getIdStr());
}

“mybatis-plus怎么使用雪花算法ID生成策略”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站,小编将为大家输出更多高质量的实用文章!

推荐阅读:
  1. 雪花算法(03)生成时间
  2. 雪花算法(01)介绍

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

mybatis-plus

上一篇:怎么用Python模拟谷歌的小恐龙游戏

下一篇:HashMap和Hashtable的区别是什么

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》