ea03582d85
Change-Id: Iae6feaf9f51f32e20376a539e46700f07783419e Signed-off-by: Karol Latecki <karol.latecki@intel.com> Signed-off-by: Pawel Kaminski <pawelx.kaminski@intel.com> Reviewed-on: https://review.gerrithub.io/417683 Tested-by: SPDK CI Jenkins <sys_sgci@intel.com> Chandler-Test-Pool: SPDK Automated Test System <sys_sgsw@intel.com> Reviewed-by: Maciej Szwed <maciej.szwed@intel.com> Reviewed-by: Ben Walker <benjamin.walker@intel.com> Reviewed-by: Jim Harris <james.r.harris@intel.com>
205 lines
7.4 KiB
Python
205 lines
7.4 KiB
Python
from .ui_node import UINode, UIBdevs, UILvolStores, UIVhosts
|
|
import rpc.client
|
|
import rpc
|
|
|
|
|
|
class UIRoot(UINode):
|
|
"""
|
|
Root node for CLI menu tree structure. Refreshes running config on startup.
|
|
"""
|
|
def __init__(self, s, shell):
|
|
UINode.__init__(self, "/", shell=shell)
|
|
self.current_bdevs = []
|
|
self.current_lvol_stores = []
|
|
self.current_vhost_ctrls = []
|
|
self.set_rpc_target(s)
|
|
|
|
def refresh(self):
|
|
self._children = set([])
|
|
UIBdevs(self)
|
|
UILvolStores(self)
|
|
UIVhosts(self)
|
|
|
|
def set_rpc_target(self, s):
|
|
self.client = rpc.client.JSONRPCClient(s)
|
|
|
|
def print_array(self, a):
|
|
return " ".join(a)
|
|
|
|
def get_bdevs(self, bdev_type):
|
|
self.current_bdevs = rpc.bdev.get_bdevs(self.client)
|
|
# Following replace needs to be done in order for some of the bdev
|
|
# listings to work: logical volumes, split disk.
|
|
# For example logical volumes: listing in menu is "Logical_Volume"
|
|
# (cannot have space), but the product name in SPDK is "Logical Volume"
|
|
bdev_type = bdev_type.replace("_", " ")
|
|
for bdev in filter(lambda x: bdev_type in x["product_name"].lower(),
|
|
self.current_bdevs):
|
|
test = Bdev(bdev)
|
|
yield test
|
|
|
|
def get_bdevs_iostat(self, **kwargs):
|
|
return rpc.bdev.get_bdevs_iostat(self.client, **kwargs)
|
|
|
|
def split_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_split_vbdev(self.client, **kwargs)
|
|
return self.print_array(response)
|
|
|
|
def destruct_split_bdev(self, **kwargs):
|
|
rpc.bdev.destruct_split_vbdev(self.client, **kwargs)
|
|
|
|
def delete_bdev(self, name):
|
|
rpc.bdev.delete_bdev(self.client, bdev_name=name)
|
|
|
|
def create_malloc_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_malloc_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def delete_malloc_bdev(self, **kwargs):
|
|
rpc.bdev.delete_malloc_bdev(self.client, **kwargs)
|
|
|
|
def create_iscsi_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_iscsi_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def delete_iscsi_bdev(self, **kwargs):
|
|
rpc.bdev.delete_iscsi_bdev(self.client, **kwargs)
|
|
|
|
def create_aio_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_aio_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def delete_aio_bdev(self, **kwargs):
|
|
rpc.bdev.delete_aio_bdev(self.client, **kwargs)
|
|
|
|
def create_lvol_bdev(self, **kwargs):
|
|
response = rpc.lvol.construct_lvol_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def create_nvme_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_nvme_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def create_null_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_null_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def delete_null_bdev(self, **kwargs):
|
|
rpc.bdev.delete_null_bdev(self.client, **kwargs)
|
|
|
|
def create_error_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_error_bdev(self.client, **kwargs)
|
|
|
|
def delete_error_bdev(self, **kwargs):
|
|
rpc.bdev.delete_error_bdev(self.client, **kwargs)
|
|
|
|
def get_lvol_stores(self):
|
|
self.current_lvol_stores = rpc.lvol.get_lvol_stores(self.client)
|
|
for lvs in self.current_lvol_stores:
|
|
yield LvolStore(lvs)
|
|
|
|
def create_lvol_store(self, **kwargs):
|
|
response = rpc.lvol.construct_lvol_store(self.client, **kwargs)
|
|
new_lvs = rpc.lvol.get_lvol_stores(self.client,
|
|
response,
|
|
lvs_name=None)
|
|
return new_lvs[0]["name"]
|
|
|
|
def delete_lvol_store(self, **kwargs):
|
|
rpc.lvol.destroy_lvol_store(self.client, **kwargs)
|
|
|
|
def create_pmem_pool(self, **kwargs):
|
|
response = rpc.pmem.create_pmem_pool(self.client, **kwargs)
|
|
return response
|
|
|
|
def delete_pmem_pool(self, **kwargs):
|
|
rpc.pmem.delete_pmem_pool(self.client, **kwargs)
|
|
|
|
def create_pmem_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_pmem_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def create_rbd_bdev(self, **kwargs):
|
|
response = rpc.bdev.construct_rbd_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def create_virtio_dev(self, **kwargs):
|
|
response = rpc.vhost.construct_virtio_dev(self.client, **kwargs)
|
|
return self.print_array(response)
|
|
|
|
def remove_virtio_scsi_bdev(self, **kwargs):
|
|
response = rpc.vhost.remove_virtio_scsi_bdev(self.client, **kwargs)
|
|
return response
|
|
|
|
def get_virtio_scsi_devs(self):
|
|
for bdev in rpc.vhost.get_virtio_scsi_devs(self.client):
|
|
test = Bdev(bdev)
|
|
yield test
|
|
|
|
def list_vhost_ctrls(self):
|
|
self.current_vhost_ctrls = rpc.vhost.get_vhost_controllers(self.client)
|
|
|
|
def get_vhost_ctrlrs(self, ctrlr_type):
|
|
for ctrlr in filter(lambda x: ctrlr_type in x["backend_specific"].keys(),
|
|
self.current_vhost_ctrls):
|
|
yield VhostCtrlr(ctrlr)
|
|
|
|
def remove_vhost_controller(self, **kwargs):
|
|
rpc.vhost.remove_vhost_controller(self.client, **kwargs)
|
|
self.current_vhost_ctrls = rpc.vhost.get_vhost_controllers(self.client)
|
|
|
|
def create_vhost_scsi_controller(self, **kwargs):
|
|
rpc.vhost.construct_vhost_scsi_controller(self.client, **kwargs)
|
|
self.current_vhost_ctrls = rpc.vhost.get_vhost_controllers(self.client)
|
|
|
|
def create_vhost_blk_controller(self, **kwargs):
|
|
rpc.vhost.construct_vhost_blk_controller(self.client, **kwargs)
|
|
self.current_vhost_ctrls = rpc.vhost.get_vhost_controllers(self.client)
|
|
|
|
def remove_vhost_scsi_target(self, **kwargs):
|
|
rpc.vhost.remove_vhost_scsi_target(self.client, **kwargs)
|
|
self.current_vhost_ctrls = rpc.vhost.get_vhost_controllers(self.client)
|
|
|
|
def add_vhost_scsi_lun(self, **kwargs):
|
|
rpc.vhost.add_vhost_scsi_lun(self.client, **kwargs)
|
|
self.current_vhost_ctrls = rpc.vhost.get_vhost_controllers(self.client)
|
|
|
|
def set_vhost_controller_coalescing(self, **kwargs):
|
|
rpc.vhost.set_vhost_controller_coalescing(self.client, **kwargs)
|
|
|
|
|
|
class Bdev(object):
|
|
def __init__(self, bdev_info):
|
|
"""
|
|
All class attributes are set based on what information is received
|
|
from get_bdevs RPC call.
|
|
# TODO: Document in docstring parameters which describe bdevs.
|
|
# TODO: Possible improvement: JSON schema might be used here in future
|
|
"""
|
|
for i in bdev_info.keys():
|
|
setattr(self, i, bdev_info[i])
|
|
|
|
|
|
class LvolStore(object):
|
|
def __init__(self, lvs_info):
|
|
"""
|
|
All class attributes are set based on what information is received
|
|
from get_bdevs RPC call.
|
|
# TODO: Document in docstring parameters which describe bdevs.
|
|
# TODO: Possible improvement: JSON schema might be used here in future
|
|
"""
|
|
for i in lvs_info.keys():
|
|
setattr(self, i, lvs_info[i])
|
|
|
|
|
|
class VhostCtrlr(object):
|
|
def __init__(self, ctrlr_info):
|
|
"""
|
|
All class attributes are set based on what information is received
|
|
from get_vhost_controllers RPC call.
|
|
# TODO: Document in docstring parameters which describe bdevs.
|
|
# TODO: Possible improvement: JSON schema might be used here in future
|
|
"""
|
|
for i in ctrlr_info.keys():
|
|
setattr(self, i, ctrlr_info[i])
|