From 867183699202225af1aff812a2111f3b821f1e11 Mon Sep 17 00:00:00 2001
From: lxw <810412026@qq.com>
Date: 星期二, 20 九月 2022 14:09:25 +0800
Subject: [PATCH] 删除台站

---
 src/main/java/com/whyc/service/StationInfoService.java |   47 +++++++++++++++++++++++++++++++----------------
 1 files changed, 31 insertions(+), 16 deletions(-)

diff --git a/src/main/java/com/whyc/service/StationInfoService.java b/src/main/java/com/whyc/service/StationInfoService.java
index 6a18e73..aa0fc18 100644
--- a/src/main/java/com/whyc/service/StationInfoService.java
+++ b/src/main/java/com/whyc/service/StationInfoService.java
@@ -217,33 +217,48 @@
     @Transactional
     public Response deleteStation(String stationName1, String stationName2, String stationName3) {
         String stationName="";
-        UpdateWrapper wrapper=new UpdateWrapper();
+        int stationFlag=0;
         if(stationName1!=null&&!stationName1.isEmpty()){
             stationName=stationName+stationName1;
+            stationFlag=1;
         }
         if(stationName2!=null&&!stationName2.isEmpty()){
             stationName=stationName+"-"+stationName2;
+            stationFlag=2;
         }
         if(stationName3!=null&&!stationName3.isEmpty()){
             stationName=stationName+"-"+stationName3;
+            stationFlag=3;
         }
-        wrapper.eq("station_name",stationName);
-        //1.鍏堟煡鍑烘満鎴跨殑id
-        Integer stationId=mapper.selectIdByName(stationName);
-        if(stationId==null){
+        List<Integer> idStations=new ArrayList<>();
+        if(stationFlag==1){
+            //鎵惧嚭鏍硅妭鐐规槸stationName1鐨勬墍鏈夋満鎴縤d
+            idStations=mapper.selectIdByName1(stationName1);
+        }
+        if(stationFlag==2){
+            //鎵惧嚭鏍硅妭鐐规槸stationName1,stationName2鐨勬墍鏈夋満鎴縤d
+            idStations=mapper.selectIdByName2(stationName1,stationName2);
+        }
+        if(stationFlag==3){
+            //鎵惧嚭鏍硅妭鐐规槸stationName1,stationName2,stationName3鐨勬墍鏈夋満鎴縤d
+           idStations=mapper.selectIdByName3(stationName1,stationName2,stationName3);
+        }
+        if(idStations!=null&&idStations.size()>0){
+            for (int stationId:idStations) {
+                //3.鍒犻櫎鎵�鏈夋枃浠舵暟鎹�
+                dataMapper.deleteDataAndInfoByStationId(stationId);
+                //4.鍒犻櫎鎵�鏈夋枃浠跺弬鏁�
+                paramMapper.deleteParamByStationId(stationId);
+                //鏈�缁堝垹闄ゅ熀绔�
+                UpdateWrapper wrapper=new UpdateWrapper();
+                wrapper.eq("station_id",stationId);
+                wrapper.eq("station_name",stationName);
+                mapper.delete(wrapper);
+            }
+            return new Response().set(1,true,"鍒犻櫎鎴愬姛");
+        }else{
             return new Response().set(1,false,"鏈烘埧涓嶅瓨鍦�");
-        }else if(stationId==0){
-            //鏈�缁堝垹闄ゅ熀绔�
-            mapper.delete(wrapper);
-        }else {
-            //3.鍒犻櫎鎵�鏈夋枃浠舵暟鎹�
-            dataMapper.deleteDataAndInfoByStationId(stationId);
-            //4.鍒犻櫎鎵�鏈夋枃浠跺弬鏁�
-            paramMapper.deleteParamByStationId(stationId);
-            //鏈�缁堝垹闄ゅ熀绔�
-            mapper.delete(wrapper);
         }
-        return new Response().set(1,true,"鍒犻櫎鎴愬姛");
     }
     //绉婚櫎鍙扮珯涓嬫枃浠讹紙鍗曚釜鏂囦欢锛�
     @Transactional

--
Gitblit v1.9.1