1
zj
2024-04-26 cf4abf670a2fea1151c136e9bdf0c7fcedca27e9
src/main/java/com/nq/service/impl/StockDzServiceImpl.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.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.nq.common.ServerResponse;
@@ -8,7 +9,7 @@
import com.nq.pojo.Stock;
import com.nq.pojo.StockDz;
import com.nq.service.StockDzService;
import com.nq.utils.DateTimeUtil;
import com.nq.utils.timeutil.DateTimeUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
@@ -58,21 +59,29 @@
    }
    @Override
    public ServerResponse addByAdmin(String stockCode, String stockNum, String password, String startTime, String endTime, String discount) {
    public ServerResponse addByAdmin(String stockCode, String stockNum, String password, String startTime, String endTime, String discount,Integer period) {
        if (stockCode == null || stockCode.equals("")||stockNum == null || stockNum.equals("")||password == null || discount == null|| discount.equals("")||
                password.equals("")|| startTime == null || startTime.equals("")||endTime == null || endTime.equals("")){
                password.equals("")|| startTime == null || startTime.equals("")||endTime == null || endTime.equals("") || period == null){
            return ServerResponse.createByErrorMsg("参数不能为空");
        }
        Stock stock = stockMapper.findStockByCode(stockCode);
        QueryWrapper queryWrapper = new QueryWrapper();
        queryWrapper.eq("stock_spell",stockCode);
        Stock stock = stockMapper.selectOne(queryWrapper);
        if (stock == null){
            return ServerResponse.createByErrorMsg("股票代码不存在");
        }
        Long count = stockDzMapper.selectCount(new LambdaQueryWrapper<StockDz>().eq(StockDz::getStockSpell, stockCode));
        if(count > 0){
            return ServerResponse.createByErrorMsg("股票代码已存在");
        }
        StockDz stockDz = new StockDz();
        stockDz.setStockName(stock.getStockName());
        stockDz.setStockCode(stock.getStockCode());
        stockDz.setStockType(stock.getStockType());
        stockDz.setStockGid(stock.getStockGid());
        stockDz.setStockPlate(stock.getStockPlate());
        stockDz.setStockSpell(stock.getStockSpell());
        stockDz.setIsLock(stock.getIsLock());
        stockDz.setIsShow(1);
        stockDz.setAddTime(new Date());
@@ -83,6 +92,7 @@
        stockDz.setStartTime(DateTimeUtil.strToDate(startTime));
        stockDz.setEndTime(DateTimeUtil.strToDate(endTime));
        stockDz.setDiscount(new BigDecimal(discount));
        stockDz.setPeriod(period);
        int res = stockDzMapper.insert(stockDz);
        if (res > 0) {
            return ServerResponse.createBySuccessMsg("添加成功");
@@ -105,7 +115,12 @@
    @Override
    public ServerResponse updByAdmin(StockDz model) {
//        log.info("model:{}",model);
        Long count = stockDzMapper.selectCount(new LambdaQueryWrapper<StockDz>()
                .eq(StockDz::getStockCode, model.getStockCode())
                .ne(StockDz::getId,model.getId()));
        if(count > 0){
            return ServerResponse.createByErrorMsg("股票代码已存在");
        }
        return stockDzMapper.updateById(model) > 0 ? ServerResponse.createBySuccess("修改成功") : ServerResponse.createByErrorMsg("修改失败");
    }