如何使用redis生成流水号

article/2025/11/8 16:34:02

概述
本文讲述如何使用redis生成流水号。本文是在Springboot中实现的。知道原理之后其他框架也可以轻松实现。
原理介绍
本文主要是使用redis的incr方法进行自增补零。然后结合时间、随机数、前缀组成唯一的流水号。
在这里插入图片描述
下面是流水号的结构。
在文章的最后还是简单介绍一下redis的持久化。防止redis宕机导致自增数重置。从而导致流水号重复。

流程图

本次实现分为

1、初始化流水号信息进入缓存

2、生成流水号

下面是分别的大致流程图

1、初始化流水号信息进入缓存
在这里插入图片描述
2、生成流水号

代码实现
1、流水号实体类
用于存储流水号信息

@Data
public class YuSnGenCode {// 流水号生成依赖的类，会使用类的名称作为redis的key private Class entity;// 前缀private String prefix;// 自增数位数 自增数达不到此位数自动补零private Integer num;public YuSnGenCode(Class entity, String prefix, Integer num) {this.entity = entity;this.prefix = prefix;this.num = num;}
}

2、初始化流水号进入缓存
在程序启动之后初始化流水号

@Component
@Slf4j
public class YuSnGenStart implements ApplicationRunner {@AutowiredYuSnGenUtil yuSnGenUtil;@Overridepublic void run(ApplicationArguments args) throws Exception {log.info("=====开始初始化流水号=====");List<YuSnGenCode> yuSnGenCodeList = new ArrayList<>();yuSnGenCodeList.add(new YuSnGenCode(Account.class, "AC", 6));yuSnGenCodeList.add(new YuSnGenCode(Shop.class, "DP", 6));yuSnGenCodeList.add(new YuSnGenCode(Order.class, "OR", 6));yuSnGenCodeList.add(new YuSnGenCode(Dispatch.class, "DL", 6));yuSnGenCodeList.add(new YuSnGenCode(Refund.class, "RF", 6));yuSnGenCodeList.add(new YuSnGenCode(PayLog.class, "PL", 6));yuSnGenCodeList.add(new YuSnGenCode(Withdrawal.class, "WR", 6));yuSnGenCodeList.add(new YuSnGenCode(Aftermarket.class, "AS", 6));yuSnGenCodeList.add(new YuSnGenCode(Coupon.class, "CO", 6));yuSnGenCodeList.add(new YuSnGenCode(CouponGen.class, "CG", 10));yuSnGenCodeList.add(new YuSnGenCode(Draw.class, "DP", 6));yuSnGenUtil.init(yuSnGenCodeList);log.info("=====初始化流水号完毕=====");}
}

 public void init(List<YuSnGenCode> yuSnGenCodes) {// 流水号初始化入缓存for (YuSnGenCode yuSnGenCode : yuSnGenCodes) {String redisKey = yuSnGenCode.getEntity().getName();// 存入缓存 key：key：实体类名称 value：流水号数据（前缀、自增数位数）redisTemplate.opsForValue().set(yuSnGenCode.getEntity().getName(), FastJsonUtils.toJsonStr(yuSnGenCode));log.info(yuSnGenCode.getEntity().getName() + "已初始化");}}

初始化之后的流水号如下图所示

在这里插入图片描述
3、生成流水号

 public Optional<String> gen(Class c) {// 获取实体类的名称String redisKey = c.getName();// 判断是不是有初始化此实体类if (null != redisTemplate.opsForValue().get(redisKey)) {// 从缓存获取流水号的生成信息YuSnGenCode yuSnGenCode = FastJsonUtils.toBean(redisTemplate.opsForValue().get(redisKey).toString(), YuSnGenCode.class);// 根据流水号的前缀判断今天是否有生成过流水号if (redisTemplate.opsForValue().get(yuSnGenCode.getPrefix()) == null) {// 没有则新建一个存入缓存 格式（key:OR  value:0）// 设置到第二天早上00：00：01过期Long todayTime = LocalDate.now().plusDays(1).atTime(0, 0, 0, 1).atOffset(ZoneOffset.ofHours(8)).toEpochSecond();Long nowTime = LocalDateTime.now().atOffset(ZoneOffset.ofHours(8)).toEpochSecond();Long expireTime = todayTime - nowTime;redisTemplate.opsForValue().set(yuSnGenCode.getPrefix(), 0, expireTime*1000, TimeUnit.MILLISECONDS);}// 进行自增操作StringBuffer sn = new StringBuffer();// 和前缀、时间、随机数进行组合sn.append(yuSnGenCode.getPrefix());String date = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));sn.append(date);Long num = redisTemplate.opsForValue().increment(yuSnGenCode.getPrefix());sn.append(addZero(String.valueOf(num), yuSnGenCode.getNum()));String random = String.valueOf(new Random().nextInt(1000));sn.append(random);// 生成最终的流水号返回return Optional.ofNullable(sn.toString());}return Optional.ofNullable(null);}

 // 自动补零public String addZero(String numStr, Integer maxNum) {int addNum = maxNum - numStr.length();StringBuffer rStr = new StringBuffer();for (int i = 0; i < addNum; i++) {rStr.append("0");}rStr.append(numStr);return rStr.toString();}

第一次生成流水号之后redis中会有一个以此流水号的前缀作为key的数据，后续直接进行自增。无需新增

在这里插入图片描述
代码测试
普通测试

   public Response test2() {return Response.success(yuSnGenUtil.gen(Order.class).get());}

结果如下
在这里插入图片描述
并发测试
这里我们使用ab进行压力并发测试

测试前我们流水号数据为0
在这里插入图片描述
进行5000请求100并发的测试

请求完毕之后流水号的数据是5000是没有问题的。

redis 持久化
redis持久化指redis意外退出之后重启仍然能够恢复之前数据。我们这里使用redis持久化防止redis意外退出重启导致流水号数据重置，从而导致我们的流水号生成重复。

Redis 提供了不同级别的持久化方式:

RDB

RDB持久化方式能够在指定的时间间隔能对你的数据进行快照存储.

AOF

AOF持久化方式记录每次对服务器写的操作,当服务器重启的时候会重新执行这些命令来恢复原始的数据

我们本次使用的是AOF的方式
在这里插入图片描述
根据官网的提示。我们只需要去redis.config文件中开启appendonly就可以了。这样redis会生成appendonly.aof文件，服务器重启的时候会重新执行这些命令来恢复原始的数据。

注意
因为AOF的默认备份方式有三种
1、always每次有新命令追加到 AOF 文件时就执行一次 fsync ：非常慢，也非常安全