From bbd6ab5d328556ec20432cd6016f8f352465b232 Mon Sep 17 00:00:00 2001
From: gongchunyi <deslre0381@gmail.com>
Date: 星期五, 10 四月 2026 15:28:59 +0800
Subject: [PATCH] fix: 审批节点生成流程编号重复问题
---
src/main/java/com/ruoyi/approve/utils/DailyRedisCounter.java | 110 +++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 103 insertions(+), 7 deletions(-)
diff --git a/src/main/java/com/ruoyi/approve/utils/DailyRedisCounter.java b/src/main/java/com/ruoyi/approve/utils/DailyRedisCounter.java
index eec0e4a..b62d7e9 100644
--- a/src/main/java/com/ruoyi/approve/utils/DailyRedisCounter.java
+++ b/src/main/java/com/ruoyi/approve/utils/DailyRedisCounter.java
@@ -1,18 +1,25 @@
package com.ruoyi.approve.utils;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.ruoyi.approve.mapper.ApproveProcessMapper;
+import com.ruoyi.approve.pojo.ApproveProcess;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.boot.ApplicationArguments;
+import org.springframework.boot.ApplicationRunner;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.time.LocalDate;
+import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.concurrent.TimeUnit;
//鍩轰簬redis鐨勪竴涓瘡鏃ヨ鏁板櫒
@Component
-public class DailyRedisCounter {
+public class DailyRedisCounter implements ApplicationRunner {
private static final String KEY_PREFIX = "daily_counter:";
private static final DateTimeFormatter DATE_FORMAT = DateTimeFormatter.ofPattern("yyyyMMdd");
private final StringRedisTemplate redisTemplate;
@@ -21,7 +28,48 @@
this.redisTemplate = redisTemplate;
}
+ @Value("${ruoyi.approvalNumberPrefix}")
+ private String approvalNumberPrefix;
+
+ @Autowired
+ private ApproveProcessMapper approveProcessMapper;
+
+ @Override
+ public void run(ApplicationArguments args) {
+ syncFromDb();
+ }
+
/**
+ * 寮哄埗浠庢暟鎹簱鍚屾褰撳墠璁℃暟鍒� Redis
+ */
+ public void syncFromDb() {
+ String key = approvalNumberPrefix + ":approveNum";
+ String lockKey = key + ":sync_lock";
+ Boolean lock = redisTemplate.opsForValue().setIfAbsent(lockKey, "1", 30, TimeUnit.SECONDS);
+ if (Boolean.TRUE.equals(lock)) {
+ try {
+ long dbCount = getCountFromDb();
+ redisTemplate.opsForValue().set(key, String.valueOf(dbCount), calculateSecondsUntilMidnight(), TimeUnit.SECONDS);
+ } finally {
+ redisTemplate.delete(lockKey);
+ }
+ }
+ }
+
+ /**
+ * 浠庢暟鎹簱鑾峰彇浠婃棩瀹℃壒鍗曟�婚噺
+ */
+ private long getCountFromDb() {
+ StartAndEndDateDto dateTime = getDateTime();
+ LambdaQueryWrapper<ApproveProcess> wrapper = new LambdaQueryWrapper<>();
+ wrapper.eq(ApproveProcess::getApproveDelete, 0)
+ .ge(ApproveProcess::getCreateTime, dateTime.getStartDate())
+ .lt(ApproveProcess::getCreateTime, dateTime.getEndDate());
+ Long count = approveProcessMapper.selectCount(wrapper);
+ return count == null ? 0 : count;
+ }
+
+ /**鏌ョ紦瀛�
* 鑾峰彇鎸囧畾璁℃暟鍣ㄥ湪浠婃棩鐨勬暟鍊硷紝骞惰嚜澧�1
* @param counterName 璁℃暟鍣ㄥ悕绉帮紙渚嬪锛歭ogin_count銆乷rder_count锛�
* @return 浠婃棩鑷鍚庣殑璁℃暟鍊�
@@ -40,6 +88,57 @@
}
/**
+ * 鑾峰彇褰撳墠鏃堕棿鐨� 寮�濮嬫棩鏈� 锛岀粨鏉熸棩鏈�
+ * @return
+ */
+ public static StartAndEndDateDto getDateTime(){
+ LocalDate now = LocalDate.now();
+ String startDateTime = now.toString();
+ String endDateTime = now.plusDays(1).toString();
+ StartAndEndDateDto startAndEndDateDto = new StartAndEndDateDto();
+ startAndEndDateDto.setStartDate(startDateTime);
+ startAndEndDateDto.setEndDate(endDateTime);
+ return startAndEndDateDto;
+ }
+
+ /**鏌ユ暟鎹簱
+ * 鑾峰彇鎸囧畾璁℃暟鍣ㄥ湪浠婃棩鐨勬暟鍊硷紝骞惰嚜澧�1
+ * @return 浠婃棩鑷鍚庣殑璁℃暟鍊�
+ */
+ public long incrementAndGetByDb() {
+ String key = approvalNumberPrefix + ":approveNum";
+ // 鑾峰彇鐜版湁鍊�
+ String approveId = redisTemplate.opsForValue().get(key);
+
+ if (approveId == null) {
+ // 缂撳瓨涓嶅瓨鍦ㄨ繘琛屽垵濮嬪寲
+ String lockKey = key + ":lock";
+ Boolean lock = redisTemplate.opsForValue().setIfAbsent(lockKey, "1", 10, TimeUnit.SECONDS);
+ if (Boolean.TRUE.equals(lock)) {
+ try {
+ approveId = redisTemplate.opsForValue().get(key);
+ if (approveId == null) {
+ long count = getCountFromDb();
+ long nextVal = count + 1;
+ redisTemplate.opsForValue().set(key, String.valueOf(nextVal), calculateSecondsUntilMidnight(), TimeUnit.SECONDS);
+ return nextVal;
+ }
+ } finally {
+ redisTemplate.delete(lockKey);
+ }
+ } else {
+ try {
+ Thread.sleep(100);
+ } catch (InterruptedException e) {
+ Thread.currentThread().interrupt();
+ }
+ return incrementAndGetByDb();
+ }
+ }
+ return redisTemplate.opsForValue().increment(key);
+ }
+
+ /**
* 鑾峰彇鎸囧畾璁℃暟鍣ㄥ湪浠婃棩鐨勫綋鍓嶆暟鍊�
* @param counterName 璁℃暟鍣ㄥ悕绉�
* @return 浠婃棩褰撳墠璁℃暟鍊硷紝鑻ヤ笉瀛樺湪鍒欒繑鍥�0
@@ -54,12 +153,9 @@
* 璁＄畻璺濈娆℃棩鍑屾櫒鐨勭鏁�
*/
private long calculateSecondsUntilMidnight() {
- LocalDate tomorrow = LocalDate.now().plusDays(1);
- LocalDate midnight = tomorrow.atStartOfDay().toLocalDate();
- return java.time.Duration.between(
- LocalDate.now().atTime(23, 59, 59),
- midnight.atTime(0, 0, 0)
- ).getSeconds() + 1;
+ LocalDateTime now = LocalDateTime.now();
+ LocalDateTime midnight = now.toLocalDate().plusDays(1).atStartOfDay();
+ return java.time.Duration.between(now, midnight).getSeconds();
}
/**
--
Gitblit v1.9.3