|
|
@@ -0,0 +1,102 @@
|
|
|
+"""
|
|
|
+Licensed to the Apache Software Foundation (ASF) under one
|
|
|
+or more contributor license agreements. See the NOTICE file
|
|
|
+distributed with this work for additional information
|
|
|
+regarding copyright ownership. The ASF licenses this file
|
|
|
+to you under the Apache License, Version 2.0 (the
|
|
|
+"License"); you may not use this file except in compliance
|
|
|
+with the License. You may obtain a copy of the License at
|
|
|
+
|
|
|
+ http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+
|
|
|
+Unless required by applicable law or agreed to in writing, software
|
|
|
+distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
+See the License for the specific language governing permissions and
|
|
|
+limitations under the License.
|
|
|
+
|
|
|
+"""
|
|
|
+import os
|
|
|
+import shutil
|
|
|
+import tarfile
|
|
|
+import tempfile
|
|
|
+
|
|
|
+from resource_management.core.logger import Logger
|
|
|
+from resource_management.core.exceptions import Fail
|
|
|
+
|
|
|
+BACKUP_TEMP_DIR = "falcon-upgrade-backup"
|
|
|
+BACKUP_DATA_ARCHIVE = "falcon-local-backup.tar"
|
|
|
+BACKUP_CONF_ARCHIVE = "falcon-conf-backup.tar"
|
|
|
+
|
|
|
+def post_stop_backup():
|
|
|
+ """
|
|
|
+ Backs up the falcon configuration and data directories as part of the
|
|
|
+ upgrade process.
|
|
|
+ :return:
|
|
|
+ """
|
|
|
+ Logger.info('Backing up Falcon data and configuration directories before upgrade...')
|
|
|
+ directoryMappings = _get_directory_mappings()
|
|
|
+
|
|
|
+ absolute_backup_dir = os.path.join(tempfile.gettempdir(), BACKUP_TEMP_DIR)
|
|
|
+ if not os.path.isdir(absolute_backup_dir):
|
|
|
+ os.makedirs(absolute_backup_dir)
|
|
|
+
|
|
|
+ for directory in directoryMappings:
|
|
|
+ if not os.path.isdir(directory):
|
|
|
+ raise Fail("Unable to backup missing directory {0}".format(directory))
|
|
|
+
|
|
|
+ archive = os.path.join(absolute_backup_dir, directoryMappings[directory])
|
|
|
+ Logger.info('Compressing {0} to {1}'.format(directory, archive))
|
|
|
+
|
|
|
+ if os.path.exists(archive):
|
|
|
+ os.remove(archive)
|
|
|
+
|
|
|
+ tarball = None
|
|
|
+ try:
|
|
|
+ tarball = tarfile.open(archive, "w")
|
|
|
+ tarball.add(directory, arcname=os.path.basename(directory))
|
|
|
+ finally:
|
|
|
+ if tarball:
|
|
|
+ tarball.close()
|
|
|
+
|
|
|
+
|
|
|
+def pre_start_restore():
|
|
|
+ """
|
|
|
+ Restores the data and configuration backups to their proper locations
|
|
|
+ after an upgrade has completed.
|
|
|
+ :return:
|
|
|
+ """
|
|
|
+ Logger.info('Restoring Falcon data and configuration directories after upgrade...')
|
|
|
+ directoryMappings = _get_directory_mappings()
|
|
|
+
|
|
|
+ for directory in directoryMappings:
|
|
|
+ archive = os.path.join(tempfile.gettempdir(), BACKUP_TEMP_DIR,
|
|
|
+ directoryMappings[directory])
|
|
|
+
|
|
|
+ if not os.path.isfile(archive):
|
|
|
+ raise Fail("Unable to restore missing backup archive {0}".format(archive))
|
|
|
+
|
|
|
+ Logger.info('Extracting {0} to {1}'.format(archive, directory))
|
|
|
+
|
|
|
+ tarball = None
|
|
|
+ try:
|
|
|
+ tarball = tarfile.open(archive, "r")
|
|
|
+ tarball.extractall(directory)
|
|
|
+ finally:
|
|
|
+ if tarball:
|
|
|
+ tarball.close()
|
|
|
+
|
|
|
+ # cleanup
|
|
|
+ shutil.rmtree(os.path.join(tempfile.gettempdir(), BACKUP_TEMP_DIR))
|
|
|
+
|
|
|
+
|
|
|
+def _get_directory_mappings():
|
|
|
+ """
|
|
|
+ Gets a dictionary of directory to archive name that represents the
|
|
|
+ directories that need to be backed up and their output tarball archive targets
|
|
|
+ :return: the dictionary of directory to tarball mappings
|
|
|
+ """
|
|
|
+ import params
|
|
|
+
|
|
|
+ return { params.falcon_local_dir : BACKUP_DATA_ARCHIVE,
|
|
|
+ params.falcon_conf_dir : BACKUP_CONF_ARCHIVE }
|
Jonathan Hurley