From 0789083f63c2d80a41fcc08d88d3dc9b254337ae Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 10 Oct 2024 14:46:32 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserPositionServiceImpl.java |   12 +++++++-----
 1 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java b/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
index a1000fa..fc2eea5 100644
--- a/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserPositionServiceImpl.java
@@ -336,7 +336,8 @@
                 .eq(UserAssets::getUserId, userPosition.getUserId())
                 .eq(UserAssets::getAccectType, "IN")
         );
-        if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+        Long count = userStockSubscribeMapper.selectCount(new LambdaQueryWrapper<>(UserStockSubscribe.class).eq(UserStockSubscribe::getNewCode, userPosition.getStockCode()));
+        if(count > 0 && userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
             return ServerResponse.createByErrorMsg("请先缴清待补资金", request);
         }
         Boolean b = tradingHourService.timeCheck(userPosition.getStockCode());
@@ -344,7 +345,7 @@
             return ServerResponse.createByErrorMsg("订单失败,不在交易时间之内", request);
         }
         if(userPosition.getPositionType() == 3){
-            StockDz stockDz = stockDzMapper.selectOne(new LambdaQueryWrapper<StockDz>().eq(StockDz::getStockCode, userPosition.getStockCode()));
+            StockDz stockDz = stockDzMapper.selectOne(new LambdaQueryWrapper<StockDz>().eq(StockDz::getId, userPosition.getDzId()));
             LocalDateTime buyOrderLocalDateTime = LocalDateTime.ofInstant(userPosition.getBuyOrderTime().toInstant(), ZoneId.systemDefault());
             // 计算天数差
             long daysBetween = ChronoUnit.DAYS.between(buyOrderLocalDateTime, LocalDateTime.now());
@@ -1637,7 +1638,7 @@
 
         //判断审核开关
         if(stockDz.getSwitchType() == 1){
-            UserPosition userPosition = getUserPosition(num, user, stockDz, nowPrice, stock, buyAmt);
+            UserPosition userPosition = getUserPosition(dzId,num, user, stockDz, nowPrice, stock, buyAmt);
             UserPositionCheckDz userPositionCheckDz = Convert.convert(UserPositionCheckDz.class, userPosition);
             userPositionCheckDz.setDzId(dzId);
             userPositionCheckDzService.save(userPositionCheckDz);
@@ -1645,13 +1646,13 @@
         }
 
         // 创建UserPosition对象
-        UserPosition userPosition = getUserPosition(num, user, stockDz, nowPrice, stock, buyAmt);
+        UserPosition userPosition = getUserPosition(dzId,num, user, stockDz, nowPrice, stock, buyAmt);
         userPositionMapper.insert(userPosition);
         userAssetsServices.availablebalanceChange(EStockType.IN.getCode(), user.getId(), EUserAssets.BUY, buyAmt.negate(),"","");
         return ServerResponse.createBySuccess("购买成功", request);
     }
 
-    private UserPosition getUserPosition(Integer num, User user, StockDz stockDz, BigDecimal nowPrice, Stock stock, BigDecimal buyAmt) {
+    private UserPosition getUserPosition(Integer dzId,Integer num, User user, StockDz stockDz, BigDecimal nowPrice, Stock stock, BigDecimal buyAmt) {
         UserPosition userPosition = new UserPosition();
         userPosition.setPositionType(3);
         userPosition.setPositionSn(KeyUtils.getUniqueKey());
@@ -1681,6 +1682,7 @@
         userPosition.setOrderStayDays(Integer.valueOf(0));
         userPosition.setOrderStayFee(new BigDecimal("0"));
         userPosition.setOrderSpread(BigDecimal.ZERO);
+        userPosition.setDzId(dzId);
         return userPosition;
     }
 

--
Gitblit v1.9.3