From f1efb265c177508ccf57c9e2aa5573037272a1b7 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 25 Apr 2024 19:41:54 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java |   12 ++++++++++++
 1 files changed, 12 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 bf3c632..d520922 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -87,6 +87,12 @@
         if(null == stock){
             return ServerResponse.createByErrorMsg("新股代码不存在");
         }
+        Long count = stockSubscribeMapper.selectCount(new LambdaQueryWrapper<StockSubscribe>()
+                .eq(StockSubscribe::getCode, model.getCode())
+                .eq(StockSubscribe::getType, model.getType()));
+        if(count > 0){
+            return ServerResponse.createByErrorMsg("新股已存在");
+        }
         int resultCount = this.stockSubscribeMapper.insert(model);
         if (resultCount > 0) {
             return ServerResponse.createBySuccessMsg("添加新股成功");
@@ -107,6 +113,12 @@
         if (stockSubscribe == null) {
             return ServerResponse.createByErrorMsg("新股不存在");
         }
+        Long count = stockSubscribeMapper.selectCount(new LambdaQueryWrapper<StockSubscribe>()
+                .eq(StockSubscribe::getCode, model.getCode())
+                .eq(StockSubscribe::getType, model.getType()).ne(StockSubscribe::getNewlistId,model.getNewlistId()));
+        if(count > 0){
+            return ServerResponse.createByErrorMsg("新股已存在");
+        }
         //修改新股
         int resultCount = this.stockSubscribeMapper.updateById(model);
         if (resultCount > 0) {

--
Gitblit v1.9.3