1

zj

2024-05-17 b271ad62a0e9ea1e16a95f342bbdbc3f93783148

 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;
@@ -58,18 +59,22 @@
    }

    @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("参数不能为空");
        }

        QueryWrapper queryWrapper = new QueryWrapper();
        queryWrapper.eq("stock_spell",stockCode);
        queryWrapper.eq("stock_code",stockCode);

        Stock stock = stockMapper.selectOne(queryWrapper);
        if (stock == null){
            return ServerResponse.createByErrorMsg("股票代码不存在");
        }
        Long count = stockDzMapper.selectCount(new LambdaQueryWrapper<StockDz>().eq(StockDz::getStockCode, stockCode));
        if(count > 0){
            return ServerResponse.createByErrorMsg("股票代码已存在");
        }
        StockDz stockDz = new StockDz();
        stockDz.setStockName(stock.getStockName());
@@ -87,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("添加成功");
@@ -109,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("修改失败");
    }