From 0a1d636393cca0d33305e8c0230a33c321ae08e4 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 12 Mar 2026 18:20:04 +0800
Subject: [PATCH] 1

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

diff --git a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
index bc4037e..0f9d52b 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -112,11 +112,15 @@
     public ServerResponse add(StockSubscribe model, HttpServletRequest request) {
         //判断是否已经存在
         log.info("model:{}", model);
-        Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockCode, model.getCode()).last(" limit 1"));
-        if(null == stock){
-            return ServerResponse.createByErrorMsg("新股代码不存在");
+//        Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockCode, model.getCode()).last(" limit 1"));
+//        if(null == stock){
+//            return ServerResponse.createByErrorMsg("新股代码不存在");
+//        }
+        Long count = stockSubscribeMapper.selectCount(new LambdaQueryWrapper<StockSubscribe>()
+                .eq(StockSubscribe::getCode, model.getCode()));
+        if(count > 0){
+            return ServerResponse.createByErrorMsg("新股代码已存在");
         }
-
         int resultCount = this.stockSubscribeMapper.insert(model);
         if (resultCount > 0) {
             return ServerResponse.createBySuccessMsg("添加新股成功");
@@ -137,6 +141,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);

--
Gitblit v1.9.3