使用Redis自动生成业务递增流水号

使用场景

我们平时在功能开发的时候可能会遇到需要提供根据日期加编号生成的流水号业务场景。有时候为了图省事我们会自定义拼接生成一些流水号。但是这种写法有很大的破绽。遇到稍微的并发场景就会出现流水号重复，这样就会造成一些不必要的损失。为了避免这种重复的情况，我们可以利用redis的单线程模式生成流水号。

两种方式对比

第一种：普通情况下自动拼接生成流水号

优点： 代码实现简单，不需要第三方数据库的协助。

缺点： 使用业务场景单一，稍微复杂一点的业务就会出错。适合单机处理。增加数据压力。

package com.**.utils;

import cn.hutool.extra.spring.SpringUtil;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.redis.core.StringRedisTemplate;

import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.concurrent.TimeUnit;

public class BusinessNumsGenerator {

	/**
	 * 1-2位 为数据业务表缩写标识
	 * 3-8位 为顺序码 100000 开始
	 * @return
	 */
	/**
	 * 获取业务数据编码
	 * 1-2位 为数据业务表缩写标识
	 * 3-8位 为顺序码 100000 开始
	 * @param currentValue 当前业务数据库记录的最大值，无可传null或者“”
	 * @return 最新业务数据编码
	 */
	public static String getNumCode(String currentValue) {
		int i = 10000;
		String dateStr = getDateStr();
		if(StringUtils.isNotEmpty(currentValue) && currentValue.length() == 13){
			String ss = currentValue.substring(8, 13);
			String substring = currentValue.substring(2, 8);
			if (!substring.equals(dateStr)) {
				i = i + 1;
			} else {
				i = Integer.parseInt(ss) + 1;
			}
		}
		return "XQ" + dateStr + i;
	}
	/**
	 * 获取日期戳
	 * @return
	 */
	public static String getDateStr(){
		return  new SimpleDateFormat("yyMMdd").format(new Date());
	}
}

第二种：使用redis生成自增流水号

优点： 安全，能经受起发数据量的压力，效率高。

缺点： 需要依赖第三方数据库。

package com.**.utils;

import cn.hutool.extra.spring.SpringUtil;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.redis.core.StringRedisTemplate;

import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.concurrent.TimeUnit;

public class BusinessNumsGenerator {

	/**
	 *
	 * 通过redis 自增生成流水号
	 * @param
	 * @author:  zfm
	 * @date:  11:07  2023/2/14
	 * @return:  {@link String}
	 */
	public String createAutoID() {
		StringRedisTemplate redisTemplate = SpringUtil.getBean(StringRedisTemplate.class);
		//时间戳 后面拼接流水号 如果需要  可以加上时分秒
		String datetime = getDateStr();
		//这里是 Redis key的前缀，如: sys:表名:日期  如果不需要去掉表名也可以
		String key = MessageFormat.format("{0}:{1}:{2}", "sys", "material", datetime);
		//查询 key 是否存在， 不存在返回 1 ，存在的话则自增加1
		Long autoID = redisTemplate.opsForValue().increment(key, 1);
		// 设置key过期时间, 保证每天的流水号从1开始
		if (autoID == 1) {
			redisTemplate.expire(key, getSecondsNextEarlyMorning(), TimeUnit.SECONDS);
		}
		//这里是 6 位id，如果位数不够可以自行修改 ，下面的意思是 得到上面 key 的 值，位数为6 ，不够的话在左边补 0 ，比如  110 会变成  000110
		String value = StringUtils.leftPad(String.valueOf(autoID), 5, "0");
		//然后把 时间戳和优化后的 ID 拼接
		String code = MessageFormat.format("{0}{1}{2}", "XQ", datetime, value);
		return code;
	}

	/**
	 * 判断当前时间距离第二天凌晨的秒数
	 *
	 * @return 返回值单位为[s:秒]
	 */
	public Long getSecondsNextEarlyMorning() {
		Calendar cal = Calendar.getInstance();
		cal.add(Calendar.DAY_OF_YEAR, 1);
		cal.set(Calendar.HOUR_OF_DAY, 0);
		cal.set(Calendar.SECOND, 0);
		cal.set(Calendar.MINUTE, 0);
		cal.set(Calendar.MILLISECOND, 0);
		return (cal.getTimeInMillis() - System.currentTimeMillis()) / 1000;
	}
}