Redis唯一ID生成器的实现

public class RedisIdWorker {

  
  private static final long BEGIN_TIMESTAMP = 1640995200L;
  
  private static final int COUNT_BITS = 32;

  private StringRedisTemplate stringRedisTemplate;
  //构造方法形式注入
  public RedisIdWorker(StringRedisTemplate stringRedisTemplate) {
    this.stringRedisTemplate = stringRedisTemplate;
  }

  public long nextId(String keyPrefix){
    //1. 生成时间戳
    LocalDateTime now = LocalDateTime.now();
    long nowSecond = now.toEpochSecond(ZoneOffset.UTC);
    long timestamp = nowSecond - BEGIN_TIMESTAMP;
    //2.生成序列号
    // 2.1 获取当前日期，精确到天
    String date = now.fORMat(DateTimeFormatter.ofPattern("yyyy:MM:dd"));
    long count = stringRedisTemplate.opsForValue().increment("icr:" + keyPrefix + ":" + date);
    //3.拼接并返回

    return timestamp << COUNT_BITS | count;
  }
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.Calendar;
import java.util.concurrent.TimeUnit;


@Component
public class GenerateIDUtil {

  @Autowired
  private RedisTemplate redisTemplate;

   public String initPrimaryId(String key) {
    Assert.hasLength(key, "hashName不能为空");
    String hashCol = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
    //自定义编号规则
    String hashColVal = hashCol + "00001";
//    redisTemplate.opsForHash().putIfAbsent(hashName, hashCol, hashColVal);

    Long expiresTime = getSecondsNextEarlyMorning();
    redisTemplate.opsForValue().set(key, Long.valueOf(hashColVal), expiresTime, TimeUnit.SECONDS);
    return hashColVal;
  }


  
  public String getPrimaryId(String key) {

    String id = "";
    if(redisTemplate.hasKey(key)){
      // redisTemplate.opsForValue().get(key);
      // redisTemplate.delete(key);
      id = String.valueOf(redisTemplate.opsForValue().increment(key, 1));
    } else {
      id = initPrimaryId(key);
    }
    return id;
  }


  
  public Long getSecondsNextEarlyMorning() {
    Calendar cal = Calendar.getInstance();
    cal.add(Calendar.DAY_OF_YEAR, 1);
    cal.set(Calendar.HOUR_OF_DAY, 0);
    cal.set(Calendar.SECOND, 0);
    cal.set(Calendar.MINUTE, 0);
    cal.set(Calendar.MILLISECOND, 0);
    return (cal.getTimeInMillis() - System.currentTimeMillis()) / 1000;
  }
}