From 6edf5438c56814b26bf4308286ebe26ac93ccb5c Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 13 Jan 2026 19:24:30 +0800
Subject: [PATCH] 1
---
src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java | 74 ++++++++++++++++++++++++++++++++-----
1 files changed, 64 insertions(+), 10 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..b57068e 100644
--- a/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -1,25 +1,26 @@
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.PageHelper;
import com.github.pagehelper.PageInfo;
+import com.nq.common.ResponseCode;
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.subscribe.StockSubscribeVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
-import javax.xml.stream.events.EntityDeclaration;
import java.util.Date;
import java.util.List;
@@ -39,7 +40,9 @@
UserStockSubscribeMapper userStockSubscribeMapper;
@Autowired
- StockMapper stockMapper;
+ UserPositionMapper userPositionMapper;
+ @Autowired
+ IUserService iUserService;
/**
* @Description: 用户新股列表
@@ -107,11 +110,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 +139,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 +155,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 +188,42 @@
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);*/
+ User user = this.iUserService.getCurrentRefreshUser(request);
+ if (user == null) {
+ return ServerResponse.createByErrorCodeMsg(ResponseCode.NEED_LOGIN.getCode(),"请先登录", request);
+ }
+ PageHelper.startPage(pageNum, pageSize);
+ List<StockSubscribeVO> subscribeVOs = stockSubscribeMapper.selectStockSubscribeVO(0 ,1 ,user.getId());
+ PageInfo<StockSubscribeVO> pageInfo = new PageInfo<>(subscribeVOs);
+
+ 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