From 06b7c3a60732ad12ee6e26bfbdcf830487ea3516 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期五, 20 九月 2024 16:39:52 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/RouteController.java |   18 ++++++++++++++++++
 1 files changed, 18 insertions(+), 0 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/RouteController.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/RouteController.java
index e1d0eb6..3a3dc37 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/RouteController.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/controller/RouteController.java
@@ -68,6 +68,21 @@
     @OperationLog("Create Route")
     @PostMapping("/route/save")
     public R save(@RequestBody Route route) {
+        if (route.getStartCode() == null || route.getEndCode() == null) {
+            return R.error("Save Fail");
+        }
+        if (route.getStartCode().equals(route.getEndCode())) {
+            return R.error("Save Fail");
+        }
+        if (routeService.count(new LambdaQueryWrapper<Route>()
+                .eq(Route::getStartCode, route.getStartCode())
+                .eq(Route::getEndCode, route.getEndCode())) > 0 ||
+                routeService.count(new LambdaQueryWrapper<Route>()
+                        .eq(Route::getStartCode, route.getStartCode())
+                        .eq(Route::getEndCode, route.getEndCode())) > 0
+        ) {
+            return R.error("Save Fail");
+        }
         route.setCreateBy(getLoginUserId());
         route.setCreateTime(new Date());
         route.setUpdateBy(getLoginUserId());
@@ -82,6 +97,9 @@
     @OperationLog("Update Route")
     @PostMapping("/route/update")
     public R update(@RequestBody Route route) {
+        if (route.getStartCode().equals(route.getEndCode())) {
+            return R.error("Update Fail");
+        }
         route.setUpdateBy(getLoginUserId());
         route.setUpdateTime(new Date());
         if (!routeService.updateById(route)) {

--
Gitblit v1.9.1