zyy
2025-07-21 f25b39e0d5fcae415a8824fc03d42fe681e96bf8
src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -3,15 +3,18 @@
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;
@@ -37,7 +40,9 @@
    UserStockSubscribeMapper userStockSubscribeMapper;
    @Autowired
    StockMapper stockMapper;
    UserPositionMapper userPositionMapper;
    @Autowired
    IUserService iUserService;
    /**
     * @Description: 用户新股列表
@@ -47,7 +52,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();
@@ -66,11 +71,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);
@@ -88,15 +110,14 @@
    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())
                .eq(StockSubscribe::getType, model.getType()));
                .eq(StockSubscribe::getCode, model.getCode()));
        if(count > 0){
            return ServerResponse.createByErrorMsg("新股已存在");
            return ServerResponse.createByErrorMsg("新股代码已存在");
        }
        int resultCount = this.stockSubscribeMapper.insert(model);
        if (resultCount > 0) {
@@ -120,19 +141,25 @@
        }
        Long count = stockSubscribeMapper.selectCount(new LambdaQueryWrapper<StockSubscribe>()
                .eq(StockSubscribe::getCode, model.getCode())
                .eq(StockSubscribe::getType, model.getType()).ne(StockSubscribe::getNewlistId,model.getNewlistId()));
                .ne(StockSubscribe::getNewlistId,model.getNewlistId()));
        if(count > 0){
            return ServerResponse.createByErrorMsg("新股已存在");
            return ServerResponse.createByErrorMsg("新股代码已存在");
        }
        //修改新股
        int resultCount = this.stockSubscribeMapper.updateById(model);
        //修改用户申购的新股代码
        List<UserStockSubscribe> userStockSubscribes = userStockSubscribeMapper
                .selectList(new LambdaQueryWrapper<UserStockSubscribe>()
                        .eq(UserStockSubscribe::getNewCode, stockSubscribe.getCode()));
                        .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("修改新股成功");
@@ -161,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();
    }
}