简单优化BusServer的Controller层代码

王勇
src/main/java/com/sunyo/wlpt/message/bus/service/controller/BusServerController.java
--- a/src/main/java/com/sunyo/wlpt/message/bus/service/controller/BusServerController.java
查看文件 @a7ca021
+++ b/src/main/java/com/sunyo/wlpt/message/bus/service/controller/BusServerController.java
查看文件 @a7ca021
 package com.sunyo.wlpt.message.bus.service.controller;
 import com.github.pagehelper.PageInfo;
-import com.sunyo.wlpt.message.bus.service.domain.BusQueue;
 import com.sunyo.wlpt.message.bus.service.domain.BusServer;
 import com.sunyo.wlpt.message.bus.service.response.ResultJson;
 import com.sunyo.wlpt.message.bus.service.service.BusServerService;
@@ -11,6 +10,8 @@ import org.springframework.web.bind.annotation.*;
 import javax.annotation.Resource;
 import java.util.List;
+import static com.sunyo.wlpt.message.bus.service.common.Constant.EXIST_SERVERNAME;
+
 /**
  * @author 子诚
  * Description：关于MQ服务器静态信息的控制器
@@ -33,26 +34,16 @@ public class BusServerController {
      * @return 成功返回MQ服务器列表
      */
     @GetMapping("/list")
-    public ResultJson<PageInfo> selectBusServerList(
+    public ResultJson selectBusServerList(
             @RequestParam(value = "serverName", required = false) String serverName,
             @RequestParam(value = "pageNum", defaultValue = "1") Integer pageNum,
             @RequestParam(value = "pageSize", defaultValue = "10") Integer pageSize) {
-
-        ResultJson<PageInfo> result = new ResultJson<>();
         BusServer busServer = new BusServer();
         // 获取参数，服务器名称
         busServer.setServerName(serverName);
         // 分页查询
         PageInfo pageInfo = busServerService.selectBusServerList(busServer, pageNum, pageSize);
-        if (pageInfo.getTotal() > 0) {
-            result.setCode("200");
-            result.setData(pageInfo);
-            result.setMsg("查询MQ服务器列表,成功!");
-        } else {
-            result.setCode("500");
-            result.setMsg("查询MQ服务器列表,失败!");
-        }
-        return result;
+        return pageInfo.getTotal() > 0 ? new ResultJson<>("200", "查询MQ服务器列表,成功!", pageInfo) : new ResultJson<>("500", "查询MQ服务器列表,失败!");
     }
     /**
@@ -63,16 +54,8 @@ public class BusServerController {
      */
     @DeleteMapping("/delete")
     public ResultJson deleteBusServer(@RequestBody BusServer busServer) {
-        ResultJson result = new ResultJson<>();
         int num = busServerService.deleteByPrimaryKey(busServer.getId());
-        if (num > 0) {
-            result.setCode("200");
-            result.setMsg("删除MQ服务器，成功");
-        } else {
-            result.setCode("500");
-            result.setMsg("删除MQ服务器，失败");
-        }
-        return result;
+        return num > 0 ? new ResultJson<>("200", "删除MQ服务器，成功") : new ResultJson<>("500", "删除MQ服务器，失败");
     }
     /**
@@ -83,16 +66,8 @@ public class BusServerController {
      */
     @GetMapping("/batchRemove")
     public ResultJson batchRemoveBusServer(String ids) {
-        ResultJson result = new ResultJson<>();
         int num = busServerService.deleteByPrimaryKey(ids);
-        if (num > 0) {
-            result.setCode("200");
-            result.setMsg("批量删除服务器,成功!");
-        } else {
-            result.setCode("500");
-            result.setMsg("批量删除服务器,失败!");
-        }
-        return result;
+        return num > 0 ? new ResultJson<>("200", "批量删除服务器，成功") : new ResultJson<>("500", "批量删除服务器，失败");
     }
     /**
@@ -105,25 +80,15 @@ public class BusServerController {
     public ResultJson updateBusServer(@RequestBody BusServer busServer) {
         //先验证，修改好的核心信息（ip和port，同时存在）是否已存在
         String message = validateBusServer(busServer);
-        // 判断服务器名称是否已存在
-        String existName = "MQ服务器名称已存在";
-        if (existName.equals(message)) {
-            return new ResultJson("402", "MQ服务器名称已存在");
+        if (EXIST_SERVERNAME.equals(message)) {
+            return new ResultJson<>("402", "MQ服务器名称已存在");
         }
         //验证通过
         if (message == null) {
-            ResultJson result = new ResultJson<>();
             int num = busServerService.updateByPrimaryKeySelective(busServer);
-            if (num > 0) {
-                result.setCode("200");
-                result.setMsg("修改服务器信息，成功");
-            } else {
-                result.setCode("500");
-                result.setMsg("修改服务器信息，失败");
-            }
-            return result;
+            return num > 0 ? new ResultJson<>("200", "修改服务器信息，成功") : new ResultJson<>("500", "修改服务器信息，失败");
         }
-        return new ResultJson("400", message != null ? message : "修改服务器信息,失败!");
+        return new ResultJson<>("400", message);
     }
     /**
@@ -136,27 +101,17 @@ public class BusServerController {
     public ResultJson insertBusServer(@RequestBody BusServer busServer) {
         //先验证，增加的服务器的核心信息（ip和port，同时存在）是否已存在
         String message = validateBusServer(busServer);
-        // 判断服务器名称是否已存在
-        String existName = "MQ服务器名称已存在";
-        if (existName.equals(message)) {
-            return new ResultJson("402", "MQ服务器名称已存在");
+        if (EXIST_SERVERNAME.equals(message)) {
+            return new ResultJson<>("402", "MQ服务器名称已存在");
         }
         //验证通过
         if (message == null) {
-            ResultJson result = new ResultJson<>();
             // 设置id
             busServer.setId(IdUtils.generateId());
             int num = busServerService.insertSelective(busServer);
-            if (num > 0) {
-                result.setCode("200");
-                result.setMsg("添加服务器信息，成功!");
-            } else {
-                result.setCode("500");
-                result.setMsg("添加服务器信息，失败!");
-            }
-            return result;
+            return num > 0 ? new ResultJson<>("200", "添加服务器信息，成功") : new ResultJson<>("500", "添加服务器信息，失败");
         }
-        return new ResultJson("400", message != null ? message : "该服务器信息，已存在!");
+        return new ResultJson<>("400", message);
     }
     /**
@@ -187,7 +142,7 @@ public class BusServerController {
             if (!oldBusServer.getServerName().equals(busServer.getServerName())) {
                 List<BusServer> serverNames = busServerService.validateServerName(busServer);
                 if (serverNames.size() > 0) {
-                    return "MQ服务器名称已存在";
+                    return EXIST_SERVERNAME;
                 }
             }
             /**
@@ -215,7 +170,7 @@ public class BusServerController {
             // 必须保证，修改或者增加 后的 serverName 是唯一的，即数据库中不存在
             List<BusServer> serverNames = busServerService.validateServerName(busServer);
             if (serverNames.size() > 0) {
-                return "MQ服务器名称已存在";
+                return EXIST_SERVERNAME;
             }
             // 根据 ServerIp以及ServerPort（同时），进行查询校验
             List<BusServer> info = busServerService.validateBusServer(busServer);