<fix>[zstacklib]: Support upgrade from c76/c79 to h76c/h79c

littleya · littleya · commit 1faa8c7aaf93 · 2023-06-06T15:18:51.000+08:00
Support upgrade from c76/c79 to h76c/h79c, there are two scenarios:
1. mn is h76c/h79c, host needs to upgrade from c76/c79 to h76c/h79c
2. mn is other distro, but hash h76c/h79c repo, and host is c76/c79,
   need to upgrade to h76c/h79c

The upgrade process is as follow:
1. copy hexli-release to host and install it
2. copy zstack-release to host and install it

Resolves: ZSTAC-55827

Change-Id: I69757a6c6962726166626a66616f676c7a796366
diff --git a/appbuildsystem/ansible/appbuild.py b/appbuildsystem/ansible/appbuild.py
@@ -54,6 +54,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/appliancevm/ansible/appliancevm.py b/appliancevm/ansible/appliancevm.py
@@ -52,9 +52,9 @@
 host_post_info.remote_port = remote_port
 if remote_pass is not None and remote_user != 'root':
     host_post_info.become = True
-
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/baremetalpxeserver/ansible/baremetalpxeserver.py b/baremetalpxeserver/ansible/baremetalpxeserver.py
@@ -57,9 +57,9 @@
 host_post_info.remote_port = remote_port
 if remote_pass is not None and remote_user != 'root':
     host_post_info.become = True
-
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/cephbackupstorage/ansible/cephb.py b/cephbackupstorage/ansible/cephb.py
@@ -58,9 +58,9 @@
 host_post_info.remote_port = remote_port
 if remote_pass is not None and remote_user != 'root':
     host_post_info.become = True
-
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/cephprimarystorage/ansible/cephp.py b/cephprimarystorage/ansible/cephp.py
@@ -60,6 +60,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/consoleproxy/ansible/consoleproxy.py b/consoleproxy/ansible/consoleproxy.py
@@ -51,6 +51,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/imagestorebackupstorage/ansible/imagestorebackupstorage.py b/imagestorebackupstorage/ansible/imagestorebackupstorage.py
@@ -64,6 +64,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/iscsifilesystemagent/ansible/iscsi.py b/iscsifilesystemagent/ansible/iscsi.py
@@ -54,6 +54,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/kvmagent/ansible/kvm.py b/kvmagent/ansible/kvm.py
@@ -80,6 +80,7 @@
     host_post_info.become = True
 
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/sftpbackupstorage/ansible/sftpbackupstorage.py b/sftpbackupstorage/ansible/sftpbackupstorage.py
@@ -57,6 +57,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/virtualrouter/ansible/virtualrouter.py b/virtualrouter/ansible/virtualrouter.py
@@ -55,6 +55,7 @@
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/zstacklib/ansible/zstacklib.py b/zstacklib/ansible/zstacklib.py
@@ -2129,6 +2129,38 @@ def check_umask(host_post_info):
     run_remote_command(check_umask_cmd, host_post_info)
 
 
+def upgrade_to_helix(host_info, host_post_info):
+    releasever = get_host_releasever(host_info)
+    if releasever in ['c76', 'c79']:
+        distro_name = {
+            'c76': 'h76c',
+            'c79': 'h79c'
+            }.get(releasever)
+        if not os.path.exists('/opt/zstack-dvd/x86_64/%s' % distro_name):
+            return host_info
+        pkg_name = {
+            'c76': 'helix-release-7-6c.0.h7.helix.x86_64.rpm',
+            'c79': 'helix-release-7-9c.0.h7.helix.x86_64.rpm'
+        }.get(releasever)
+        helix_release_pkg = '/opt/zstack-dvd/x86_64/%s/Packages/%s' % (
+            distro_name, pkg_name)
+        install_cmd = ('yum install -y /opt/%s && sed -i "/distroverpkg=/d; '
+                       '/bugtracker_url=/d" /etc/yum.conf') % pkg_name
+
+        copy_arg = CopyArg()
+        copy_arg.src = helix_release_pkg
+        copy_arg.dest = '/opt'
+        copy(copy_arg, host_post_info)
+        run_remote_command(install_cmd, host_post_info)
+
+        # flush ansible cache after upgrading
+        global _ansible_cache
+        _ansible_cache = {}
+        host_info = get_remote_host_info_obj(host_post_info)
+
+    return host_info
+
+
 def install_release_on_host(is_rpm, host_info, host_post_info):
     releasever = get_host_releasever(host_info)
     if is_rpm:
diff --git a/zstacknetwork/ansible/zsnagentansible.py b/zstacknetwork/ansible/zsnagentansible.py
@@ -69,6 +69,7 @@ def add_true_in_command(cmd):
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
diff --git a/zstacksharedblock/ansible/zsblkagentansible.py b/zstacksharedblock/ansible/zsblkagentansible.py
@@ -71,6 +71,7 @@ def add_true_in_command(cmd):
 
 # include zstacklib.py
 host_info = get_remote_host_info_obj(host_post_info)
+host_info = upgrade_to_helix(host_info, host_post_info)
 releasever = get_host_releasever(host_info)
 host_post_info.releasever = releasever
 
