Add storage and image management support

Signed-off-by: Zhenchao Liu <[email protected]>

avocado_vt/plugins/vt_cluster.py

@@ -8,6 +8,8 @@
 from avocado.utils.stacktrace import log_exc_info
 
 from virttest.vt_cluster import cluster, node_metadata
+from virttest.vt_imgr import vt_imgr
+from virttest.vt_resmgr import startup_resmgr, teardown_resmgr
 
 
 class ClusterCreationError(Exception):
@@ -31,7 +33,10 @@ def pre_tests(self, job):
             try:
                 for node in cluster.get_all_nodes():
                     node.start_agent_server()
+
                 node_metadata.load_metadata()
+                startup_resmgr()
+                vt_imgr.startup()
 
             except Exception as detail:
                 msg = "Failure trying to set Avocado-VT job env: %s" % detail
@@ -44,6 +49,8 @@ def post_tests(self, job):
             cluster_dir = os.path.join(job.logdir, "cluster")
             for node in cluster.get_all_nodes():
                 try:
+                    teardown_resmgr()
+                    vt_imgr.teardown()
                     node_dir = os.path.join(cluster_dir, node.name)
                     os.makedirs(node_dir)
                     node.upload_agent_log(node_dir)

virttest/bootstrap.py

@@ -12,6 +12,7 @@
 from avocado.utils import process
 
 from virttest.vt_cluster import cluster, node
+from virttest.vt_resmgr import setup_resmgr
 
 from . import arch, asset, cartesian_config, data_dir, defaults, utils_selinux
 from .compat import get_opt
@@ -894,6 +895,27 @@ def _register_hosts(hosts_configs):
             LOG.debug("Host %s registered", host)
 
 
+def _setup_managers(pools_params):
+    def _verify_pools_params():
+        # Check if the pools' params are set correctly
+        required_options = ["type", "access.nodes"]
+        for params in pools_params.values():
+            for pool_name, pool_params in params.items():
+                for option in required_options:
+                    opts = option.split('.')
+                    i, sub_params = 0, pool_params
+                    while i < len(opts):
+                        if opts[i] not in sub_params:
+                            raise ValueError(f"Missed '{opts[i]}' for '{pool_name}'")
+                        elif not sub_params[opts[i]]:
+                            raise ValueError(f"Missed a value for '{opts[i]}' for '{pool_name}'")
+                        sub_params = sub_params.get(opts[i])
+                        i += 1
+
+    _verify_pools_params()
+    setup_resmgr(pools_params)
+
+
 def _config_master_server(master_config):
     """Configure the master server."""
     if master_config:
@@ -1083,6 +1105,7 @@ def bootstrap(options, interactive=False):
         cluster_config = _load_cluster_config(vt_cluster_config)
         _register_hosts(cluster_config.get("hosts"))
         _config_master_server(cluster_config.get("master"))
+        _setup_managers(cluster_config.get("pools"))
 
     LOG.info("")
     LOG.info("VT-BOOTSTRAP FINISHED")

virttest/env_process.py

@@ -58,6 +58,8 @@
 from virttest.test_setup.storage import StorageConfig
 from virttest.utils_conn import SSHConnection
 from virttest.utils_version import VersionInterval
+from virttest.vt_imgr import vt_imgr
+
 
 utils_libvirtd = lazy_import("virttest.utils_libvirtd")
 virsh = lazy_import("virttest.virsh")
