SpringBoot集成yitter

SpringBoot集成yitter

目录场景实现方案基于第一种实现方案关键部分代码

场景

yitter-idgenerator 是基于雪花算法进行改造的分布式ID自增算法，集成时需要为每个服务设置唯一的机器号，才能保证生成的Id不会重复

实现方案

基于服务启动时指定唯一机器号

在程序服务启动时通过分布式锁 Redisson（基于Redis实现），对每台机器通过IP 对应一个 唯一的机器号（自增）映射，并保存在Redis中。缓存一次后，下次启动直接读取缓存即可

基于注册中心指定唯一机器号

从注册中心读取服务，增加多一个机器号服务来统一分配

基于第一种实现方案

Maven依赖

org.redisson

redisson-spring-boot-starter

3.10.6

cn.hutool

hutool-core

5.7.19

com.github.yitter

yitter-idgenerator

1.0.6

关键部分代码

/**

* 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) {http://

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 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()

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。