opensource/RuoYi-Vue-Plus
2
0
Fork 1
mirror of https://gitee.com/dromara/RuoYi-Vue-Plus.git synced 2026-09-19 01:48:47 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Pre Merge pull request !718 from AprilWind/dev

Browse Source
This commit is contained in:
AprilWind 2025-07-04 08:55:28 +00:00 committed by Gitee
commit 1cee8d8280
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 36 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
ruoyi-common/ruoyi-common-redis/src/main/java/org/dromara/common/redis/utils/SequenceUtils.java
View File

@ -1,9 +1,9 @@
package org.dromara.common.redis.utils; package org.dromara.common.redis.utils;
import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateUtil;
import lombok.AccessLevel; import lombok.AccessLevel;
import lombok.NoArgsConstructor; import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.dromara.common.core.utils.DateUtils;
import org.dromara.common.core.utils.SpringUtils; import org.dromara.common.core.utils.SpringUtils;
import org.dromara.common.core.utils.StringUtils; import org.dromara.common.core.utils.StringUtils;
import org.redisson.api.RIdGenerator; import org.redisson.api.RIdGenerator;
@ -17,6 +17,7 @@ import java.time.Duration;
* @author 秋辞未寒 * @author 秋辞未寒
* @date 2024-12-10 * @date 2024-12-10
*/ */
@Slf4j
@NoArgsConstructor(access = AccessLevel.PRIVATE) @NoArgsConstructor(access = AccessLevel.PRIVATE)
public class SequenceUtils { public class SequenceUtils {
@ -33,12 +34,17 @@ public class SequenceUtils {
/** /**
* 默认过期时间-天 * 默认过期时间-天
*/ */
public static final Duration DEFAULT_EXPIRE_TIME_DAY = Duration.ofDays(1); public static final Duration DEFAULT_EXPIRE_DAY = Duration.ofDays(1);
/** /**
* 默认过期时间-分钟 * 默认过期时间-分钟
*/ */
public static final Duration DEFAULT_EXPIRE_TIME_MINUTE = Duration.ofMinutes(1); public static final Duration DEFAULT_EXPIRE_MINUTE = Duration.ofMinutes(1);
/**
* 默认前缀
*/
public static final String DEFAULT_PREFIX = "SEQ:";
/** /**
* 获取Redisson客户端实例 * 获取Redisson客户端实例
@ -46,7 +52,7 @@ public class SequenceUtils {
private static final RedissonClient REDISSON_CLIENT = SpringUtils.getBean(RedissonClient.class); private static final RedissonClient REDISSON_CLIENT = SpringUtils.getBean(RedissonClient.class);
/** /**
* 获取ID生成器 * 获取 Redisson ID生成器（仅初始化一次）
* *
* @param key 业务key * @param key 业务key
* @param expireTime 过期时间 * @param expireTime 过期时间
@ -55,20 +61,32 @@ public class SequenceUtils {
* @return ID生成器 * @return ID生成器
*/ */
private static RIdGenerator getIdGenerator(String key, Duration expireTime, Long initValue, Long stepValue) { private static RIdGenerator getIdGenerator(String key, Duration expireTime, Long initValue, Long stepValue) {
if (initValue == null || initValue <= 0) { if (StringUtils.isBlank(key)) {
initValue = DEFAULT_INIT_VALUE; throw new IllegalArgumentException("key 不能为空");
} }
if (stepValue == null || stepValue <= 0) { initValue = (initValue == null || initValue <= 0) ? DEFAULT_INIT_VALUE : initValue;
stepValue = DEFAULT_STEP_VALUE; stepValue = (stepValue == null || stepValue <= 0) ? DEFAULT_STEP_VALUE : stepValue;
}
RIdGenerator idGenerator = REDISSON_CLIENT.getIdGenerator(key); RIdGenerator idGenerator = REDISSON_CLIENT.getIdGenerator(DEFAULT_PREFIX + key);
// 设置初始值和步长 try {
idGenerator.tryInit(initValue, stepValue); // 初始化成功后设置过期时间，避免多线程反复设置
// 设置过期时间 if (idGenerator.tryInit(initValue, stepValue)) {
idGenerator.expire(expireTime); idGenerator.expire(expireTime);
}
} catch (Exception e) {
log.error("ID生成器初始化失败: key={}, initValue={}, stepValue={}", key, initValue, stepValue, e);
throw e;
}
return idGenerator; return idGenerator;
} }
/**
* 构建标准的业务 key：加上默认前缀
*/
private static String generateKey(String prefix, String suffix) {
return StringUtils.format("{}{}", StringUtils.blankToDefault(prefix, ""), suffix);
}
/** /**
* 获取指定业务key的唯一id * 获取指定业务key的唯一id
* *
@ -145,12 +163,8 @@ public class SequenceUtils {
* @return 唯一id * @return 唯一id
*/ */
public static String nextIdDate(String prefix) { public static String nextIdDate(String prefix) {
// 前缀+日期 构建 prefixKey String key = generateKey(prefix, DateUtils.getCurrentDate());
String prefixKey = StringUtils.format("{}{}", StringUtils.blankToDefault(prefix, ""), DateUtil.format(DateUtil.date(), DatePattern.PURE_DATE_FORMATTER)); return key + nextId(key, DEFAULT_EXPIRE_DAY);
// 获取下一个id
long nextId = getIdGenerator(prefixKey, DEFAULT_EXPIRE_TIME_DAY, DEFAULT_INIT_VALUE, DEFAULT_STEP_VALUE).nextId();
// 返回完整id
return StringUtils.format("{}{}", prefixKey, nextId);
} }
/** /**
@ -169,12 +183,8 @@ public class SequenceUtils {
* @return 唯一id * @return 唯一id
*/ */
public static String nextIdDateTime(String prefix) { public static String nextIdDateTime(String prefix) {
// 前缀+日期时间 构建 prefixKey String key = generateKey(prefix, DateUtils.dateTimeNow());
String prefixKey = StringUtils.format("{}{}", StringUtils.blankToDefault(prefix, ""), DateUtil.format(DateUtil.date(), DatePattern.PURE_DATETIME_FORMATTER)); return key + nextId(key, DEFAULT_EXPIRE_MINUTE);
// 获取下一个id
long nextId = getIdGenerator(prefixKey, DEFAULT_EXPIRE_TIME_MINUTE, DEFAULT_INIT_VALUE, DEFAULT_STEP_VALUE).nextId();
// 返回完整id
return StringUtils.format("{}{}", prefixKey, nextId);
} }
} }