SpringBoot集成yitter-idgenerator（雪花漂移）分布式Id⾃
增的实现
⽬录
场景
实现⽅案
基于第⼀种实现⽅案
关键部分代码
场景
是基于雪花算法进⾏改造的分布式ID⾃增算法，集成时需要为每个服务设置唯⼀的机器号，才能保证⽣成的Id不会重复
实现⽅案
基于服务启动时指定唯⼀机器号
在程序服务启动时通过分布式锁 Redisson（基于Redis实现），对每台机器通过IP 对应⼀个唯⼀的机器号（⾃增）映射，并保存在Redis中。缓存⼀次后，下次启动直接读取缓存即可
基于注册中⼼指定唯⼀机器号
从注册中⼼读取服务，增加多⼀个机器号服务来统⼀分配
基于第⼀种实现⽅案
Maven依赖
<dependency>
<groupId&disson</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 = Lock(lockKey);
lock.lock();
return lock;
}
/**
* 带超时的锁
* @param lockKey
* @param timeout 超时时间单位：秒
*/
public RLock lock(String lockKey, long timeout) {
RLock lock = Lock(lockKey);
lock.lock(timeout, TimeUnit.SECONDS);
return lock;
}
/**
* 带超时的锁
* @param lockKey
* @param unit 时间单位
springboot结构* @param timeout 超时时间
*/
public RLock lock(String lockKey, TimeUnit unit ,long timeout) {
RLock lock = Lock(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 = Lock(lockKey);
try {
Lock(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 = Lock(lockKey);
try {
Lock(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 = Lock(lockKey);
try {
Lock(waitTime, leaseTime, unit);
} catch (InterruptedException e) {
return false;
}
}
/**
* 释放锁
* @param lockKey
*/
public void unlock(String lockKey) {
RLock lock = Lock(lockKey);
lock.unlock();
}
/**
* 若没⽤锁情况下，就不调⽤释放锁的代码，若有锁情况下才调⽤释放锁
* @param lockKey
*/
public void unlockIgnore(String lockKey) {
RLock lock = Lock(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 args) throws Exception {
//获取mac地址
String macAddress = Localhost().getHostAddress();
log.info("{} 配置分布式Id Work缓存========开始",macAddress);
boolean existWorkerId =  redisTemplate.opsForHash().hasKey(CACHE_ID_IP, macAddress);        //若已缓存在缓存中，直接跳过不设置
if (existWorkerId) {
log.info("{} 已配置分布式",macAddress);
return ;
}
try {
//分布式锁等待120秒，执⾏时长最⼤120秒
boolean  locked = Lock(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 =  (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);
}
}
}
直接在代码使⽤即可
到此这篇关于SpringBoot集成yitter-idgenerator(雪花漂移)分布式Id⾃增的实现的⽂章就介绍到这了，更多相关SpringBoot 分布式Id⾃增内容请搜索以前的⽂章或继续浏览下⾯的相关⽂章希望⼤家以后多多⽀持！

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。