Преглед изворни кода

编辑站点 增加站点名称 用户名错误提示

lqc пре 5 година
родитељ
комит
c600810e9a

+ 28 - 0
src/main/java/com/mokamrp/privates/controller/pangu/FosterwxStationController.java

@@ -1,5 +1,6 @@
 package com.mokamrp.privates.controller.pangu;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.mokamrp.privates.constant.Constants;
 import com.mokamrp.privates.constant.HttpStatus;
@@ -64,6 +65,20 @@ public class FosterwxStationController extends BaseController<FosterwxStation> {
         if (bindingResult.hasErrors()) {
             return AjaxResult.error(HttpStatus.ERROR, bindingResult.getFieldError().getDefaultMessage());
         }
+
+        // 判断站点名是否存在
+        LambdaQueryWrapper<FosterwxStation> fosterwxStationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        fosterwxStationLambdaQueryWrapper.eq(FosterwxStation::getStationName, editHandle.getRaw().getStationName());
+        if (fosterwxStationService.count(fosterwxStationLambdaQueryWrapper) > 0) {
+            return AjaxResult.error("站点名重复");
+        }
+        //判断用户名是否存在
+        fosterwxStationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        fosterwxStationLambdaQueryWrapper.eq(FosterwxStation::getLoginUsername, editHandle.getRaw().getLoginUsername());
+        if (fosterwxStationService.count(fosterwxStationLambdaQueryWrapper) > 0) {
+            return AjaxResult.error("用户名重复");
+        }
+
         this.onUpdate(editHandle);
         service.update(editHandle.getRaw(), new QueryWrapper<FosterwxStation>().in("id", editHandle.getIds()));
         return AjaxResult.success(Constants.SUCCESS, true);
@@ -74,6 +89,19 @@ public class FosterwxStationController extends BaseController<FosterwxStation> {
         if (bindingResult.hasErrors()) {
             return AjaxResult.error(HttpStatus.ERROR, bindingResult.getFieldError().getDefaultMessage());
         }
+        // 判断站点名是否存在
+        LambdaQueryWrapper<FosterwxStation> fosterwxStationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        fosterwxStationLambdaQueryWrapper.eq(FosterwxStation::getStationName, addHandle.getRaw().getStationName());
+        if (fosterwxStationService.count(fosterwxStationLambdaQueryWrapper) > 0) {
+            return AjaxResult.error("站点名重复");
+        }
+        //判断用户名是否存在
+        fosterwxStationLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        fosterwxStationLambdaQueryWrapper.eq(FosterwxStation::getLoginUsername, addHandle.getRaw().getLoginUsername());
+        if (fosterwxStationService.count(fosterwxStationLambdaQueryWrapper) > 0) {
+            return AjaxResult.error("用户名重复");
+        }
+
         addHandle.getRaw().setUploadUid(user.getId());
         this.onInsert(addHandle.getRaw());
         Boolean res = fosterwxStationService.save(addHandle.getRaw());