Redis编号自动生成

最新推荐文章于 2023-10-13 15:51:41 发布

Haaaaaang

最新推荐文章于 2023-10-13 15:51:41 发布

阅读量228

点赞数

文章标签： java redis 开发语言

本文链接：https://blog.csdn.net/java__project/article/details/129881843

版权

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);
    }

}