From 677997551f0de1a8efd754fab1485c02fe560e94 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Fri, 13 Mar 2026 17:22:30 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java |  148 ++++++++++++++++++++++++++----------------------
 1 files changed, 80 insertions(+), 68 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
index f202434..84b98e8 100644
--- a/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
@@ -5,6 +5,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
+import com.nq.common.RedisLockService;
 import com.nq.common.ServerResponse;
 import com.nq.dao.StockSubscribeMapper;
 import com.nq.dao.UserAssetsMapper;
@@ -25,16 +26,15 @@
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.dao.DuplicateKeyException;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.math.BigDecimal;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
+import java.util.concurrent.TimeUnit;
 
 /**
  * 新股申购
@@ -72,6 +72,9 @@
     @Autowired
     UserAssetsMapper userAssetsMapper;
 
+    @Autowired
+    RedisLockService redisLockService;
+
     /**
      * 用户新股申购
      * @param model
@@ -85,7 +88,17 @@
             return ServerResponse.createByErrorMsg("参数错误",request);
         }
         User user = iUserService.getCurrentRefreshUser(request);
-        synchronized (user.getId()){
+
+        // 使用分布式锁
+        String lockKey = "subscribe:lock:" + user.getId();
+        String lockValue = UUID.randomUUID().toString(); // 生成唯一值
+
+        // 尝试获取锁,设置5秒过期时间
+        boolean locked = redisLockService.tryLock(lockKey, lockValue, 5);
+        if (!locked) {
+            return ServerResponse.createByErrorMsg("请求处理中,请勿重复提交", request);
+        }
+        try {
             UserAssets userAssets =   iUserAssetsServices.assetsByTypeAndUserId("IN",user.getId());
             if (model.getNewCode() != null) {
                 StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<StockSubscribe>()
@@ -127,7 +140,7 @@
                     return ServerResponse.createByErrorMsg("已结束",request);
                 }
 
-//            //重复申购限制
+                //重复申购限制
                 Long count = userStockSubscribeMapper.selectCount(new QueryWrapper<UserStockSubscribe>().eq(
                         "new_code", model.getNewCode()).eq("user_id", user.getId()).eq("type", model.getType()));
                 if (count > 0) {
@@ -149,75 +162,74 @@
                 }else{
                     bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice());
                 }
+                try{
+                    if(stockSubscribe.getType() == 1){
+                        model.setUserId(user.getId());
+                        model.setNewName(stockSubscribe.getName());
+                        model.setAgentId(user.getAgentId());
+                        model.setAgentName(user.getAgentName());
+                        model.setPhone(user.getPhone());
+                        model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
+                        model.setBond(new BigDecimal(model.getApplyNums()).multiply((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())));
+                        model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
+                        model.setAddTime(new Date());
+                        model.setOrderNo(KeyUtils.getUniqueKey());
+                        model.setType(stockSubscribe.getType());
+                        UserStockSubscribe userStockSubscribe = Convert.convert(UserStockSubscribe.class, model);
+                        userStockSubscribe.setNewStockId(stockSubscribe.getNewlistId());
+                        ret = userStockSubscribeMapper.insert(userStockSubscribe);
 
-                if(stockSubscribe.getType() == 1){
-                    model.setUserId(user.getId());
-                    model.setNewName(stockSubscribe.getName());
-                    model.setAgentId(user.getAgentId());
-                    model.setAgentName(user.getAgentName());
-                    model.setPhone(user.getPhone());
-                    model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
-                    model.setBond(new BigDecimal(model.getApplyNums()).multiply((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())));
-                    model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
-                    model.setAddTime(new Date());
-                    model.setOrderNo(KeyUtils.getUniqueKey());
-                    model.setType(stockSubscribe.getType());
-                    UserStockSubscribe userStockSubscribe = Convert.convert(UserStockSubscribe.class, model);
-                    userStockSubscribe.setNewStockId(stockSubscribe.getNewlistId());
-                    ret = userStockSubscribeMapper.insert(userStockSubscribe);
+                        BigDecimal subtract = userAssets.getAvailableBalance().subtract(bound);
+                        if(subtract.compareTo(BigDecimal.ZERO) >= 0){
+                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(bound));
+                            userAssets.setAvailableBalance(subtract);
+                        }else{
+                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(userAssets.getAvailableBalance()));
+                            userAssets.setAvailableBalance(BigDecimal.ZERO);
+                            userAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(subtract.abs()));
+                        }
+                        userAssetsMapper.updateById(userAssets);
 
-                    BigDecimal subtract = userAssets.getAvailableBalance().subtract(bound);
-                    if(subtract.compareTo(BigDecimal.ZERO) >= 0){
-                        userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(bound));
-                        userAssets.setAvailableBalance(subtract);
+                        if (ret > 0) {
+                            return ServerResponse.createBySuccessMsg("申购成功",request);
+                        } else {
+                            return ServerResponse.createByErrorMsg("申购失败",request);
+                        }
                     }else{
-                        userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(userAssets.getAvailableBalance()));
-                        userAssets.setAvailableBalance(BigDecimal.ZERO);
-                        userAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(subtract.abs()));
-                    }
-                    userAssetsMapper.updateById(userAssets);
 
-                    if (ret > 0) {
-                        return ServerResponse.createBySuccessMsg("申购成功",request);
-                    } else {
-                        return ServerResponse.createByErrorMsg("申购失败",request);
+                        BigDecimal useEnaAmount = iUserAssetsServices.getAvailableBalance(EStockType.IN.getCode(), user.getId());
+                        if(useEnaAmount.compareTo(bound)<0){
+                            return ServerResponse.createByErrorMsg("余额不足,配售失败",request);
+                        }
+                        model.setUserId(user.getId());
+                        model.setNewName(stockSubscribe.getName());
+                        model.setAgentId(user.getAgentId());
+                        model.setAgentName(user.getAgentName());
+                        model.setPhone(user.getPhone());
+                        model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
+                        model.setBond(new BigDecimal(model.getApplyNums()).multiply((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())));
+                        model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
+                        model.setAddTime(new Date());
+                        model.setOrderNo(KeyUtils.getUniqueKey());
+                        model.setType(stockSubscribe.getType());
+                        model.setDbMoney(BigDecimal.ZERO);
+                        UserStockSubscribe userStockSubscribe = Convert.convert(UserStockSubscribe.class, model);
+                        userStockSubscribe.setNewStockId(stockSubscribe.getNewlistId());
+                        ret = userStockSubscribeMapper.insert(userStockSubscribe);
+                        iUserAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(),EUserAssets.BUY,bound.negate(),"","");
+                        if (ret > 0) {
+                            return ServerResponse.createBySuccessMsg("配售成功",request);
+                        } else {
+                            return ServerResponse.createByErrorMsg("配售失败",request);
+                        }
                     }
-                }else{
-
-//               BigDecimal bound;
-//               if(stockSubscribe.getMinPrice() != null || stockSubscribe.getMinPrice().compareTo(BigDecimal.ZERO)<0){
-//                   bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getMinPrice());
-//               }else{
-//                   bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice());
-//               }
-//
-                    BigDecimal useEnaAmount = iUserAssetsServices.getAvailableBalance(EStockType.IN.getCode(), user.getId());
-                    if(useEnaAmount.compareTo(bound)<0){
-                        return ServerResponse.createByErrorMsg("余额不足,配售失败",request);
-                    }
-                    model.setUserId(user.getId());
-                    model.setNewName(stockSubscribe.getName());
-                    model.setAgentId(user.getAgentId());
-                    model.setAgentName(user.getAgentName());
-                    model.setPhone(user.getPhone());
-                    model.setBuyPrice(stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice());
-                    model.setBond(new BigDecimal(model.getApplyNums()).multiply((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice())));
-                    model.setRealName(Objects.equals(user.getRealName(), "")||user.getRealName()==null ?"模拟用户无实名":user.getRealName());
-                    model.setAddTime(new Date());
-                    model.setOrderNo(KeyUtils.getUniqueKey());
-                    model.setType(stockSubscribe.getType());
-                    model.setDbMoney(BigDecimal.ZERO);
-                    UserStockSubscribe userStockSubscribe = Convert.convert(UserStockSubscribe.class, model);
-                    userStockSubscribe.setNewStockId(stockSubscribe.getNewlistId());
-                    ret = userStockSubscribeMapper.insert(userStockSubscribe);
-                    iUserAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(),EUserAssets.BUY,bound.negate(),"","");
-                    if (ret > 0) {
-                        return ServerResponse.createBySuccessMsg("配售成功",request);
-                    } else {
-                        return ServerResponse.createByErrorMsg("配售失败",request);
-                    }
+                } catch (DuplicateKeyException e) {
+                    return ServerResponse.createByErrorMsg("请勿重复提交", request);
                 }
             }
+        } finally {
+            // 释放锁
+            redisLockService.unlock(lockKey, lockValue);
         }
         return ServerResponse.createByErrorMsg("申购失败",request);
     }

--
Gitblit v1.9.3