From c8b031a405ebd2b450f5098bb37e5e63390f364c Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Fri, 18 Jul 2025 09:57:31 +0800
Subject: [PATCH] 新增贷款,优化

---
 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java |   65 +++++++++++++++++++++++++++++---
 1 files changed, 58 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
index cc11dbf..61d9db3 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -4,15 +4,17 @@
 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.PageHelper;
 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.UserStockSubscribe;
+import com.nq.pojo.*;
 import com.nq.service.IStockSubscribeService;
 import com.nq.dao.StockSubscribeMapper;
+import com.nq.service.IUserService;
+import com.nq.vo.stock.ai.StockAiOrderTypeVO;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -40,6 +42,11 @@
 
     @Autowired
     StockMapper stockMapper;
+
+    @Autowired
+    UserPositionMapper userPositionMapper;
+    @Autowired
+    IUserService iUserService;
 
     /**
      * @Description: 用户新股列表
@@ -107,11 +114,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("添加新股成功");
@@ -132,6 +143,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);
@@ -142,6 +159,11 @@
         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("修改新股成功");
@@ -170,6 +192,35 @@
         return ServerResponse.createByErrorMsg("删除新股失败");
     }
 
+    @Override
+    public ServerResponse listPageInfo(int pageNum, int pageSize, HttpServletRequest request) {
+        try {
+            QueryWrapper<StockSubscribe> queryWrapper = new QueryWrapper();
+            //显示 0
+            queryWrapper.eq("zt", 0);
+            queryWrapper.eq("type", 1);
+            queryWrapper.orderByDesc("list_date");
+
+            PageHelper.startPage(pageNum, pageSize);
+            List<StockSubscribe> stockSubscribeList = this.stockSubscribeMapper.selectList(queryWrapper);
+            // 获取分页信息
+            PageInfo<StockSubscribe> pageInfo = new PageInfo<>(stockSubscribeList);
+
+            Date newDate = new Date();
+            pageInfo.getList().forEach(stockSubscribe -> {
+                Date startDate = stockSubscribe.getSubscribeTime();
+                Date endDate = stockSubscribe.getSubscriptionTime();
+
+                boolean isBetween = newDate.after(startDate) && newDate.before(endDate);
+                stockSubscribe.setIsBetween(isBetween);
+            });
+            return ServerResponse.createBySuccess(pageInfo);
+        } catch (Exception e) {
+            log.error("IStockSubscribeService listPageInfo error", e);
+        }
+        return ServerResponse.createByError();
+    }
+
 }
 
 

--
Gitblit v1.9.3