From 59877a4089ab84920fcd2d2d4982d032dc09afc6 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Fri, 13 Mar 2026 16:52:26 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java |  295 ++++++++++++++++++++++++++++++++++++----------------------
 1 files changed, 183 insertions(+), 112 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/UserStockSubscribeServiceImpl.java
index 538a1b0..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,104 +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(new Date().before(stockSubscribe.getSubscribeTime()) && new Date().after(stockSubscribe.getSubscribeTime()) ){
-                return ServerResponse.createByErrorMsg("不在认缴或配售时间之内",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);
-            }
+        // 使用分布式锁
+        String lockKey = "subscribe:lock:" + user.getId();
+        String lockValue = UUID.randomUUID().toString(); // 生成唯一值
 
-            if ( model.getApplyNums() == null){
-                // 申请数量为空时返回错误信息
-                return ServerResponse.createByErrorMsg("申请数量不能为空", request);
-            }
-
-            if (model.getApplyNums() > stockSubscribe.getOrderNumber()){
-                // 申请数量大于订单数量时返回错误信息
-                return ServerResponse.createByErrorMsg("最多可申购:" + stockSubscribe.getOrderNumber(), 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);
-                if (ret > 0) {
-                    return ServerResponse.createBySuccessMsg("申购成功",request);
-                } else {
-                    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);
                 }
-            }else{
+                //实名认证开关
+                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);
+                }
 
-               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());
-               }
+                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);
+                }
 
-               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);
+                BigDecimal sumPrice = stockSubscribe.getPrice().multiply(new BigDecimal(model.getApplyNums()));
+                if(userAssets.getAvailableBalance().compareTo(sumPrice) < 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);
+                }
+
+                //重复申购限制
+                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);
+                }
+
+                model.setApplyNums((int) (model.getApplyNums()*stockSubscribe.getLever()));
+
+                //计算申购金额
+                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());
+                }
+                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);
+
+                        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);
+
+                        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);
     }
@@ -197,7 +244,7 @@
      */
     @Override
     @Transactional
-    public ServerResponse save(UserStockSubscribe model, HttpServletRequest request) {
+    public ServerResponse save(UserStockSubscribe model, HttpServletRequest request) throws Exception {
         int ret = 0;
         if( model.getId() != null  ){
             if (model.getStatus() == 3||model.getStatus() == 2) {
@@ -215,26 +262,45 @@
             if((model.getStatus() == 3 && model.getApplyNumber() == null) || (model.getStatus() == 3 && model.getApplyNumber() == 0) ){
                 return ServerResponse.createByErrorMsg("中签数量不能小于0");
             }
+            if(model.getStatus() == 3 && model.getApplyNumber()>userStockSubscribe.getApplyNums()){
+                return  ServerResponse.createByErrorMsg("配置中签数量不能超过申请数量",request);
+            }
+
+            //客户中签直接扣除客户账户可用资金
+            UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId("IN", userStockSubscribe.getUserId());
             if (model.getStatus() == 3 && model.getApplyNumber() != null){
                 if(stockSubscribe.getType() == 1){
                     model.setBond((stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice()).multiply(BigDecimal.valueOf(model.getApplyNumber())));
                     model.setDbMoney(model.getDbMoney());
-                    //客户中签直接扣除客户账户可用资金
-                    UserAssets userAssets = iUserAssetsServices.assetsByTypeAndUserId("IN", userStockSubscribe.getUserId());
+
                     if(null == userAssets){
                         return ServerResponse.createByErrorMsg("客户资金账户不存在");
                     }
-                    BigDecimal decimal = userStockSubscribe.getBuyPrice().multiply(new BigDecimal(model.getApplyNumber()));//中签总价
-                    BigDecimal availableBalance = userAssets.getAvailableBalance().subtract(decimal);//剩余可用余额
-                    if(decimal.compareTo(userAssets.getAvailableBalance()) <= 0){
-                        userAssets.setAvailableBalance(availableBalance);
+                    int applyNumber = userStockSubscribe.getApplyNums() - model.getApplyNumber();
+                    if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) <= 0){
+                        if(applyNumber > 0){
+                            //需要退回的资金
+                            BigDecimal refundPrice = userStockSubscribe.getBuyPrice().multiply(new BigDecimal(applyNumber));
+                            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(refundPrice));
+                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(refundPrice));
+                        }
                     }else{
-                        BigDecimal amountToBeCovered = decimal.subtract(userAssets.getAvailableBalance());//待补金额
-                        userAssets.setAvailableBalance(BigDecimal.ZERO);
-                        userAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(amountToBeCovered));
+                        if(applyNumber > 0){
+                            BigDecimal refundPrice = userStockSubscribe.getBuyPrice().multiply(new BigDecimal(applyNumber));
+                            BigDecimal subtract = refundPrice.subtract(userAssets.getAmountToBeCovered());
+                            if(subtract.compareTo(BigDecimal.ZERO) <= 0){//如果退回的资金不足补足待补则直接减
+                                userAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().subtract(refundPrice));
+                            }else{//如果退回的资金足够填补待补资金还有剩余,则返回余额
+                                userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(refundPrice).add(userAssets.getAmountToBeCovered()));
+                                userAssets.setAmountToBeCovered(BigDecimal.ZERO);
+                                userAssets.setAvailableBalance(subtract);
+                            }
+                        }
                     }
