简单业务流水号生成器的思路和实现

1. 通过业务前缀以及当前date组合为key，在redis中存入当日生成的单号的数量，过期时间为当日的23:59:59:999
2. 每次调用接口后对redis中的对应key的单号数量进行自增
3. 不同的业务前缀对应的数量互不干扰
   生成的id组成为 业务前缀+日期+自增号,
   例如2023/12/15的第一个销售单:XSD20231215000001

import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.text.SimpleDateFormat;
import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.concurrent.TimeUnit;

/**
 * 业务流号生成器
 *
 * @author manongikun
 * @since 2023/12/20
 */
@Component
public class BusinessFlowNumberGenerator {
    private static RedisTemplate<String,Object> redisTemplate;

    private static final String DEFAULT_KEY_PREFIX = "";
    private static final String DATE_KEY = "date";
    private static final String PREFIX_DATE_FORMAT = "yyMMdd";
    private static final String FORMAT_TEMPLATE = "{prefix}{date}{number}";
    @Resource
    public void setRedisTemplate(RedisTemplate<String, Object> redisTemplate) {
        BusinessFlowNumberGenerator.redisTemplate = redisTemplate;
    }
    /**
     * 获取从当前日期到今天结束数字毫秒数
     * <p>依赖系统时间</p>
     *
     * @return long
     */
    private static long getsNumberMillisecondsFromCurrentDateToEndToday(){
        LocalDateTime currentTime = LocalDateTime.now();
        LocalDateTime midnight = currentTime.withHour(23).withMinute(59).withSecond(59).withNano(999999999);
        return ChronoUnit.MILLIS.between(currentTime, midnight);
    }

    /**
     * 生成流水号
     * <p>依赖redis</p>
     *
     * @param prefix 前缀
     * @return {@link String}
     */
    public static String generateFlowNumber(String prefix) {
        String dateStr = (String) redisTemplate.opsForValue().get(DATE_KEY);
        if (dateStr == null) {
            // 如果没有当前时间，则设置当前时间到redis中，并设置过期时间为当天最后一毫秒
            dateStr = new SimpleDateFormat(PREFIX_DATE_FORMAT).format(System.currentTimeMillis());
            redisTemplate.opsForValue().set(DATE_KEY, dateStr, getsNumberMillisecondsFromCurrentDateToEndToday(), TimeUnit.MILLISECONDS);
        }
        // 业务前缀加时间组合为缓存key
        String key = prefix + dateStr;
        if (Boolean.TRUE.equals(redisTemplate.hasKey(key))) {
            // 如果redis中存在，则从redis中获取，并自增1
            Long incr = redisTemplate.opsForValue().increment(key, 1);
            return FORMAT_TEMPLATE.replace("{prefix}", prefix).replace("{date}", dateStr).replace("{number}", String.format("%06d", incr));
        } else {
            // 如果redis中不存在，则设置redis中当前key的值为1，并设置过期时间为当天最后一毫秒
            redisTemplate.opsForValue().set(key, 1, getsNumberMillisecondsFromCurrentDateToEndToday(), TimeUnit.MILLISECONDS);
            Long incr = 1L;
            return FORMAT_TEMPLATE.replace("{prefix}", prefix).replace("{date}", dateStr).replace("{number}", String.format("%06d", incr));
        }
    }

    /**
     * 生成流水号
     *
     * @return {@link String}
     */
    public static String generateFlowNumber(){
        return generateFlowNumber(DEFAULT_KEY_PREFIX);
    }

使用

System.out.println(BusinessFlowNumberGenerator.generateFlowNumber());