From 087d32825071725761ab65b7f241cae1848becff Mon Sep 17 00:00:00 2001
From: liyong <18434998025@163.com>
Date: 星期四, 21 五月 2026 13:25:38 +0800
Subject: [PATCH] refactor(product): 优化产品型号反向新增逻辑并添加分布式锁
---
src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java | 163 +++++++++++++++++++++++++++++++++++++++++++----------
1 files changed, 131 insertions(+), 32 deletions(-)
diff --git a/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java b/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
index 55bfdce..7f7a44d 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
@@ -24,12 +24,15 @@
import com.ruoyi.sales.mapper.SalesLedgerProductMapper;
import com.ruoyi.sales.pojo.SalesLedgerProduct;
import lombok.AllArgsConstructor;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import javax.validation.constraints.NotNull;
import java.util.*;
+import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
/**
@@ -41,6 +44,11 @@
@Service
@AllArgsConstructor
public class ProductModelServiceImpl extends ServiceImpl<ProductModelMapper, ProductModel> implements IProductModelService {
+ private static final String PRODUCT_MODEL_LOCK_PREFIX = "product_model_anticlockwise:";
+ private static final long LOCK_WAIT_TIMEOUT_SECONDS = 10L;
+ private static final long LOCK_EXPIRE_TIME_SECONDS = 30L;
+ private static final String FINISHED_PRODUCT_NAME = "\u6210\u54c1";
+ private final RedisTemplate redisTemplate;
private final ProductMapper productMapper;
private final SalesLedgerProductMapper salesLedgerProductMapper;
@@ -168,45 +176,136 @@
//鍙嶅悜鏂板鎴愬搧浜у搧锛屽彧鏈夐攢鍞叧鑱旀柊澧炵殑鏃跺�欒皟鐢�
@Override
+ @Transactional(rollbackFor = Exception.class)
public Long productModelAnticlockwise(ProductModelAnticlockwiseDto productModelDto) {
- ProductModel oldProductModel = productModelMapper.selectOldProductModel(productModelDto.getModel(), productModelDto.getProductName());
- //瀛樺湪灏辨洿鏂�
- if (oldProductModel != null) {
- oldProductModel.setModel(productModelDto.getModel());
- oldProductModel.setUnit(productModelDto.getUnit());
- oldProductModel.setSubUnit(productModelDto.getSubUnit());
- oldProductModel.setDeptId(SecurityUtils.getDeptId()[0]);
- productModelMapper.updateById(oldProductModel);
- Product product = productMapper.selectById(oldProductModel.getProductId());
- product.setProductName(productModelDto.getProductName());
- productMapper.updateById(product);
- return oldProductModel.getId();
- }else {
- //鎵惧埌鐖惰妭鐐�
- Product productParent = productMapper.selectOne(new QueryWrapper<Product>().lambda().eq(Product::getProductName, "鎴愬搧").last("limit 1"));
- if (ObjectUtils.isEmpty(productParent)) {
- Product product = new Product();
- product.setProductName("鎴愬搧");
- product.setDeptId(SecurityUtils.getDeptId()[0]);
- productMapper.insert(product);
- productParent.setId(product.getId());
- }
- //鏂板浜у搧澶х被
- Product product = new Product();
- product.setProductName(productModelDto.getProductName());
- product.setParentId(productParent.getId());
- product.setDeptId(SecurityUtils.getDeptId()[0]);
- productMapper.insert( product);
- //鏂板浜у搧瑙勬牸
+ Long deptId = SecurityUtils.getDeptId()[0];
+
+ // 濡傛灉鎻愪緵浜咺D锛岀洿鎺ユ洿鏂�
+ if (ObjectUtils.isNotEmpty(productModelDto.getId())) {
+ return updateProductModelById(productModelDto, deptId);
+ }
+
+ // 妫�鏌ユ槸鍚﹀凡瀛樺湪鐩稿悓鐨勪骇鍝佸瀷鍙�
+ ProductModel existingModel = productModelMapper.selectOldProductModel(
+ productModelDto.getModel(), productModelDto.getProductName()
+ );
+ if (existingModel != null) {
+ return updateExistingProductModel(existingModel, productModelDto, deptId);
+ }
+
+ // 鏂板鏃堕渶瑕佸姞閿侀槻姝㈠苟鍙戝垱寤�
+ String lockKey = buildProductModelLockKey(FINISHED_PRODUCT_NAME,
+ productModelDto.getProductName(), productModelDto.getModel());
+ String lockValue = UUID.randomUUID().toString();
+ acquireLock(lockKey, lockValue);
+ try {
+ Long productId = getOrCreateProduct(productModelDto.getProductName(), deptId);
+
ProductModel productModel = new ProductModel();
- productModel.setProductId(product.getId());
+ productModel.setProductId(productId);
productModel.setModel(productModelDto.getModel());
productModel.setUnit(productModelDto.getUnit());
productModel.setSubUnit(productModelDto.getSubUnit());
- productModel.setDeptId(SecurityUtils.getDeptId()[0]);
+ productModel.setDeptId(deptId);
productModelMapper.insert(productModel);
return productModel.getId();
-
+ } finally {
+ releaseLock(lockKey, lockValue);
}
}
+
+ private Long updateProductModelById(ProductModelAnticlockwiseDto productModelDto, Long deptId) {
+ ProductModel productModel = productModelMapper.selectById(productModelDto.getId());
+ if (productModel == null) {
+ throw new ServiceException("浜у搧鍨嬪彿涓嶅瓨鍦�");
+ }
+ return updateExistingProductModel(productModel, productModelDto, deptId);
+ }
+
+ private Long updateExistingProductModel(ProductModel productModel, ProductModelAnticlockwiseDto dto, Long deptId) {
+ productModel.setModel(dto.getModel());
+ productModel.setUnit(dto.getUnit());
+ productModel.setSubUnit(dto.getSubUnit());
+ productModel.setDeptId(deptId);
+ productModelMapper.updateById(productModel);
+
+ updateProductName(productModel.getProductId(), dto.getProductName(), deptId);
+ return productModel.getId();
+ }
+
+ private Long getOrCreateProduct(String productName, Long deptId) {
+ // 鑾峰彇鎴栧垱寤虹埗浜у搧"鎴愬搧"
+ Product parentProduct = getOrCreateParentProduct(FINISHED_PRODUCT_NAME, deptId);
+
+ // 鏌ユ壘鎴栧垱寤哄叿浣撲骇鍝�
+ Product product = productMapper.selectOne(new QueryWrapper<Product>().lambda()
+ .eq(Product::getProductName, productName)
+ .eq(Product::getParentId, parentProduct.getId())
+ .last("limit 1"));
+
+ if (product == null) {
+ product = new Product();
+ product.setProductName(productName);
+ product.setParentId(parentProduct.getId());
+ product.setDeptId(deptId);
+ productMapper.insert(product);
+ }
+ return product.getId();
+ }
+
+ private Product getOrCreateParentProduct(String parentName, Long deptId) {
+ Product parentProduct = productMapper.selectOne(new QueryWrapper<Product>().lambda()
+ .eq(Product::getProductName, parentName)
+ .last("limit 1"));
+
+ if (parentProduct == null) {
+ parentProduct = new Product();
+ parentProduct.setProductName(parentName);
+ parentProduct.setDeptId(deptId);
+ productMapper.insert(parentProduct);
+ }
+ return parentProduct;
+ }
+
+ private void updateProductName(Long productId, String productName, Long deptId) {
+ Product product = productMapper.selectById(productId);
+ if (product != null) {
+ product.setProductName(productName);
+ product.setDeptId(deptId);
+ productMapper.updateById(product);
+ }
+ }
+
+ private String buildProductModelLockKey(String parentName, String productName, String model) {
+ return PRODUCT_MODEL_LOCK_PREFIX + parentName + ":" + productName + ":" + model;
+ }
+
+ private void acquireLock(String lockKey, String lockValue) {
+ long startWaitTime = System.currentTimeMillis();
+ while (System.currentTimeMillis() - startWaitTime < LOCK_WAIT_TIMEOUT_SECONDS * 1000) {
+ Boolean locked = redisTemplate.opsForValue().setIfAbsent(
+ lockKey, lockValue, LOCK_EXPIRE_TIME_SECONDS, TimeUnit.SECONDS
+ );
+ if (Boolean.TRUE.equals(locked)) {
+ return;
+ }
+ try {
+ Thread.sleep(100L);
+ } catch (InterruptedException e) {
+ Thread.currentThread().interrupt();
+ throw new ServiceException("Get product model lock interrupted");
+ }
+ }
+ throw new ServiceException("Get product model lock timeout");
+ }
+
+ private void releaseLock(String lockKey, String lockValue) {
+ String luaScript = "if redis.call('GET', KEYS[1]) == ARGV[1] then return redis.call('DEL', KEYS[1]) else return 0 end";
+ redisTemplate.execute(
+ new DefaultRedisScript<>(luaScript, Long.class),
+ Collections.singletonList(lockKey),
+ lockValue
+ );
+ }
+
}
--
Gitblit v1.9.3