From ca455909bc92b314ef00fa0f10290aa70932b071 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 03 Sep 2024 18:53:44 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/StockSettingServicesImpl.java |   17 +++++++++++++++--
 1 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/StockSettingServicesImpl.java b/src/main/java/com/nq/service/impl/StockSettingServicesImpl.java
index 32c3f72..b84e8ec 100644
--- a/src/main/java/com/nq/service/impl/StockSettingServicesImpl.java
+++ b/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("修改失败");

--
Gitblit v1.9.3