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

本篇內(nèi)容介紹了“mybatis-plus怎么使用雪花算法ID生成策略”的有關(guān)知識(shí)，在實(shí)際案例的操作過(guò)程中，不少人都會(huì)遇到這樣的困境，接下來(lái)就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

mybatis-plus 可以通過(guò)@TableId注解指定主鍵生成策略

@TableId(value="id",type=IdType.ASSIGN_ID)

上面的注解設(shè)置了值，對(duì)應(yīng)在MybatisDefaultParameterHandler中有判斷這個(gè)類(lèi)型和設(shè)置ID的處理，判斷idType.getKey() == IdType.ASSIGN_ID.getKey()時(shí)，使用identifierGenerator.nextId(entity)創(chuàng)建了一個(gè)雪花算法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版本后默認(rèn)就是使用 IdType.ASSIGN_ID，即雪花算法，如果需要使用主鍵自增，則需要修改IdType.AUTO

看一下源碼大概找一下雪花算法的實(shí)現(xiàn)方式

首先來(lái)看主鍵生成策略的接口是IdentifierGenerator。mp中默認(rèn)有一個(gè)實(shí)現(xiàn)類(lèi)就是DefaultIdentifierGenerator，這個(gè)就是雪花算法的實(shí)現(xiàn)類(lèi)。這個(gè)類(lèi)中使用的Sequence就是雪花算法的實(shí)現(xiàn)類(lèi)了。

如果想自己創(chuàng)建雪花算法手動(dòng)創(chuàng)建ID怎么辦

com.baomidou.mybatisplus.core.toolkit包下有一個(gè)IdWork的類(lèi)，這個(gè)類(lèi)就是使用上面的DefaultIdentifierGenerator來(lái)獲取雪花算法ID，我們可以直接使用IdWork同名方法getId或者getIdStr來(lái)手動(dòng)獲取一個(gè)雪花算法ID。

Long id=IdWorker.getIdStr()

自定義ID生成器實(shí)現(xiàn)

第一種，使用spring掃描注解創(chuàng)建bean使用

@Component
public class CustomIdGenerator implements IdentifierGenerator {
    @Override
    public Long nextId(Object entity) {
       //可以將當(dāng)前傳入的class全類(lèi)名來(lái)作為bizKey,或者提取參數(shù)來(lái)生成bizKey進(jìn)行分布式Id調(diào)用生成.
       String bizKey = entity.getClass().getName();
        //根據(jù)bizKey調(diào)用分布式ID生成
        long id = ....;
       //返回生成的id值即可.
        return id;
    }
}

第二種，直接把寫(xiě)好的CustomIdGenerator實(shí)現(xiàn)類(lèi)配置為一個(gè)bean

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

Mybatis-plus 中生成雪花算法id的工具類(lèi)

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

“mybatis-plus怎么使用雪花算法ID生成策略”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！