背景:微服务架构,有三个微服务,分别是签收、对账、开票,需要生成唯一的分布式单号
格式:标识 + 年月日 + 生成顺序(三位)
例子:
- 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);