Redis 生成分布式业务单号

蔚落 2024-03-24 11:23 5阅读 0赞

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

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

例子：

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