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; import com.nq.dao.StockSettingMapper; import com.nq.pojo.Stock; import com.nq.pojo.StockSetting; import com.nq.service.IStockSettingServices; import org.apache.http.util.TextUtils; import org.springframework.stereotype.Service; import javax.annotation.Resource; @Service public class StockSettingServicesImpl implements IStockSettingServices { @Resource StockSettingMapper stockSettingMapper; @Resource StockMapper stockMapper; @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("修改失败"); } return ServerResponse.createBySuccess(); } @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("修改失败"); } return ServerResponse.createBySuccess(); } @Override public ServerResponse deleteStockSetting(Integer id) { int ref = stockSettingMapper.deleteById(id); if(ref == 0){ return ServerResponse.createByErrorMsg("修改失败"); } return ServerResponse.createBySuccess(); } @Override public ServerResponse queryAll(String stockCode) { QueryWrapper queryWrapper = new QueryWrapper<>(); if(!TextUtils.isEmpty(stockCode)){ queryWrapper.eq("stock_code",stockCode); } return ServerResponse.createBySuccess(stockSettingMapper.selectList(queryWrapper)); } }