mybatis-plus?如何使用雪花算法ID生成策略
mybatis-plus 可以通過(guò)@TableId注解指定主鍵生成策略
@TableId(value="id",type=IdType.ASSIGN_ID)
| 值 | 描述 |
|---|---|
| AUTO | 數(shù)據(jù)庫(kù)ID自增 |
| NONE | 無(wú)狀態(tài),該類型為未設(shè)置主鍵類型(注解里等于跟隨全局,全局里約等于 INPUT) |
| INPUT | insert前自行set主鍵值 |
| ASSIGN_ID | 分配ID(主鍵類型為Number(Long和Integer)或String)(since 3.3.0),使用接口IdentifierGenerator的方法nextId(默認(rèn)實(shí)現(xiàn)類為DefaultIdentifierGenerator雪花算法) |
| ASSIGN_UUID | 分配UUID,主鍵類型為String(since 3.3.0),使用接口IdentifierGenerator的方法nextUUID(默認(rèn)default方法) |
| 分布式全局唯一ID 長(zhǎng)整型類型(please use ASSIGN_ID) | |
| 32位UUID字符串(please use ASSIGN_UUID) | |
| 分布式全局唯一ID 字符串類型(please use ASSIGN_ID) |
上面的注解設(shè)置了值,對(duì)應(yīng)在MybatisDefaultParameterHandler中有判斷這個(gè)類型和設(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)類就是DefaultIdentifierGenerator,這個(gè)就是雪花算法的實(shí)現(xiàn)類。這個(gè)類中使用的Sequence就是雪花算法的實(shí)現(xiàn)類了。



如果想自己創(chuàng)建雪花算法手動(dòng)創(chuàng)建ID怎么辦
com.baomidou.mybatisplus.core.toolkit包下有一個(gè)IdWork的類,這個(gè)類就是使用上面的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)作為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;
}
}
第二種,直接把寫好的CustomIdGenerator實(shí)現(xiàn)類配置為一個(gè)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());
}
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Mybatis Interceptor 攔截器的實(shí)現(xiàn)
這篇文章主要介紹了Mybatis Interceptor 攔截器的實(shí)現(xiàn),小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
Java split函數(shù)拆分后變成null問(wèn)題解決方案
這篇文章主要介紹了Java split函數(shù)拆分后變成null問(wèn)題解決方案,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-10-10
Java編程實(shí)現(xiàn)基于TCP協(xié)議的Socket聊天室示例
這篇文章主要介紹了Java編程實(shí)現(xiàn)基于TCP協(xié)議的Socket聊天室,結(jié)合實(shí)例形式詳細(xì)分析了java基于TCP協(xié)議的Socket聊天室客戶端與服務(wù)器端相關(guān)實(shí)現(xiàn)與使用技巧,需要的朋友可以參考下2018-01-01
Springboot實(shí)例講解實(shí)現(xiàn)專業(yè)材料認(rèn)證管理系統(tǒng)流程
這是一個(gè)基于java的畢業(yè)設(shè)計(jì)項(xiàng)目,畢設(shè)課題為springboot框架的知識(shí)產(chǎn)權(quán)服務(wù)平臺(tái)系統(tǒng),是一個(gè)采用b/s結(jié)構(gòu)的javaweb項(xiàng)目,需要的朋友可以參考下2022-06-06
Java 使用getClass().getResourceAsStream()方法獲取資源
這篇文章主要介紹了Java 使用getClass().getResourceAsStream()方法獲取資源的相關(guān)資料,這里主要講解哪種方式可以獲取到文件資源,需要的朋友可以參考下2017-07-07
Java使用HttpClient實(shí)現(xiàn)Post請(qǐng)求實(shí)例
本篇文章主要介紹了Java使用HttpClient實(shí)現(xiàn)Post請(qǐng)求實(shí)例,小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-02-02
使用JPA進(jìn)行CriteriaQuery進(jìn)行查詢的注意事項(xiàng)
這篇文章主要介紹了使用JPA進(jìn)行CriteriaQuery進(jìn)行查詢的注意事項(xiàng),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2021-12-12

