SpringBoot集成yitter-idgenerator(雪花漂移)分布式Id自增的实现
作者:a_a\\\\\\ 时间:2022-03-27 10:47:08
场景
yitter-idgenerator 是基于雪花算法进行改造的分布式ID自增算法,集成时需要为每个服务设置唯一的机器号,才能保证生成的Id不会重复
实现方案
基于服务启动时指定唯一机器号
在程序服务启动时通过分布式锁 Redisson(基于Redis实现),对每台机器通过IP 对应一个 唯一的机器号(自增)映射,并保存在Redis中。缓存一次后,下次启动直接读取缓存即可
基于注册中心指定唯一机器号
从注册中心读取服务,增加多一个机器号服务来统一分配
基于第一种实现方案
Maven依赖
<dependency>
<groupId>org.redisson</groupId>
<artifactId>redisson-spring-boot-starter</artifactId>
<version>3.10.6</version>
</dependency>
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-core</artifactId>
<version>5.7.19</version>
</dependency>
<dependency>
<groupId>com.github.yitter</groupId>
<artifactId>yitter-idgenerator</artifactId>
<version>1.0.6</version>
</dependency>
关键部分代码
/**
* Redisson分布式锁工具类
*/
@Component
public class RedissonUtil {
@Autowired
private RedissonClient redissonClient;
/**
* 加锁
* @param lockKey
* @return
*/
public RLock lock(String lockKey) {
RLock lock = redissonClient.getLock(lockKey);
lock.lock();
return lock;
}
/**
* 带超时的锁
* @param lockKey
* @param timeout 超时时间 单位:秒
*/
public RLock lock(String lockKey, long timeout) {
RLock lock = redissonClient.getLock(lockKey);
lock.lock(timeout, TimeUnit.SECONDS);
return lock;
}
/**
* 带超时的锁
* @param lockKey
* @param unit 时间单位
* @param timeout 超时时间
*/
public RLock lock(String lockKey, TimeUnit unit ,long timeout) {
RLock lock = redissonClient.getLock(lockKey);
lock.lock(timeout, unit);
return lock;
}
/**
* 尝试获取锁
* @param lockKey
* @param waitTime 最多等待时间
* @param unit TimeUnit时间单位
* @return
*/
public boolean tryLock(String lockKey,long waitTime, TimeUnit unit) {
RLock lock = redissonClient.getLock(lockKey);
try {
return lock.tryLock(waitTime, unit);
} catch (InterruptedException e) {
return false;
}
}
/**
* 尝试获取锁
* @param lockKey
* @param waitTime 最多等待时间
* @param leaseTime 上锁后自动释放锁时间
* @return
*/
public boolean tryLock(String lockKey, long waitTime, long leaseTime) {
RLock lock = redissonClient.getLock(lockKey);
try {
return lock.tryLock(waitTime, leaseTime, TimeUnit.SECONDS);
} catch (InterruptedException e) {
return false;
}
}
/**
* 尝试获取锁
* @param lockKey
* @param unit 时间单位
* @param waitTime 最多等待时间
* @param leaseTime 上锁后自动释放锁时间
* @return
*/
public boolean tryLock(String lockKey, TimeUnit unit, long waitTime, long leaseTime) {
RLock lock = redissonClient.getLock(lockKey);
try {
return lock.tryLock(waitTime, leaseTime, unit);
} catch (InterruptedException e) {
return false;
}
}
/**
* 释放锁
* @param lockKey
*/
public void unlock(String lockKey) {
RLock lock = redissonClient.getLock(lockKey);
lock.unlock();
}
/**
* 若没用锁情况下,就不调用释放锁的代码,若有锁情况下才调用释放锁
* @param lockKey
*/
public void unlockIgnore(String lockKey) {
RLock lock = redissonClient.getLock(lockKey);
if ( !lock.isLocked() ) {
return ;
}
lock.unlock();
}
/**
* 释放锁
* @param lock
*/
public void unlock(RLock lock) {
lock.unlock();
}
}
启动配置代码如下
@Slf4j
@Component
@Order(0)
public class SystemInitConfig implements CommandLineRunner {
@Autowired
private RedissonUtil redissonUtil;
@Autowired
private RedisTemplate redisTemplate;
/**
* 分布式锁Key
*/
private static final String CACHE_ID_GENERATOR = "LOCK_ID_GENERATOR";
/**
* 最大机器号Key
*/
private static final String CACHE_WORKERID_MAXID= "CACHE_WORKERID_MAXID";
/**
* 已分配的机器号Key
*/
private static final String CACHE_ID_IP = "CACHE_ID_IP";
@Override
public void run(String... args) throws Exception {
//获取mac地址
String macAddress = NetUtil.getLocalhost().getHostAddress();
log.info("{} 配置分布式Id Work缓存========开始",macAddress);
boolean existWorkerId = redisTemplate.opsForHash().hasKey(CACHE_ID_IP, macAddress);
//若已缓存在缓存中,直接跳过不设置
if (existWorkerId) {
log.info("{} 已配置分布式Id Work...",macAddress);
return ;
}
try {
//分布式锁等待120秒,执行时长最大120秒
boolean locked = redissonUtil.tryLock(CACHE_ID_GENERATOR, 120, 120);
if (!locked) {
throw new RuntimeException(macAddress+"设置分布式Id机器号失败");
}
ValueOperations <String,Integer> stringOperation = redisTemplate.opsForValue();
boolean initWorkerId = stringOperation.setIfAbsent(CACHE_WORKERID_MAXID, 1);
if( !initWorkerId ) {
//若已存在key,对最大的机器号自增1
stringOperation.increment(CACHE_WORKERID_MAXID);
}
Integer workerId = stringOperation.get(CACHE_WORKERID_MAXID);
IdGeneratorOptions options = new IdGeneratorOptions( workerId.shortValue());
YitIdHelper.setIdGenerator(options);
//设置mac地址 - workerid 到hash结构
redisTemplate.opsForHash().put(CACHE_ID_IP,macAddress,workerId);
log.info("已配置分布式Id Work,{} - {}",macAddress,workerId);
} finally {
redissonUtil.unlock(CACHE_ID_GENERATOR);
log.info("{} 配置分布式Id Work缓存========结束",macAddress);
}
}
}
直接在代码使用即可
YitIdHelper.nextId()
来源:https://blog.csdn.net/jie873440996/article/details/122459656
标签:SpringBoot,分布式Id自增
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
亲手教你SpringBoot中的多数据源集成问题
2023-08-19 02:57:20
C#多线程处理多个队列数据的方法
2021-12-26 19:31:02
Java Springboot的目的你知道吗
2022-12-20 14:16:24
带你了解Java数据结构和算法之数组
2022-09-29 21:18:33
![](https://img.aspxhome.com/file/2023/7/83087_0s.png)
关于SpringMVC在Controller层方法的参数解析详解
2022-10-29 01:06:10
JAVA使用JDBC连接oracle数据库的详细过程
2021-11-20 19:20:19
![](https://img.aspxhome.com/file/2023/4/63414_0s.png)
详解IDEA多module项目maven依赖的一些说明
2023-07-21 06:09:05
![](https://img.aspxhome.com/file/2023/4/67974_0s.png)
C#中const用法详解
2021-08-12 15:30:29
Java数据机构中关于并查集的详解
2022-09-05 13:23:53
![](https://img.aspxhome.com/file/2023/2/82982_0s.png)
Android实现语音播放与录音功能
2022-01-21 15:39:09
什么是递归?用Java写一个简单的递归程序
2022-02-11 19:39:45
![](https://img.aspxhome.com/file/2023/2/60772_0s.jpg)
struts2简介_动力节点Java学院整理
2023-06-24 02:04:07
![](https://img.aspxhome.com/file/2023/3/72323_0s.jpg)
简单了解Java字符串(操作)
2022-04-22 18:37:29
HashMap 和 Hashtable的区别
2022-11-01 06:34:47
C#开发微信门户及应用(5) 用户分组信息管理
2022-08-03 04:52:20
IDEA中Spring项目的工程构建
2023-07-20 13:36:48
![](https://img.aspxhome.com/file/2023/6/72536_0s.jpg)
C#文件下载实例代码(适用于各个浏览器)
2022-12-28 06:25:48
C# Cache缓存读取的设置方法
2022-11-18 05:33:53
![](https://img.aspxhome.com/file/2023/3/81143_0s.png)
Java四位电话号码的加密方法
2022-08-20 16:04:29
![](https://img.aspxhome.com/file/2023/9/89629_0s.jpg)
Java服务限流算法的6种实现
2022-04-03 04:52:51
![](https://img.aspxhome.com/file/2023/3/62993_0s.jpg)