From fd97ec2db0324c7884eb21089dd0b7d9764f3ea0 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Sat, 18 May 2024 11:06:48 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java |   14 +++++++++++++-
 1 files changed, 13 insertions(+), 1 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..a4a35e3 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -83,9 +83,15 @@
     public ServerResponse add(StockSubscribe model, HttpServletRequest request) {
         //判断是否已经存在
         log.info("model:{}", model);
-        Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockSpell, model.getCode()).last(" limit 1"));
+        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())
+                .eq(StockSubscribe::getType, model.getType()));
+        if(count > 0){
+            return ServerResponse.createByErrorMsg("新股已存在");
         }
         int resultCount = this.stockSubscribeMapper.insert(model);
         if (resultCount > 0) {
@@ -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