From 7d008bf58373926bbdcf67fa7bcf92510f427fb6 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Fri, 27 Mar 2026 16:57:11 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java |  260 +++++++++++++++++++++++++++------------------------
 1 files changed, 137 insertions(+), 123 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
index 4877188..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,137 +88,148 @@
             return ServerResponse.createByErrorMsg("参数错误",request);
         }
         User user = iUserService.getCurrentRefreshUser(request);
-        UserAssets userAssets =   iUserAssetsServices.assetsByTypeAndUserId("IN",user.getId());
-        if (model.getNewCode() != null) {
-            StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new QueryWrapper<StockSubscribe>()
-                    .eq("newlist_id", model.getNewlistId()));
-            //判断购买新股需不需要密钥
-            if(StringUtils.isNotEmpty(stockSubscribe.getPassword()) && !stockSubscribe.getPassword().equals(model.getPassword())){
-                return ServerResponse.createByErrorMsg("密钥输入错误",request);
-            }
-            //实名认证开关
-            SiteProduct siteProduct = iSiteProductService.getProductSetting();
-            if (siteProduct.getRealNameDisplay() && user.getIsActive() != 2) {
-                return ServerResponse.createByErrorMsg("订单失败,请先实名认证",request);
-            }
-            if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
-                return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
-            }
 
-            if ( model.getApplyNums() == null){
-                // 申请数量为空时返回错误信息
-                return ServerResponse.createByErrorMsg("申请数量不能为空", request);
-            }
-            if(model.getApplyNums() < stockSubscribe.getMinNumber() ){
-                return ServerResponse.createByErrorMsg("最少申购:" + stockSubscribe.getMinNumber(), request);
-            }
-            if (model.getApplyNums() > stockSubscribe.getOrderNumber()){
-                return ServerResponse.createByErrorMsg("最多可申购:" + stockSubscribe.getOrderNumber(), request);
-            }
+        // 使用分布式锁
+        String lockKey = "subscribe:lock:" + user.getId();
+        String lockValue = UUID.randomUUID().toString(); // 生成唯一值
 
-            BigDecimal sumPrice = stockSubscribe.getPrice().multiply(new BigDecimal(model.getApplyNums()));
-            if(userAssets.getAvailableBalance().compareTo(sumPrice) < 0){
-                return ServerResponse.createByErrorMsg("账户余额不足", request);
-            }
+        // 尝试获取锁,设置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>()
+                        .eq("newlist_id", model.getNewlistId()));
+                //判断购买新股需不需要密钥
+                if(StringUtils.isNotEmpty(stockSubscribe.getPassword()) && !stockSubscribe.getPassword().equals(model.getPassword())){
+                    return ServerResponse.createByErrorMsg("密钥输入错误",request);
+                }
+                //实名认证开关
+                SiteProduct siteProduct = iSiteProductService.getProductSetting();
+                if (siteProduct.getRealNameDisplay() && user.getIsActive() != 2) {
+                    return ServerResponse.createByErrorMsg("订单失败,请先实名认证",request);
+                }
+                if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+                    return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
+                }
 
-            Date currentDate = new Date();
-            if(currentDate.before(stockSubscribe.getSubscriptionTime())){
-                return ServerResponse.createByErrorMsg("未开售",request);
-            }
-            if(currentDate.after(stockSubscribe.getSubscribeTime())){
-                return ServerResponse.createByErrorMsg("已结束",request);
-            }
+                if ( model.getApplyNums() == null){
+                    // 申请数量为空时返回错误信息
+                    return ServerResponse.createByErrorMsg("申请数量不能为空", request);
+                }
+                if(model.getApplyNums() < stockSubscribe.getMinNumber() ){
+                    return ServerResponse.createByErrorMsg("最少申购:" + stockSubscribe.getMinNumber(), request);
+                }
+                if (model.getApplyNums() > stockSubscribe.getOrderNumber()){
+                    return ServerResponse.createByErrorMsg("最多可申购:" + stockSubscribe.getOrderNumber(), request);
+                }
 
