package cn.iocoder.yudao.module.mes.dal.redis.md.autocode;
|
|
import cn.iocoder.yudao.module.mes.dal.redis.RedisKeyConstants;
|
import jakarta.annotation.Resource;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.stereotype.Repository;
|
|
import java.time.Duration;
|
|
/**
|
* MES 编码规则的 Redis DAO
|
*
|
* @author 芋道源码
|
*/
|
@Repository
|
public class MesMdAutoCodeRedisDAO {
|
|
@Resource
|
private StringRedisTemplate stringRedisTemplate;
|
|
/**
|
* 递增序号（带初始值和循环）
|
*
|
* @param keySuffix key 后缀（不包含 prefix）
|
* @param duration 过期时间
|
* @param step 步长
|
* @param start 起始值（仅在 key 不存在时使用）
|
* @return 递增后的值
|
*/
|
public Long increment(String keySuffix, Duration duration, Integer start, Integer step) {
|
String key = RedisKeyConstants.AUTO_CODE + keySuffix;
|
|
// 情况一：如果 key 不存在，初始化为 startNo
|
Boolean success = stringRedisTemplate.opsForValue().setIfAbsent(key, String.valueOf(start));
|
if (Boolean.TRUE.equals(success)) {
|
// 设置成功，说明是第一次，设置过期时间并返回 startNo
|
if (duration != null) {
|
stringRedisTemplate.expire(key, duration);
|
}
|
return start.longValue();
|
}
|
|
// 情况二：key 已存在，递增
|
Long value = stringRedisTemplate.opsForValue().increment(key, step);
|
if (duration != null) {
|
stringRedisTemplate.expire(key, duration);
|
}
|
return value;
|
}
|
|
}
|
