Redis生成唯一id操作类

看好了,我只操作一次

@Repository
@Slf4j
public class RedisCounterRepository {
    
    

    private final DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern("yyyyMMdd");

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    // 根据获取的自增数据,添加日期标识构造分布式全局唯一标识

    /**
     * 序列号长度:yyyymmdd + 6位递增字符
     * @param changeNumPrefix 自增的redis原子
     * @return
     */
    public String getNumFromRedis(String changeNumPrefix) {
    
    
        return getNumFromRedis(changeNumPrefix,6);
    }

    /**
     * 序列号长度:yyyymmdd + count位递增字符
     * @param changeNumPrefix  自增的redis原子
     * @param count            自定义序列号长度
     * @return
     */
    public String getNumFromRedis(String changeNumPrefix,int count) {
    
    
        String dateStr = LocalDate.now().format(dateTimeFormatter);
        Long value = incrementNum(changeNumPrefix + dateStr);
        String formatStr = "%0"+count+"d";
        return dateStr + String.format(formatStr,value);
    }

    //

    /**
     * 从redis中获取自增数据(redis保证自增是原子操作)
     * @param key  redis自增的key值
     * @return
     */
    private long incrementNum(String key) {
    
    
        RedisConnectionFactory factory = redisTemplate.getConnectionFactory();
        if (null == factory) {
    
    
            log.error("Unable to connect to redis.");
        }
        RedisAtomicLong redisAtomicLong = new RedisAtomicLong(key, factory);
        long increment = redisAtomicLong.incrementAndGet();
        if (1 == increment) {
    
    
            // 如果数据是初次设置,需要设置超时时间
            redisAtomicLong.expire(1, TimeUnit.DAYS);
        }
        return increment;
    }
}

猜你喜欢

转载自blog.csdn.net/m0_46580493/article/details/130359205