@@ -130,32 +132,51 @@ def preprocess_image(test, params, image_name, vm_process_status=None):
                               only for keep it work with process_images()
     :note: Currently this function just creates an image if requested.
     """
+    # FIXME:
+    image_id = None
+    if params.get_boolean("multihost"):
+        nodes = params.objects("nodes")
+        image_config = vt_imgr.define_image_config(image_name, params, nodes)
+        image_id = vt_imgr.create_image_object(image_config)
+
+    params = params.object_params(image_name)
     base_dir = params.get("images_base_dir", data_dir.get_data_dir())
 
     if not storage.preprocess_image_backend(base_dir, params, image_name):
         LOG.error("Backend can't be prepared correctly.")
 
-    image_filename = storage.get_image_filename(params, base_dir)
+    image_filename = None
+    if not params.get_boolean("multihost"):
+        image_filename = storage.get_image_filename(params, base_dir)
 
     create_image = False
     if params.get("force_create_image") == "yes":
         create_image = True
-    elif params.get("create_image") == "yes" and not storage.file_exists(
-        params, image_filename
-    ):
-        create_image = True
+    elif params.get("create_image") == "yes":
+        if params.get_boolean("multihost"):
+            volume = vt_imgr.query_image(
+                image_id,
+                request=f"spec.virt-images.{image_name}.spec.volume.meta"
+            )
+            create_image = True if not volume["meta"]["allocated"] else False
+        else:
+            create_image = True if not storage.file_exists(params, image_filename) else False
 
     if params.get("backup_image_before_testing", "no") == "yes":
+        # FIXME: add backup_image
         image = qemu_storage.QemuImg(params, base_dir, image_name)
         image.backup_image(params, base_dir, "backup", True, True)
     if create_image:
-        if storage.file_exists(params, image_filename):
-            # As rbd image can not be covered, so need remove it if we need
-            # force create a new image.
-            storage.file_remove(params, image_filename)
-        image = qemu_storage.QemuImg(params, base_dir, image_name)
-        LOG.info("Create image on %s." % image.storage_type)
-        image.create(params)
+        if params.get_boolean("multihost"):
+            vt_imgr.handle_image(image_id, {"create": {}})
+        else:
+            if storage.file_exists(params, image_filename):
+                # As rbd image can not be covered, so need remove it if we need
+                # force create a new image.
+                storage.file_remove(params, image_filename)
+            image = qemu_storage.QemuImg(params, base_dir, image_name)
+            LOG.info("Create image on %s." % image.storage_type)
+            image.create(params)
 
 
 def preprocess_fs_source(test, params, fs_name, vm_process_status=None):
@@ -535,6 +556,17 @@ def postprocess_image(test, params, image_name, vm_process_status=None):
         )
         return
 
+    # FIXME: multihost
+    image_id = None
+    if params.get_boolean("multihost"):
+        image_id = vt_imgr.get_image_by_tag(image_name)
+        if image_id is None:
+            LOG.warning(f"Cannot find image {image_name}")
+            nodes = params.objects("nodes")
+            image_config = vt_imgr.define_image_config(image_name, params, nodes)
+            image_id = vt_imgr.create_image_object(image_config)
+    params = params.object_params(image_name)
+
     restored, removed = (False, False)
     clone_master = params.get("clone_master", None)
     base_dir = params.get("images_base_dir", data_dir.get_data_dir())
@@ -592,10 +624,18 @@ def postprocess_image(test, params, image_name, vm_process_status=None):
         )
         LOG.info("Remove image on %s." % image.storage_type)
         if clone_master is None:
-            image.remove()
+            if params.get_boolean("multihost"):
+                vt_imgr.handle_image(image_id, {"destroy": {}})
+                vt_imgr.destroy_image_object(image_id)
+            else:
+                image.remove()
         elif clone_master == "yes":
             if image_name in params.get("master_images_clone").split():
-                image.remove()
+                if params.get_boolean("multihost"):
+                    vt_imgr.handle_image(image_id, {"destroy": {}})
+                    vt_imgr.destroy_image_object(image_id)
+                else:
+                    image.remove()
 
 
 def postprocess_fs_source(test, params, fs_name, vm_process_status=None):
@@ -825,8 +865,9 @@ def _process_images_serial(
                               or None for no vm exist.
     """
     for image_name in images:
-        image_params = params.object_params(image_name)
-        image_func(test, image_params, image_name, vm_process_status)
+        #image_params = params.object_params(image_name)
+        #image_func(test, image_params, image_name, vm_process_status)
+        image_func(test, params, image_name, vm_process_status)
         if exit_event and exit_event.is_set():
             LOG.error("Received exit_event, stop processing of images.")
             break

virttest/vt_agent/agents/__init__.py

@@ -0,0 +1,2 @@
+from .image_agent import image_agent
+from .resource_backing_agent import resbacking_agent

virttest/vt_agent/agents/image_agent.py

@@ -0,0 +1,21 @@
+import logging
+
+from .images import get_image_handler
+
+
+LOG = logging.getLogger("avocado.service." + __name__)
+
+
+class _ImageAgent(object):
+
+    def __init__(self):
+        pass
+
+    def handle_image(self, image_config, update_config):
+        cmd, arguments = update_config.popitem()
+        image_type = image_config["meta"]["type"]
+        handler = get_image_handler(image_type, cmd)
+        return handler(image_config, arguments)
+
+
+image_agent = _ImageAgent()

virttest/vt_agent/agents/images/__init__.py

@@ -0,0 +1,16 @@
+from .qemu import get_qemu_image_handler
+#from .xen import get_xen_image_handler
+
+
+_image_handler_getters = {
+    "qemu": get_qemu_image_handler,
+    #"xen": get_xen_image_handler,
+}
+
+
+def get_image_handler(image_type, cmd):
+    getter = _image_handler_getters.get(image_type)
+    return getter(cmd)
+
+
+__all__ = ["get_image_handler"]

virttest/vt_agent/agents/images/qemu/__init__.py

@@ -0,0 +1 @@
+from .qemu_image_handlers import get_qemu_image_handler