diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/customWorkGroup/controller/WorkGroupController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/customWorkGroup/controller/WorkGroupController.java
index ba18cc02..40b54d6d 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/customWorkGroup/controller/WorkGroupController.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/customWorkGroup/controller/WorkGroupController.java
@@ -5,6 +5,7 @@ import com.adc.da.http.ResponseMessage;
import com.adc.da.http.Result;
import com.adc.da.slrs.customWorkGroup.entity.WorkGroupQueryPage;
import com.adc.da.slrs.customWorkGroup.service.IWorkGroupService;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import io.swagger.annotations.ApiModelProperty;
import org.apache.ibatis.annotations.Delete;
@@ -14,6 +15,8 @@ import com.adc.da.slrs.customWorkGroup.entity.WorkGroup;
import io.swagger.annotations.Api;
import com.adc.da.base.web.BaseController;
+import java.util.List;
+
/**
*
* 前端控制器
@@ -34,9 +37,18 @@ public class WorkGroupController extends BaseController {
@ApiModelProperty("新增工作组")
@PostMapping
public ResponseMessage addWorkGroup(WorkGroup workGroup) {
- boolean add = iWorkGroupService.add(workGroup);
+ String meg="";
+ if (isDistinct(workGroup.getWorkGroup())){
+ if (iWorkGroupService.add(workGroup)){
+ meg="新增成功!";
+ }else {
+ meg="新增失败!";
+ }
+ }else {
+ meg="组名已存在!";
+ }
- return Result.success(add);
+ return Result.success("200",meg);
}
@@ -53,7 +65,6 @@ public class WorkGroupController extends BaseController {
@GetMapping
public ResponseMessage> getWorkGroupLists(WorkGroupQueryPage queryPage){
IPage workGroup = iWorkGroupService.getWorkGroup(queryPage);
-
return Result.success(workGroup);
}
@@ -61,11 +72,34 @@ public class WorkGroupController extends BaseController {
@ApiModelProperty("修改工作组")
@PutMapping
public ResponseMessage updateWorkGroup(WorkGroup workGroup){
- boolean update = iWorkGroupService.update(workGroup);
+ String meg="";
+ if (isDistinct(workGroup.getWorkGroup())){
+ if (iWorkGroupService.update(workGroup)){
+ meg="修改成功!";
+ }else {
+ meg="修改失败";
+ }
- return Result.success(update);
+ }else {
+ meg="组名已存在!";
+ }
+
+ return Result.success("200",meg);
}
+ /**
+ * 判断提交的工作组名称是否重复
+ * @param groupName
+ * @return true 没有重复 false 有重复
+ */
+ private boolean isDistinct(String groupName){
+ QueryWrapper ew = new QueryWrapper<>();
+
+ ew.eq("work_group_name",groupName);
+
+ List list = iWorkGroupService.list(ew);
+ return list.isEmpty();
+ }
}