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

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

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

@TableId(value="id",type=IdType.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));                }            }        }     }

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

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

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

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

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

Long id=IdWorker.getIdStr()

自定义ID生成器实现

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

@Componentpublic 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

@Beanpublic 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生成策略”的内容就介绍到这里了，感谢大家的阅读。如果想了解更多行业相关的知识可以关注编程网网站，小编将为大家输出更多高质量的实用文章！