-                    userAssetsMapper.updateById(userAssets);
                     ret = userStockSubscribeMapper.update1(model);
+                    userAssetsMapper.updateById(userAssets);
+                    iUserPositionService.newStockToPosition(model.getId(),userAssets.getAmountToBeCovered());//转持仓
+                    model.setStatus(5);
                 }else{
                     if(model.getApplyNumber()>model.getApplyNums()){
                         return  ServerResponse.createByErrorMsg("配置中签数量不能超过申请数量",request);
@@ -249,9 +315,15 @@
                     ret = userStockSubscribeMapper.update1(model);
                 }
             }else if (model.getStatus() == 2){
+                BigDecimal bond = (stockSubscribe.getMinPrice() != null ? stockSubscribe.getMinPrice() : stockSubscribe.getPrice()).multiply(BigDecimal.valueOf(model.getApplyNums()));
+                BigDecimal subtractMoney = bond.subtract(userAssets.getAmountToBeCovered());//需要退回的钱
+                userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(subtractMoney.abs()));
+                userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(subtractMoney));
+                userAssets.setAmountToBeCovered(BigDecimal.ZERO);
+                userAssetsMapper.updateById(userAssets);
                 ret = userStockSubscribeMapper.update1(model);
             }else if (model.getStatus() == 5){
-                    return iUserPositionService.newStockToPosition(model.getId());
+                    return iUserPositionService.newStockToPosition(model.getId(),BigDecimal.ZERO);
 
             }
                 if(ret>0 && model.getStatus() == 3 ){
@@ -297,8 +369,6 @@
                     }
 
                 }
-
-
         } else{
             if(model.getPhone() != null&&model.getId()==null) {
                 User user = userMapper.findByPhone(model.getPhone());
@@ -324,7 +394,8 @@
                 model.setAddTime(DateTimeUtil.getCurrentDate());
                 model.setOrderNo(KeyUtils.getUniqueKey());
                 model.setType(stockSubscribe.getType());
-                ret = userStockSubscribeMapper.insert(model);
+                UserStockSubscribeAddIn convert = Convert.convert(UserStockSubscribeAddIn.class, model);
+                return insert(convert,request);
             }
         }
         if(ret>0){
@@ -377,9 +448,9 @@
 
     /*新股申购-查询列表*/
     @Override
-    public ServerResponse<PageInfo> getList(int pageNum, int pageSize, String keyword, HttpServletRequest request){
+    public ServerResponse<PageInfo> getList(int pageNum, int pageSize, String keyword, String agentId,HttpServletRequest request){
         PageHelper.startPage(pageNum, pageSize);
-        List<UserStockSubscribe> listData = this.userStockSubscribeMapper.pageList(pageNum, pageSize, keyword);
+        List<UserStockSubscribe> listData = this.userStockSubscribeMapper.pageList(pageNum, pageSize, keyword,agentId);
         PageInfo pageInfo = new PageInfo(listData);
         pageInfo.setList(listData);
         return ServerResponse.createBySuccess(pageInfo);

--
Gitblit v1.9.3