From e89e71158de719c70c487110a0f8d00d4a30d30b Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Thu, 12 Feb 2026 11:02:48 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java |   66 ++++++++++++++++++++++++++++++---
 1 files changed, 60 insertions(+), 6 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..0f9d52b 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -1,13 +1,18 @@
 package com.nq.service.impl;
 
+import cn.hutool.core.convert.Convert;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 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;
 import lombok.extern.slf4j.Slf4j;
@@ -16,6 +21,7 @@
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
+import javax.xml.stream.events.EntityDeclaration;
 import java.util.Date;
 import java.util.List;
 
@@ -32,7 +38,13 @@
     StockSubscribeMapper stockSubscribeMapper;
 
     @Autowired
+    UserStockSubscribeMapper userStockSubscribeMapper;
+
+    @Autowired
     StockMapper stockMapper;
+
+    @Autowired
+    UserPositionMapper userPositionMapper;
 
     /**
      * @Description: 用户新股列表
@@ -42,7 +54,7 @@
      * @Date: 2022/10/25
      */
     @Override
-    public ServerResponse list(String name, String code, Integer zt,Integer getType, Integer isLock, Integer type, HttpServletRequest request) {
+    public ServerResponse list(String name, String code, Integer zt,Integer getType, Integer isLock, Integer type, Integer stockType,Integer postType,HttpServletRequest request) {
 
         //name和code模糊查询
         QueryWrapper<StockSubscribe> queryWrapper = new QueryWrapper();
@@ -61,11 +73,28 @@
         if (type != null && !type.equals("")) {
             queryWrapper.eq("type", type);
         }
-        if(getType == 1){
-            queryWrapper.gt("list_date", new Date());
+//        if(getType == 1){
+//            queryWrapper.gt("list_date", new Date());
+//        }
+        if(stockType != null){
+            queryWrapper.eq("stock_type", stockType);
         }
         queryWrapper.orderByDesc("newlist_id");
+//        if(postType == 0){
+//            queryWrapper.select(StockSubscribe.class, info -> !info.getColumn().equals("password"));
+//        }
         List<StockSubscribe> stockSubscribeList = this.stockSubscribeMapper.selectList(queryWrapper);
+
+        Date newDate = new Date();
+
+        for (StockSubscribe stockSubscribe : stockSubscribeList) {
+            Date startDate = stockSubscribe.getSubscribeTime();
+            Date endDate = stockSubscribe.getSubscriptionTime();
+
+            boolean isBetween = newDate.after(startDate) && newDate.before(endDate);
+            stockSubscribe.setIsBetween(isBetween);
+        }
+
         PageInfo pageInfo = new PageInfo(stockSubscribeList);
         pageInfo.setList(stockSubscribeList);
 
@@ -83,9 +112,14 @@
     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"));
-        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) {
@@ -107,8 +141,28 @@
         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);
+        //修改用户申购的新股代码
+        List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper
+                .selectList(new LambdaQueryWrapper<UserStockSubscribe>()
+                        .eq(UserStockSubscribe::getNewStockId, stockSubscribe.getNewlistId()));
+        for (UserStockSubscribe u : userStockSubscribes) {
+            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