1

zj

2024-10-11 5f97f550b3fb60ac2142d7ca3c78d6bd04f18b80

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

@@ -1,5 +1,6 @@
package com.nq.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.nq.common.ServerResponse;
import com.nq.dao.StockMapper;
@@ -26,6 +27,15 @@

    @Override
    public ServerResponse updateStockSetting(StockSetting stockSetting) {
        Stock stock = stockMapper.findStockByCode(stockSetting.getStockCode());
        if(stock == null){
            return ServerResponse.createByErrorMsg("股票不存在");
        }
        Long count = stockSettingMapper.selectCount(new LambdaQueryWrapper<>(StockSetting.class)
                .eq(StockSetting::getStockCode, stockSetting.getStockCode()).ne(StockSetting::getId,stockSetting.getId()));
        if(count > 0){
            return ServerResponse.createByErrorMsg("盘前股票已存在！");
        }
       int ref =  stockSettingMapper.updateById(stockSetting);
       if(ref == 0){
           return   ServerResponse.createByErrorMsg("修改失败");
@@ -35,12 +45,15 @@

    @Override
    public ServerResponse saveStockSetting(StockSetting stockSetting) {


        Stock stock = stockMapper.findStockByCode(stockSetting.getStockCode());
        if(stock == null){
           return ServerResponse.createByErrorMsg("添加股票不存在");
        }
        Long count = stockSettingMapper.selectCount(new LambdaQueryWrapper<>(StockSetting.class)
                .eq(StockSetting::getStockCode, stockSetting.getStockCode()));
        if(count > 0){
            return ServerResponse.createByErrorMsg("盘前股票已存在！");
        }
        int ref =  stockSettingMapper.insert(stockSetting);
        if(ref == 0){
            return  ServerResponse.createByErrorMsg("修改失败");