From 752c5bbdf26f7eaa56f25a284eacb6eb37ed15c0 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 11 Jun 2025 18:04:47 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java |   16 ++++++++++++++++
 1 files changed, 16 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
index cc11dbf..4d7613f 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -7,9 +7,11 @@
 import com.github.pagehelper.PageInfo;
 import com.nq.common.ServerResponse;
 import com.nq.dao.StockMapper;
+import com.nq.dao.UserPositionMapper;
 import com.nq.dao.UserStockSubscribeMapper;
 import com.nq.pojo.Stock;
 import com.nq.pojo.StockSubscribe;
+import com.nq.pojo.UserPosition;
 import com.nq.pojo.UserStockSubscribe;
 import com.nq.service.IStockSubscribeService;
 import com.nq.dao.StockSubscribeMapper;
@@ -40,6 +42,9 @@
 
     @Autowired
     StockMapper stockMapper;
+
+    @Autowired
+    UserPositionMapper userPositionMapper;
 
     /**
      * @Description: 用户新股列表
@@ -132,6 +137,12 @@
         if (stockSubscribe == null) {
             return ServerResponse.createByErrorMsg("新股不存在");
         }
+        Long count = stockSubscribeMapper.selectCount(new LambdaQueryWrapper<StockSubscribe>()
+                .eq(StockSubscribe::getCode, model.getCode())
+                .ne(StockSubscribe::getNewlistId,model.getNewlistId()));
+        if(count > 0){
+            return ServerResponse.createByErrorMsg("新股代码已存在");
+        }
 
         //修改新股
         int resultCount = this.stockSubscribeMapper.updateById(model);
@@ -143,6 +154,11 @@
             u.setNewCode(model.getCode());
             userStockSubscribeMapper.updateById(u);
         }
+        List<UserPosition> userPositions = userPositionMapper.selectList(new LambdaQueryWrapper<UserPosition>().eq(UserPosition::getNewId, stockSubscribe.getNewlistId()));
+        for (UserPosition u : userPositions) {
+            u.setStockCode(model.getCode());
+            userPositionMapper.updateById(u);
+        }
         if (resultCount > 0) {
             return ServerResponse.createBySuccessMsg("修改新股成功");
         }

--
Gitblit v1.9.3