mgr/dashboard: add traddr param to nvmeof cli/api endpoints

Signed-off-by: Tomer Haskalovitch <[email protected]>

Author: Tomer Haskalovitch

src/pybind/mgr/dashboard/controllers/nvmeof.py

@@ -35,8 +35,8 @@ class NVMeoFGateway(RESTController):
         @NvmeofCLICommand("nvmeof gw info")
         @convert_to_model(model.GatewayInfo)
         @handle_nvmeof_error
-        def list(self, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.get_gateway_info(
+        def list(self, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.get_gateway_info(
                 NVMeoFClient.pb2.get_gateway_info_req()
             )
 
@@ -58,8 +58,8 @@ def group(self):
         @NvmeofCLICommand("nvmeof gw version")
         @convert_to_model(model.GatewayVersion)
         @handle_nvmeof_error
-        def version(self, gw_group: Optional[str] = None):
-            gw_info = NVMeoFClient(gw_group=gw_group).stub.get_gateway_info(
+        def version(self, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            gw_info = NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.get_gateway_info(
                 NVMeoFClient.pb2.get_gateway_info_req()
             )
             return NVMeoFClient.pb2.gw_version(status=gw_info.status,
@@ -71,8 +71,9 @@ def version(self, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof gw get_log_level")
         @convert_to_model(model.GatewayLogLevelInfo)
         @handle_nvmeof_error
-        def get_log_level(self, gw_group: Optional[str] = None):
-            gw_log_level = NVMeoFClient(gw_group=gw_group).stub.get_gateway_log_level(
+        def get_log_level(self, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            gw_log_level = NVMeoFClient(gw_group=gw_group,
+                                        traddr=traddr).stub.get_gateway_log_level(
                 NVMeoFClient.pb2.get_gateway_log_level_req()
             )
             return gw_log_level
@@ -82,9 +83,11 @@ def get_log_level(self, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof gw set_log_level")
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
-        def set_log_level(self, log_level: str, gw_group: Optional[str] = None):
+        def set_log_level(self, log_level: str, gw_group: Optional[str] = None,
+                          traddr: Optional[str] = None):
             log_level = log_level.lower()
-            gw_log_level = NVMeoFClient(gw_group=gw_group).stub.set_gateway_log_level(
+            gw_log_level = NVMeoFClient(gw_group=gw_group,
+                                        traddr=traddr).stub.set_gateway_log_level(
                 NVMeoFClient.pb2.set_gateway_log_level_req(log_level=log_level)
             )
             return gw_log_level
@@ -97,8 +100,9 @@ class NVMeoFSpdk(RESTController):
         @NvmeofCLICommand("nvmeof spdk_log_level get")
         @convert_to_model(model.SpdkNvmfLogFlagsAndLevelInfo)
         @handle_nvmeof_error
-        def get_spdk_log_level(self, gw_group: Optional[str] = None):
-            spdk_log_level = NVMeoFClient(gw_group=gw_group).stub.get_spdk_nvmf_log_flags_and_level(
+        def get_spdk_log_level(self, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            spdk_log_level = NVMeoFClient(gw_group=gw_group,
+                                          traddr=traddr).stub.get_spdk_nvmf_log_flags_and_level(
                 NVMeoFClient.pb2.get_spdk_nvmf_log_flags_and_level_req()
             )
             return spdk_log_level
@@ -109,10 +113,12 @@ def get_spdk_log_level(self, gw_group: Optional[str] = None):
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
         def set_spdk_log_level(self, log_level: Optional[str] = None,
-                               print_level: Optional[str] = None, gw_group: Optional[str] = None):
+                               print_level: Optional[str] = None,
+                               gw_group: Optional[str] = None, traddr: Optional[str] = None):
             log_level = log_level.upper() if log_level else None
             print_level = print_level.upper() if print_level else None
-            spdk_log_level = NVMeoFClient(gw_group=gw_group).stub.set_gateway_log_level(
+            spdk_log_level = NVMeoFClient(gw_group=gw_group,
+                                          traddr=traddr).stub.set_gateway_log_level(
                 NVMeoFClient.pb2.set_spdk_nvmf_logs_req(log_level=log_level,
                                                         print_level=print_level)
             )
@@ -123,8 +129,10 @@ def set_spdk_log_level(self, log_level: Optional[str] = None,
         @NvmeofCLICommand("nvmeof spdk_log_level disable")
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
-        def disable_spdk_log_level(self, gw_group: Optional[str] = None):
-            spdk_log_level = NVMeoFClient(gw_group=gw_group).stub.disable_spdk_nvmf_logs(
+        def disable_spdk_log_level(self, gw_group: Optional[str] = None,
+                                   traddr: Optional[str] = None):
+            spdk_log_level = NVMeoFClient(gw_group=gw_group,
+                                          traddr=traddr).stub.disable_spdk_nvmf_logs(
                 NVMeoFClient.pb2.disable_spdk_nvmf_logs_req()
             )
             return spdk_log_level
@@ -137,8 +145,8 @@ class NVMeoFSubsystem(RESTController):
         @NvmeofCLICommand("nvmeof subsystem list")
         @convert_to_model(model.SubsystemList)
         @handle_nvmeof_error
-        def list(self, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_subsystems(
+        def list(self, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_subsystems(
                 NVMeoFClient.pb2.list_subsystems_req()
             )
 
@@ -153,8 +161,8 @@ def list(self, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof subsystem get")
         @convert_to_model(model.SubsystemList)
         @handle_nvmeof_error
-        def get(self, nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_subsystems(
+        def get(self, nqn: str, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_subsystems(
                 NVMeoFClient.pb2.list_subsystems_req(subsystem_nqn=nqn)
             )
 
@@ -172,8 +180,8 @@ def get(self, nqn: str, gw_group: Optional[str] = None):
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
         def create(self, nqn: str, enable_ha: bool = True, max_namespaces: int = 1024,
-                   gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.create_subsystem(
+                   gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.create_subsystem(
                 NVMeoFClient.pb2.create_subsystem_req(
                     subsystem_nqn=nqn, max_namespaces=max_namespaces, enable_ha=enable_ha
                 )
@@ -191,8 +199,9 @@ def create(self, nqn: str, enable_ha: bool = True, max_namespaces: int = 1024,
         @NvmeofCLICommand("nvmeof subsystem del")
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
-        def delete(self, nqn: str, force: Optional[str] = "false", gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.delete_subsystem(
+        def delete(self, nqn: str, force: Optional[str] = "false", gw_group: Optional[str] = None,
+                   traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.delete_subsystem(
                 NVMeoFClient.pb2.delete_subsystem_req(
                     subsystem_nqn=nqn, force=str_to_bool(force)
                 )
@@ -212,8 +221,8 @@ class NVMeoFListener(RESTController):
         @NvmeofCLICommand("nvmeof listener list")
         @convert_to_model(model.ListenerList)
         @handle_nvmeof_error
-        def list(self, nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_listeners(
+        def list(self, nqn: str, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_listeners(
                 NVMeoFClient.pb2.list_listeners_req(subsystem=nqn)
             )
 
@@ -301,8 +310,8 @@ class NVMeoFNamespace(RESTController):
         @NvmeofCLICommand("nvmeof ns list")
         @convert_to_model(model.NamespaceList)
         @handle_nvmeof_error
-        def list(self, nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_namespaces(
+        def list(self, nqn: str, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_namespaces(
                 NVMeoFClient.pb2.list_namespaces_req(subsystem=nqn)
             )
 
@@ -318,8 +327,9 @@ def list(self, nqn: str, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof ns get")
         @convert_to_model(model.NamespaceList)
         @handle_nvmeof_error
-        def get(self, nqn: str, nsid: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_namespaces(
+        def get(self, nqn: str, nsid: str, gw_group: Optional[str] = None,
+                traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_namespaces(
                 NVMeoFClient.pb2.list_namespaces_req(subsystem=nqn, nsid=int(nsid))
             )
 
@@ -336,8 +346,9 @@ def get(self, nqn: str, nsid: str, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof ns get_io_stats")
         @convert_to_model(model.NamespaceIOStats)
         @handle_nvmeof_error
-        def io_stats(self, nqn: str, nsid: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.namespace_get_io_stats(
+        def io_stats(self, nqn: str, nsid: str, gw_group: Optional[str] = None,
+                     traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.namespace_get_io_stats(
                 NVMeoFClient.pb2.namespace_get_io_stats_req(
                     subsystem_nqn=nqn, nsid=int(nsid))
             )
@@ -380,10 +391,11 @@ def create(
             block_size: int = 512,
             load_balancing_group: Optional[int] = None,
             gw_group: Optional[str] = None,
+            traddr: Optional[str] = None,
             force: Optional[bool] = False,
             no_auto_visible: Optional[bool] = False
         ):
-            return NVMeoFClient(gw_group=gw_group).stub.namespace_add(
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.namespace_add(
                 NVMeoFClient.pb2.namespace_add_req(
                     subsystem_nqn=nqn,
                     rbd_image_name=rbd_image_name,
@@ -428,6 +440,7 @@ def update(
             r_mbytes_per_second: Optional[int] = None,
             w_mbytes_per_second: Optional[int] = None,
             gw_group: Optional[str] = None,
+            traddr: Optional[str] = None,
             trash_image: Optional[bool] = None,
         ):
             contains_failure = False
@@ -436,7 +449,7 @@ def update(
                 mib = 1024 * 1024
                 new_size_mib = int((rbd_image_size + mib - 1) / mib)
 
-                resp = NVMeoFClient(gw_group=gw_group).stub.namespace_resize(
+                resp = NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.namespace_resize(
                     NVMeoFClient.pb2.namespace_resize_req(
                         subsystem_nqn=nqn, nsid=int(nsid), new_size=new_size_mib
                     )
@@ -482,7 +495,7 @@ def update(
             if contains_failure:
                 cherrypy.response.status = 202
 
-            response = NVMeoFClient(gw_group=gw_group).stub.list_namespaces(
+            response = NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_namespaces(
                 NVMeoFClient.pb2.list_namespaces_req(subsystem=nqn, nsid=int(nsid))
             )
             return response
@@ -505,9 +518,10 @@ def delete(
             nqn: str,
             nsid: str,
             gw_group: Optional[str] = None,
+            traddr: Optional[str] = None,
             force: Optional[str] = "false"
         ):
-            return NVMeoFClient(gw_group=gw_group).stub.namespace_delete(
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.namespace_delete(
                 NVMeoFClient.pb2.namespace_delete_req(
                     subsystem_nqn=nqn,
                     nsid=int(nsid),
@@ -536,8 +550,8 @@ class NVMeoFHost(RESTController):
             else o,
         )
         @handle_nvmeof_error
-        def list(self, nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_hosts(
+        def list(self, nqn: str, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_hosts(
                 NVMeoFClient.pb2.list_hosts_req(subsystem=nqn)
             )
 
@@ -553,8 +567,9 @@ def list(self, nqn: str, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof host add")
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
-        def create(self, nqn: str, host_nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.add_host(
+        def create(self, nqn: str, host_nqn: str, gw_group: Optional[str] = None,
+                   traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.add_host(
                 NVMeoFClient.pb2.add_host_req(subsystem_nqn=nqn, host_nqn=host_nqn)
             )
 
@@ -570,8 +585,9 @@ def create(self, nqn: str, host_nqn: str, gw_group: Optional[str] = None):
         @NvmeofCLICommand("nvmeof host del")
         @convert_to_model(model.RequestStatus)
         @handle_nvmeof_error
-        def delete(self, nqn: str, host_nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.remove_host(
+        def delete(self, nqn: str, host_nqn: str, gw_group: Optional[str] = None,
+                   traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.remove_host(
                 NVMeoFClient.pb2.remove_host_req(subsystem_nqn=nqn, host_nqn=host_nqn)
             )
 
@@ -589,8 +605,8 @@ class NVMeoFConnection(RESTController):
         @NvmeofCLICommand("nvmeof connection list")
         @convert_to_model(model.ConnectionList)
         @handle_nvmeof_error
-        def list(self, nqn: str, gw_group: Optional[str] = None):
-            return NVMeoFClient(gw_group=gw_group).stub.list_connections(
+        def list(self, nqn: str, gw_group: Optional[str] = None, traddr: Optional[str] = None):
+            return NVMeoFClient(gw_group=gw_group, traddr=traddr).stub.list_connections(
                 NVMeoFClient.pb2.list_connections_req(subsystem=nqn)
             )