Fix listing management server by parameters

Author: davidjumani

api/src/main/java/org/apache/cloudstack/api/command/admin/management/ListMgmtsCmd.java

@@ -22,7 +22,6 @@
 import org.apache.cloudstack.api.BaseCmd;
 import org.apache.cloudstack.api.BaseListCmd;
 import org.apache.cloudstack.api.Parameter;
-import org.apache.cloudstack.api.response.HostResponse;
 import org.apache.cloudstack.api.response.ListResponse;
 import org.apache.cloudstack.api.response.ManagementServerResponse;
 import org.apache.log4j.Logger;
@@ -38,7 +37,7 @@ public class ListMgmtsCmd extends BaseListCmd {
     //////////////// API parameters /////////////////////
     /////////////////////////////////////////////////////
 
-    @Parameter(name = ApiConstants.ID, type = CommandType.UUID, entityType = HostResponse.class, description = "the id of the management server")
+    @Parameter(name = ApiConstants.ID, type = CommandType.UUID, entityType = ManagementServerResponse.class, description = "the id of the management server")
     private Long id;
 
     @Parameter(name = ApiConstants.NAME, type = CommandType.STRING, description = "the name of the management server")

api/src/main/java/org/apache/cloudstack/management/ManagementServerHost.java

@@ -16,15 +16,14 @@
 // under the License.
 package org.apache.cloudstack.management;
 
-public interface ManagementServerHost {
+import org.apache.cloudstack.api.Identity;
+import org.apache.cloudstack.api.InternalIdentity;
+
+public interface ManagementServerHost extends InternalIdentity, Identity {
     enum State {
         Up, Down
     }
 
-    long getId();
-
-    String getUuid();
-
     long getMsid();
 
     State getState();

server/src/main/java/com/cloud/api/query/QueryManagerImpl.java

@@ -3828,20 +3828,43 @@ protected ResourceDetailResponse createResourceDetailsResponse(ResourceDetail re
     @Override
     public ListResponse<ManagementServerResponse> listManagementServers(ListMgmtsCmd cmd) {
         ListResponse<ManagementServerResponse> response = new ListResponse<>();
-        List<ManagementServerResponse> result = new ArrayList<>();
-        for (ManagementServerHostVO mgmt : managementServerHostDao.listAll()) {
-            ManagementServerResponse mgmtResponse = new ManagementServerResponse();
-            mgmtResponse.setId(mgmt.getUuid());
-            mgmtResponse.setName(mgmt.getName());
-            mgmtResponse.setState(mgmt.getState());
-            mgmtResponse.setVersion(mgmt.getVersion());
-            mgmtResponse.setObjectName("managementserver");
-            result.add(mgmtResponse);
-        }
-        response.setResponses(result);
+        Pair<List<ManagementServerHostVO>, Integer> result = listManagementServersInternal(cmd);
+        List<ManagementServerResponse> hostResponses = new ArrayList<>();
+
+        for(ManagementServerHostVO host : result.first()) {
+            ManagementServerResponse hostResponse = createManagementServerResponse(host);
+            hostResponses.add(hostResponse);
+         }
+
+        response.setResponses(hostResponses);
         return response;
     }
 
+    protected Pair<List<ManagementServerHostVO>, Integer> listManagementServersInternal(ListMgmtsCmd cmd) {
+        Long id = cmd.getId();
+        String name = cmd.getHostName();
+
+        SearchBuilder<ManagementServerHostVO> sb = managementServerHostDao.createSearchBuilder();
+        SearchCriteria<ManagementServerHostVO> sc = sb.create();
+        if(id != null) {
+            sc.addAnd("id", SearchCriteria.Op.EQ, id);
+        }
+        if(name != null) {
+            sc.addAnd("name", SearchCriteria.Op.EQ, name);
+        }
+        return managementServerHostDao.searchAndCount(sc, null);
+    }
+
+    protected ManagementServerResponse createManagementServerResponse(ManagementServerHostVO mgmt) {
+        ManagementServerResponse mgmtResponse = new ManagementServerResponse();
+        mgmtResponse.setId(mgmt.getUuid());
+        mgmtResponse.setName(mgmt.getName());
+        mgmtResponse.setState(mgmt.getState());
+        mgmtResponse.setVersion(mgmt.getVersion());
+        mgmtResponse.setObjectName("managementserver");
+        return mgmtResponse;
+    }
+
     @Override
     public ListResponse<TemplateOVFPropertyResponse> listTemplateOVFProperties(ListTemplateOVFProperties cmd) {
         ListResponse<TemplateOVFPropertyResponse> response = new ListResponse<>();