Redis编号自动生成

最新推荐文章于 2024-12-02 18:03:47 发布

原创最新推荐文章于 2024-12-02 18:03:47 发布 · 351 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #redis #开发语言

该代码段展示了一个公共类`NumberUtils`，用于生成带日期的四位数字编号。它利用`RedisTemplate`与Redis交互，通过日期格式化和增量操作保证编号的唯一性。如果Redis中不存在指定key，则设置初始值为1并返回。

public class NumberUtils {

    /**
     * 编号自动生成
     *
     * @param redisTemplate redis模板
     * @return 编号
     */
    public static String generateNo(RedisTemplate<String, Object> redisTemplate) throws ServiceException {
        try {
            // nums位数字，不够的前面补0
            val format = String.format("%%0%dd", 4);

            val key = CommonConstant.NO + new SimpleDateFormat(DatePattern.PURE_DATE_PATTERN).format(new Date());

            // redisKey组成：前缀 + 格式化后的具体日期，确保每天都是新的key，编号从0001开始
            return key + String.format(format, getIncrement(redisTemplate, key));
        } catch (Exception e) {
            throw new ServiceException(ExceptionCode.ERROR_20019);
        }
    }

    private static Integer getIncrement(RedisTemplate<String, Object> redisTemplate, String redisKey) {
        return Optional
                .ofNullable(redisTemplate.opsForValue().get(redisKey))
                // 返回redis里的值 + 1
                .map(num -> Objects.requireNonNull(redisTemplate.opsForValue().increment(redisKey)).intValue())
                // 如果redis里没有，则插入1并通过判断插入是否成功来返回初始化值1，如果插入失败，则返回拼接null
                .orElse(Boolean.TRUE.equals(redisTemplate.opsForValue().setIfAbsent(redisKey, 1)) ? 1 : null);
    }

}