基于Redis自增函数生成分布式单号

57 阅读 0 评论 38 点赞

我是靠谱客的博主懦弱饼干，最近开发中收集的这篇文章主要介绍基于Redis自增函数生成分布式单号，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

1. 背景

在分布式系统中，不可避免的需要协调多个服务器节点生成分布式的单号问题，如果采取单节点的方式，可能会出现单号重复的问题。从具体的实现方式而言，可以采取数据库的方案，但这个可能存在多节点同时争抢同一条数据造成死锁的问题。我们这里采取分布式中间件Redis的自增函数，实现分布式单号的生成。

2. 业务要求

分布式的单号生成组件，需要根据不同的业务要求生成不同规则的业务单号，并且在高并发的场景下保证单号唯一。

3. 分布式单号组件的实现

3.1 单号的生成业务规则接口

/**
* 分布式单号的格式化前缀接口
*/
public interface DistributedSeqNoFormat<T>{
/**
* 根据特定的业务对象类型，进行定制化的单号前缀格式化输出
* @param t 单号的前缀格式化操作对象
* @return 单号的前缀
* @author chenyz18
*/
String format(T t);
}

3.2 单号生成器

/**
* 分布式单号生成器
*
*/
public class DistributedSeqNoGenerator <T>{
private static final Logger logger = LoggerFactory.getLogger(DistributedSeqNoGenerator.class);
/**
* 根据业务定制化的前缀格式化生成器，生成分布式的业务单号
* @param format 格式化生成器的实现实例
* @param expiredTime 单号的前缀超时时间, 如果设置为-1则不会超时
* @param t 格式化生成器的格式化业务操作类型实例对象
//* @param <T> 格式化生成器的格式化业务操作类型
* @return 分布式的单号
*/
public String getBizSerialNo(DistributedSeqNoFormat<T> format, T t, Long expiredTime) {
StringBuilder serialNo = new StringBuilder();
String prefix = format.format(t);
serialNo.append(prefix);
serialNo.append(getCommonSerialNo(prefix, expiredTime));
return serialNo.toString();
}
/**
* 根据业务定制化的前缀格式化生成器，生成定长的分布式业务单号
* @param format 格式化生成器的实现实例
* @param t 格式化生成器的格式化业务操作类型实例对象
* @param maxSerialNo 序列化的对打长度，如果生成的序列号不足该长度，则补零
* @param expiredTime 单号的前缀超时时间, 如果设置为-1则不会超时
//* @param <T> 格式化生成器的格式化业务操作类型
* @return 分布式的单号
*/
public
String getFixedLengthBizSerialNo(DistributedSeqNoFormat<T> format, T t, Integer maxSerialNo, Long expiredTime) {
String prefix = format.format(t);
Long seqNo = getCommonSerialNo(prefix, expiredTime);
int startNoLen = Long.toString(seqNo).length();
if (startNoLen <= maxSerialNo) {
String tmpStr = "";
//空缺的为止用0补全
for (int i = 0; i < (maxSerialNo - startNoLen); i++) {
tmpStr = tmpStr + "0";
}
String serialNo = tmpStr + seqNo;
serialNo = prefix + serialNo;
return serialNo;
} else {
return getFixedLengthBizSerialNo(format, t, maxSerialNo, expiredTime);
}
}
/**
* 根据redis生成分布式的序列号
*
* @param prefix
序列号前缀
* @param expiredTime 序列号的过期时间
* @return 根据序列号前缀生成的序列号
*/
public static Long getCommonSerialNo(String prefix, Long expiredTime) {
if (StringUtils.isBlank(prefix)) {
throw new IllegalArgumentException("单号前缀不能为空!");
}
long resultSeq;
//get redis client implements instantce 
CodisImpl cache = (CodisImpl) SpringUtil.getBean("redisBean");
resultSeq = cache.incr(prefix);
if (expiredTime > 0) {
cache.expire(prefix, expiredTime);
}
return resultSeq;
}
}

3.3测试类

	public static String getBillNO( String prefix){
DistributedSeqNoGenerator<String> generator = new DistributedSeqNoGenerator<>();
return generator.getFixedLengthBizSerialNo(new DistributedSeqNoFormat<String>() {
@Override
public String format(String s) {
SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMdd");
String str = dateFormat.format(new Date());
return s+str;
}
},prefix, maxSeriaNo,86400L);
}