package cn.iocoder.yudao.module.mes.dal.redis;
|
|
import cn.iocoder.yudao.framework.common.util.number.BizNoUtils;
|
import jakarta.annotation.Resource;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.data.redis.core.script.DefaultRedisScript;
|
import org.springframework.data.redis.core.script.RedisScript;
|
import org.springframework.stereotype.Repository;
|
|
import java.util.List;
|
|
/**
|
* MES 业务编码生成的 Redis DAO（实例码：袋码 / 托盘码 / 临时码等）。
|
*
|
* <p>需求：所有业务编码统一为「16 位纯数字」，且全系统唯一。
|
* 这里复用全系统统一的 Redis 自增 KEY，保证与其它模块的编码也不重复。
|
*/
|
@Repository
|
public class MesCodeRedisDAO {
|
|
/**
|
* 袋序号自增 KEY（全局袋序号，保证同一批/跨批生成时不重复）
|
*/
|
private static final String BAG_NO_SEQ_KEY = "mes:bag_no:seq";
|
|
/**
|
* 原子分配袋序号区间的 Lua 脚本。
|
* <p>
|
* 先按库内最大序号校正基准值（Redis 首次使用、或库内数据被直接导入导致基准落后时），
|
* 再原子自增并返回自增后的值。整个「校正 + 自增」在 Redis 单线程内完成，避免并发下互相覆盖计数。
|
* <p>
|
* KEYS[1] = 序号 KEY；ARGV[1] = 本次需要的个数；ARGV[2] = 库内当前最大序号
|
*/
|
private static final RedisScript<Long> ALLOCATE_BAG_NO_SCRIPT = new DefaultRedisScript<>(
|
"local cur = redis.call('GET', KEYS[1]) "
|
+ "local base = tonumber(ARGV[2]) "
|
+ "if (not cur) or (tonumber(cur) < base) then redis.call('SET', KEYS[1], base) end "
|
+ "return redis.call('INCRBY', KEYS[1], ARGV[1])",
|
Long.class);
|
|
@Resource
|
private StringRedisTemplate stringRedisTemplate;
|
|
/**
|
* 生成一个 16 位纯数字业务编码（全系统唯一）
|
*
|
* @return 16 位纯数字编码
|
*/
|
public String generate() {
|
// 随机生成 16 位纯数字编码，并用 Redis 集合查重，保证全系统唯一（不使用自增/顺序）
|
for (int i = 0; i < BizNoUtils.MAX_RETRY; i++) {
|
String code = BizNoUtils.randomCode();
|
Long added = stringRedisTemplate.opsForSet().add(BizNoUtils.USED_SET_KEY, code);
|
if (added != null && added > 0) {
|
return code;
|
}
|
}
|
throw new IllegalStateException("生成唯一业务编码失败，请重试");
|
}
|
|
/**
|
* 原子分配一段连续的袋序号，返回区间起始值。
|
* <p>
|
* 袋序号原先由「查库内最大值再 +1」得到，并发生成时两个请求会读到同一个最大值，
|
* 从而产出重复的袋序号（袋序号会印在标签上）。此处改为 Redis 原子自增分配，
|
* 并用库内当前最大值做基准校正，保证序号既不重复也不回退。
|
*
|
* @param count 本次需要的连续序号个数（须大于 0）
|
* @param dbMaxBagNo 数据库当前最大袋序号（含已逻辑删除记录，传 null 视为 0）
|
* @return 区间起始序号（含），可用区间为 [start, start + count - 1]
|
*/
|
public int allocateBagNoStart(int count, Integer dbMaxBagNo) {
|
if (count <= 0) {
|
throw new IllegalArgumentException("袋序号分配个数必须大于 0");
|
}
|
Long end = stringRedisTemplate.execute(ALLOCATE_BAG_NO_SCRIPT, List.of(BAG_NO_SEQ_KEY),
|
String.valueOf(count), String.valueOf(dbMaxBagNo == null ? 0 : dbMaxBagNo));
|
if (end == null) {
|
throw new IllegalStateException("分配袋序号失败，请重试");
|
}
|
return (int) (end - count + 1);
|
}
|
|
}
|
