From 02a1ae81eba7b48223052fb833859f0a65dd7d21 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期一, 19 十月 2026 17:31:08 +0800
Subject: [PATCH] feat(mes): 实现袋码大批量生成异步任务功能
---
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/redis/MesCodeRedisDAO.java | 105 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 105 insertions(+), 0 deletions(-)
diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/redis/MesCodeRedisDAO.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/redis/MesCodeRedisDAO.java
index b0a9318..dd24ca3 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/redis/MesCodeRedisDAO.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/redis/MesCodeRedisDAO.java
@@ -3,7 +3,12 @@
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.ArrayList;
+import java.util.List;
/**
* MES 涓氬姟缂栫爜鐢熸垚鐨� Redis DAO锛堝疄渚嬬爜锛氳鐮� / 鎵樼洏鐮� / 涓存椂鐮佺瓑锛夈��
@@ -13,6 +18,49 @@
*/
@Repository
public class MesCodeRedisDAO {
+
+ /**
+ * 琚嬪簭鍙疯嚜澧� KEY锛堝叏灞�琚嬪簭鍙凤紝淇濊瘉鍚屼竴鎵�/璺ㄦ壒鐢熸垚鏃朵笉閲嶅锛�
+ */
+ private static final String BAG_NO_SEQ_KEY = "mes:bag_no:seq";
+
+ /**
+ * 鍗曟鎵归噺鍙栫爜鐨勬渶澶т釜鏁�
+ * <p>
+ * 鍙� Redis 鍗曟潯鍛戒护鍙傛暟瑙勬ā涓庡搷搴斾綋澶у皬绾︽潫锛岃繃澶у弽鑰屽鍔犲崟娆￠樆濉炴椂闂淬��
+ */
+ public static final int MAX_BATCH_SIZE = 5000;
+
+ /**
+ * 鎵归噺鍐欏叆骞惰繑鍥炲疄闄呯敓鏁堢殑缂栫爜銆�
+ * <p>
+ * 閫愪釜 SADD 鍦ㄥ悓涓�鑴氭湰鍐呭畬鎴愶細涓庤皟鐢ㄦ柟閫愭潯 SADD 鐩告瘮锛岀綉缁滃線杩斾粠 N 娆￠檷涓� 1 娆★紱
+ * 鍚屾椂鍥犱负鑴氭湰鍘熷瓙鎵ц锛屽苟鍙戣皟鐢ㄤ箣闂翠篃涓嶄細浜х敓閲嶅缂栫爜銆�
+ */
+ private static final RedisScript<List> ADD_UNIQUE_CODES_SCRIPT = new DefaultRedisScript<>(
+ "local res = {} "
+ + "for i = 1, #ARGV do "
+ + " if redis.call('SADD', KEYS[1], ARGV[i]) == 1 then "
+ + " res[#res + 1] = ARGV[i] "
+ + " end "
+ + "end "
+ + "return res",
+ List.class);
+
+ /**
+ * 鍘熷瓙鍒嗛厤琚嬪簭鍙峰尯闂寸殑 Lua 鑴氭湰銆�
+ * <p>
+ * 鍏堟寜搴撳唴鏈�澶у簭鍙锋牎姝ｅ熀鍑嗗�硷紙Redis 棣栨浣跨敤銆佹垨搴撳唴鏁版嵁琚洿鎺ュ鍏ュ鑷村熀鍑嗚惤鍚庢椂锛夛紝
+ * 鍐嶅師瀛愯嚜澧炲苟杩斿洖鑷鍚庣殑鍊笺�傛暣涓�屾牎姝� + 鑷銆嶅湪 Redis 鍗曠嚎绋嬪唴瀹屾垚锛岄伩鍏嶅苟鍙戜笅浜掔浉瑕嗙洊璁℃暟銆�
+ * <p>
+ * KEYS[1] = 搴忓彿 KEY锛汚RGV[1] = 鏈闇�瑕佺殑涓暟锛汚RGV[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;
@@ -34,4 +82,61 @@
throw new IllegalStateException("鐢熸垚鍞竴涓氬姟缂栫爜澶辫触锛岃閲嶈瘯");
}
+ /**
+ * 鎵归噺鐢熸垚 count 涓敮涓�鐨� 16 浣嶇函鏁板瓧涓氬姟缂栫爜锛堝叏绯荤粺鍞竴锛�
+ * <p>
+ * 澶ф壒閲忕敓鎴愭椂涓嶈寰幆璋冪敤 {@link #generate()}锛氶偅浼氫骇鐢� N 娆� Redis 寰�杩旓紝
+ * 鍑犲崄涓囨潯鍗冲彲鎶婃帴鍙ｆ嫋鍒板垎閽熺骇銆傝繖閲屾瘡鎵规渶澶� {@link #MAX_BATCH_SIZE} 涓�欓�夌爜
+ * 涓�娆¤剼鏈啓鍏ワ紝纰版挒锛堟鐜囨瀬浣庯級鐨勭己棰濆啀琛ヤ竴鎵广��
+ *
+ * @param count 闇�瑕佺殑缂栫爜涓暟
+ * @return 鍞竴缂栫爜鍒楄〃锛岄『搴忔棤淇濊瘉浣嗘暟閲忕瓑浜� count
+ */
+ public List<String> generateBatch(int count) {
+ if (count <= 0) {
+ return List.of();
+ }
+ List<String> result = new ArrayList<>(Math.min(count, 1024));
+ // 16 浣嶉殢鏈虹爜绌洪棿涓� 10^16锛屾甯告儏鍐典笅涓�杞嵆鍙叏閮ㄥ啓鍏ワ紱闄愬埗杞鍙敤浜庡厹搴曞紓甯�
+ for (int round = 0; round < 100 && result.size() < count; round++) {
+ int batchSize = Math.min(count - result.size(), MAX_BATCH_SIZE);
+ List<String> candidates = new ArrayList<>(batchSize);
+ for (int i = 0; i < batchSize; i++) {
+ candidates.add(BizNoUtils.randomCode());
+ }
+ List<?> added = stringRedisTemplate.execute(ADD_UNIQUE_CODES_SCRIPT,
+ List.of(BizNoUtils.USED_SET_KEY), candidates.toArray(new String[0]));
+ if (added != null) {
+ added.forEach(code -> result.add(String.valueOf(code)));
+ }
+ }
+ if (result.size() < count) {
+ throw new IllegalStateException("鎵归噺鐢熸垚鍞竴涓氬姟缂栫爜澶辫触锛岃閲嶈瘯");
+ }
+ return result;
+ }
+
+ /**
+ * 鍘熷瓙鍒嗛厤涓�娈佃繛缁殑琚嬪簭鍙凤紝杩斿洖鍖洪棿璧峰鍊笺��
+ * <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);
+ }
+
}
--
Gitblit v1.9.3