-//            //重复申购限制
-//            UserStockSubscribe userStockSubscribe = userStockSubscribeMapper.selectOne(new QueryWrapper<UserStockSubscribe>().eq(
-//                    "new_code", model.getNewCode()).eq("user_id", user.getId()).eq("type",model.getType()));
-//            if (userStockSubscribe != null) {
-//                return ServerResponse.createByErrorMsg("请勿重复申请",request);
-//            }
-            if (siteProduct.getRealNameDisplay() && user.getIsLock().intValue() == 1) {
-                return ServerResponse.createByErrorMsg("订单失败,帐户已被锁定",request);
-            }
-            if (stockSubscribe == null) {
-                return ServerResponse.createByErrorMsg("新的股票代码不存在",request);
-            }
+                BigDecimal sumPrice = stockSubscribe.getPrice().multiply(new BigDecimal(model.getApplyNums()));
+                if(userAssets.getAvailableBalance().compareTo(sumPrice) < 0){
+                    return ServerResponse.createByErrorMsg("账户余额不足", request);
+                }
 
-            model.setApplyNums((int) (model.getApplyNums()*stockSubscribe.getLever()));
+                Date currentDate = new Date();
+                if(currentDate.before(stockSubscribe.getSubscriptionTime())){
+                    return ServerResponse.createByErrorMsg("未开售",request);
+                }
+                if(currentDate.after(stockSubscribe.getSubscribeTime())){
+                    return ServerResponse.createByErrorMsg("已结束",request);
+                }
 
-            //计算申购金额
-            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());
-            }
+                //重复申购限制
+                Long count = userStockSubscribeMapper.selectCount(new QueryWrapper<UserStockSubscribe>().eq(
+                        "new_code", model.getNewCode()).eq("user_id", user.getId()).eq("type", model.getType()));
+                if (count > 0) {
+                    return ServerResponse.createByErrorMsg("请勿重复申请",request);
+                }
+                if (siteProduct.getRealNameDisplay() && user.getIsLock().intValue() == 1) {
+                    return ServerResponse.createByErrorMsg("订单失败,帐户已被锁定",request);
+                }
+                if (stockSubscribe == null) {
+                    return ServerResponse.createByErrorMsg("新的股票代码不存在",request);
+                }
 
-            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);
+                model.setApplyNums((int) (model.getApplyNums()*stockSubscribe.getLever()));
 
-                BigDecimal subtract = userAssets.getAvailableBalance().subtract(bound);
-                if(subtract.compareTo(BigDecimal.ZERO) >= 0){
-                    userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(bound));
-                    userAssets.setAvailableBalance(subtract);
+                //计算申购金额
+                BigDecimal bound;
+                if (stockSubscribe.getMinPrice() != null && stockSubscribe.getMinPrice().compareTo(BigDecimal.ZERO) > 0) {
+                    bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getMinPrice());
                 }else{
-                    userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(userAssets.getAvailableBalance()));
-                    userAssets.setAvailableBalance(BigDecimal.ZERO);
-                    userAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(subtract.abs()));
+                    bound =  new BigDecimal(model.getApplyNums()).multiply(stockSubscribe.getPrice());
                 }
-                userAssetsMapper.updateById(userAssets);
+                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 (ret > 0) {
-                    return ServerResponse.createBySuccessMsg("申购成功",request);
-                } else {
-                    return ServerResponse.createByErrorMsg("申购失败",request);
-                }
-            }else{
+                        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 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);
+                        if (ret > 0) {
+                            return ServerResponse.createBySuccessMsg("申购成功",request);
+                        } else {
+                            return ServerResponse.createByErrorMsg("申购失败",request);
+                        }
+                    }else{
+
+                        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);
     }
@@ -268,7 +282,7 @@
                             //需要退回的资金
                             BigDecimal refundPrice = userStockSubscribe.getBuyPrice().multiply(new BigDecimal(applyNumber));
                             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(refundPrice));
-                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(refundPrice));
+                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(refundPrice));
                         }
                     }else{
                         if(applyNumber > 0){
@@ -284,9 +298,9 @@
                         }
                     }
                     ret = userStockSubscribeMapper.update1(model);
+                    userAssetsMapper.updateById(userAssets);
                     iUserPositionService.newStockToPosition(model.getId(),userAssets.getAmountToBeCovered());//转持仓
                     model.setStatus(5);
-                    userAssetsMapper.updateById(userAssets);
                 }else{
                     if(model.getApplyNumber()>model.getApplyNums()){
                         return  ServerResponse.createByErrorMsg("配置中签数量不能超过申请数量",request);

--
Gitblit v1.9.3