Redis 生成分布式业务单号

背景：微服务架构，有三个微服务，分别是签收、对账、开票，需要生成唯一的分布式单号

格式：标识 + 年月日 + 生成顺序（三位）

例子：

• QS20230301001，即2023年三月一日第一张签收单
• DZ20230212002，即2023年二月十二日第二张对账单

原理：利用 Redis 的原子性，保证三位生成顺序的唯一性

新建业务id枚举类 IdEnum

public enum IdEnum {

    SIGN("QS"),
    VERIFY("DZ"),
    INVOICE("KP"),
    RECEIPT("SK"),
    ;

    /**
     * 单号前缀
     */
    private String prefix;

    IdEnum(String prefix) {
        this.prefix = prefix;
    }

    public String getPrefix() {
        return prefix;
    }
}

新建 IdUtil

• getCacheKey：获取 redis 的 KEY
• getDay：获取 服务标识 + 格式化的日期
• completionSerial：从redis获取生成顺序后，格式化为固定三位

public class IdUtil {

    static final String  PREFIX = "ID_CASH_";

    public static String getCacheKey(String serialPrefix) {
        return PREFIX.concat(serialPrefix);
    }

    public static String getDay(IdEnum idEnum) {
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyyMMdd");
        StringBuffer sb = new StringBuffer();
        sb.append(idEnum.getPrefix());
        sb.append(formatter.format(LocalDateTime.now()));
        return sb.toString();
    }

    public static String completionSerial(Long serial) {
        Format formatCount = new DecimalFormat("000");
        String serialFormat = formatCount.format(serial);
        return  serialFormat;
    }
}

新建 IdService 获取分布式单号

• dayWithPrefix ：获取 服务标识 + 格式化的日期
• redisCacheKey：Redis key
• serial：利用 redis increment 获取原子自增数

@Service
public class IdService {

    @Resource
    private RedisTemplate redisTemplate;

    public String generateNumber(IdEnum idEnum) {
        String dayWithPrefix = IdUtil.getDay(idEnum);
        String redisCacheKey = IdUtil.getCacheKey(dayWithPrefix);
        Long serial = redisTemplate.opsForValue().increment(redisCacheKey);
        redisTemplate.expire(redisCacheKey, 2, TimeUnit.DAYS);

        String num =  dayWithPrefix + IdUtil.completionSerial(serial);
        return num;
    }
}

最后使用

String number = idService.generateNumber(IdEnum.INVOICE);