适配CM组件

Signed-off-by: coolany <kyosang@163.com> support cgroup 追加合入
2022-03-03 20:37:33 +08:00
parent 8874d89c6f
commit eae422baf3
229 changed files with 17230 additions and 13148 deletions
--- a/script/base_utils/os/compress_util.py
+++ b/script/base_utils/os/compress_util.py
@ -0,0 +1,114 @@
+# -*- coding:utf-8 -*-
+#############################################################################
+# Copyright (c) 2020 Huawei Technologies Co.,Ltd.
+#
+# openGauss is licensed under Mulan PSL v2.
+# You can use this software according to the terms
+# and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS,
+# WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# ----------------------------------------------------------------------------
+# Description  : compress_util.py is utility to support compress and decompress.
+#############################################################################
+
+import os
+import subprocess
+from gspylib.common.ErrorCode import ErrorCode
+from base_utils.os.cmd_util import CmdUtil
+
+
+class CompressUtil:
+    """compress util"""
+
+    @staticmethod
+    def getCompressFilesCmd(tar_name, file_src):
+        """
+        function: get compress file cmd
+        input  : tar_name, file_src
+        output : str
+        """
+        return "%s -zvcf '%s' %s" % (CmdUtil.getTarCmd(), tar_name, file_src)
+
+    @staticmethod
+    def getDecompressFilesCmd(src_package, dest):
+        """
+        function: get decompress file cmd
+        input  : src_package, dest
+        output : str
+        """
+        return "%s -zxvf '%s' -C '%s'" % (CmdUtil.getTarCmd(), src_package, dest)
+
+    @staticmethod
+    def getCompressZipFilesCmd(zip_name, file_src):
+        """
+        function: get compress zip files cmd
+        input  : zip_name, file_src
+        output : str
+        """
+        return "cd %s && %s -r '%s.zip' ./*" % (file_src, CmdUtil.getZipCmd(),
+                                                zip_name)
+
+    @staticmethod
+    def getDecompressZipFilesCmd(src_package, dest):
+        """
+        function: get decompress zip files cmd
+        input  : src_package, dest
+        output : str
+        """
+        return "%s -o '%s' -d '%s'" % (CmdUtil.getUnzipCmd(), src_package, dest)
+
+    @staticmethod
+    def decompressFiles(src_package, dest):
+        """
+        function:decompress package to files
+        input:src_package, dest
+        output:NA
+        """
+        if not os.path.exists(src_package):
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50201"] % src_package)
+        if not os.path.exists(dest):
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50201"] % dest)
+        cmd = CompressUtil.getDecompressFilesCmd(src_package, dest)
+        (status, output) = subprocess.getstatusoutput(cmd)
+        if status != 0:
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50231"] % src_package +
+                            " Error:\n%s" % output + "\nThe cmd is %s" % cmd)
+
+    @staticmethod
+    def compressZipFiles(zip_name, dir_path):
+        """
+        function:compress directory to a package
+        input:zip_name, directory
+        output:NA
+        """
+        if not os.path.exists(dir_path):
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50201"] % dir_path)
+        cmd = CompressUtil.getCompressZipFilesCmd(zip_name, dir_path)
+        (status, output) = subprocess.getstatusoutput(cmd)
+        if status != 0:
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50227"] % cmd +
+                            " Error:\n%s" % output)
+
+    @staticmethod
+    def decompressZipFiles(src_package, dest):
+        """
+        function:decompress package to files
+        input:src_package, dest
+        output:NA
+        """
+        if not os.path.exists(src_package):
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50201"] % src_package)
+        if not os.path.exists(dest):
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50201"] % dest)
+        cmd = CompressUtil.getDecompressZipFilesCmd(src_package, dest)
+        (status, output) = subprocess.getstatusoutput(cmd)
+        if status != 0:
+            raise Exception(ErrorCode.GAUSS_502["GAUSS_50231"] % src_package +
+                            " Error:\n%s" % output + "\nThe cmd is %s" % cmd)