From f7cc4ee83280e7f9ab3b931de37ca0698ed33f67 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 25 Apr 2024 20:09:21 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java |   18 +++++++++++++++++-
 1 files changed, 17 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 6e9b5bd..d520922 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -16,6 +16,7 @@
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
+import java.util.Date;
 import java.util.List;
 
 /**
@@ -41,7 +42,7 @@
      * @Date: 2022/10/25
      */
     @Override
-    public ServerResponse list(String name, String code, Integer zt, Integer isLock, Integer type, HttpServletRequest request) {
+    public ServerResponse list(String name, String code, Integer zt,Integer getType, Integer isLock, Integer type, HttpServletRequest request) {
 
         //name和code模糊查询
         QueryWrapper<StockSubscribe> queryWrapper = new QueryWrapper();
@@ -59,6 +60,9 @@
         }
         if (type != null && !type.equals("")) {
             queryWrapper.eq("type", type);
+        }
+        if(getType == 1){
+            queryWrapper.gt("list_date", new Date());
         }
         queryWrapper.orderByDesc("newlist_id");
         List<StockSubscribe> stockSubscribeList = this.stockSubscribeMapper.selectList(queryWrapper);
@@ -83,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("添加新股成功");
@@ -103,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