演示站

zj

2025-11-05 9c6296d07a668b15b9794709172e28ce3d9603e2

 src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java

@@ -7,9 +7,11 @@
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;
@@ -40,6 +42,9 @@

    @Autowired
    StockMapper stockMapper;

    @Autowired
    UserPositionMapper userPositionMapper;

    /**
     * @Description: 用户新股列表
@@ -107,11 +112,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 +141,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);
@@ -143,6 +158,11 @@
            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("修改新股成功");
        }