backup revision df4ff7ec6be98e7ae5830731becc1a3d55105378
0N/A#!/usr/bin/env ruby
0N/A
0N/Arequire 'fileutils'
0N/Arequire 'pathname'
0N/A
0N/Amodule Backup
0N/A class Backup
0N/A # Amount of backups that have to be there at least
0N/A BACKUPS_COUNT = 3
0N/A # Backups are valid for 7 days
0N/A BACKUPS_VALIDITY_TIME = 7 * 60 * 60 * 24
0N/A
0N/A MAINTENANCE_FILE = 'maintenance.txt'
0N/A
0N/A SQL_DUMP_FILE = 'ontohub_sql_dump.postgresql'
0N/A REPOSITORY_FILE = 'ontohub_repositories.tar.gz'
0N/A
0N/A DATA_DIRS = %w(data/repositories data/git_daemon)
0N/A
0N/A attr_reader :db_name, :data_root, :backup_root, :backup_instance_dir
0N/A attr_reader :dry_run, :verbose, :sql_dump_without_su
0N/A
0N/A def initialize(db_name, data_root, backup_root,
0N/A verbose: false, dry_run: true, sql_dump_without_su: false)
661N/A @db_name = db_name
661N/A @data_root = Pathname.new(data_root)
0N/A @backup_root = Pathname.new(backup_root)
0N/A
0N/A @dry_run = dry_run
0N/A @verbose = verbose
0N/A @sql_dump_without_su = sql_dump_without_su
0N/A end
0N/A
0N/A def create
0N/A enable_maintenance_mode
0N/A initialize_backup
0N/A create_sql_dump
0N/A create_repository_archive
0N/A # We needed to create the directory for the script to continue later on.
0N/A Dir.rmdir(backup_instance_dir) if dry_run
0N/A disable_maintenance_mode
0N/A puts "created backup in #{backup_instance_dir}"
0N/A self.class.prune(backup_root)
0N/A end
0N/A
0N/A def restore(backup_name)
0N/A enable_maintenance_mode
0N/A initialize_restore(backup_name)
0N/A restore_sql_dump
0N/A restore_repository_archive
0N/A disable_maintenance_mode
0N/A puts "restored backup from #{backup_instance_dir}"
0N/A end
0N/A
0N/A def self.prune(backup_root)
0N/A if !Dir.exists?(backup_root)
0N/A puts "Nothing to prune: There is no backup directory."
0N/A return
0N/A end
0N/A now = Time.now
0N/A backup_dirs_allowed_to_delete(Dir.new(backup_root).entries).each do |dir|
0N/A backup = backup_root.join(dir)
0N/A if now - File.new(backup).ctime > BACKUPS_VALIDITY_TIME
0N/A puts "removing old backup: #{dir}"
0N/A FileUtils.rm_r(backup)
0N/A end
0N/A end
0N/A end
0N/A
0N/A protected
0N/A
0N/A def new_backup_name
0N/A Time.now.strftime("%Y-%m-%d_%H-%M-%S")
661N/A end
661N/A
661N/A def initialize_backup
661N/A @backup_instance_dir = backup_root.join(new_backup_name)
661N/A puts "FileUtils.mkdir_p #{backup_instance_dir}" if verbose
661N/A # Create directory even in dry run to let the script continue.
0N/A FileUtils.mkdir_p(backup_instance_dir)
0N/A end
0N/A
0N/A def create_sql_dump
0N/A Dir.chdir(backup_instance_dir) do
0N/A if sql_dump_without_su
0N/A exec "pg_dump -U postgres -Fc #{db_name} > #{SQL_DUMP_FILE}"
0N/A else
0N/A exec "sudo -u postgres pg_dump -Fc #{db_name} > #{SQL_DUMP_FILE}"
0N/A end
661N/A end
661N/A end
661N/A
0N/A def create_repository_archive
661N/A Dir.chdir(data_root.join('..')) do
661N/A archive_file = backup_instance_dir.join(REPOSITORY_FILE)
661N/A exec(["tar -czvf #{archive_file}", *DATA_DIRS].join(' '))
661N/A end
661N/A end
0N/A
0N/A def initialize_restore(backup_name)
0N/A @backup_instance_dir = backup_root.join(backup_name)
0N/A unless Dir.exists?(backup_instance_dir)
0N/A puts "Error: Backup '#{backup_name}' does not exist in #{backup_root}."
0N/A exit
0N/A end
0N/A end
0N/A
0N/A def restore_sql_dump
0N/A Dir.chdir(backup_instance_dir) do
0N/A if sql_dump_without_su
0N/A exec "pg_restore -c -U postgres -d #{db_name} #{SQL_DUMP_FILE}"
0N/A else
0N/A exec "sudo -u postgres pg_restore -c -d #{db_name} #{SQL_DUMP_FILE}"
0N/A end
0N/A end
0N/A end
0N/A
0N/A def restore_repository_archive
0N/A Dir.chdir(data_root.join('..')) do
0N/A puts "FileUtils.rm_r #{DATA_DIRS}" if verbose
0N/A FileUtils.rm_r DATA_DIRS unless dry_run
0N/A archive_file = backup_instance_dir.join(REPOSITORY_FILE)
0N/A exec(["tar -xzvf #{archive_file}", *DATA_DIRS].join(' '))
661N/A end
661N/A end
661N/A
0N/A def enable_maintenance_mode
0N/A puts "FileUtils.touch #{maintenance_file}" if verbose
0N/A FileUtils.touch maintenance_file unless dry_run
0N/A end
0N/A
0N/A def disable_maintenance_mode
0N/A puts "FileUtils.rm #{maintenance_file}" if verbose
0N/A FileUtils.rm maintenance_file unless dry_run
0N/A end
0N/A
0N/A def exec(command)
0N/A puts "[executing next command in #{Dir.getwd}]" if verbose
0N/A puts command if verbose
0N/A system(command) unless dry_run
0N/A end
0N/A
0N/A def maintenance_file
0N/A data_root.join(MAINTENANCE_FILE)
0N/A end
0N/A
0N/A def self.backup_dirs_allowed_to_delete(entries)
0N/A entries.reject{ |entry| %w(. ..).include?(entry) }[0..-(BACKUPS_COUNT+1)]
0N/A end
0N/A end
0N/Aend
0N/A
0N/Adef data_root(rails_root)
0N/A File.realpath(rails_root.join('data'))
0N/Aend
0N/A
0N/Adef on_development_system?(rails_root)
0N/A !File.symlink?(rails_root.join('data'))
0N/Aend
0N/A
0N/A# We assume, this script runs in "RAILS_ROOT/script/".
0N/ARAILS_ROOT = Pathname.new(__FILE__).dirname.join('..')
0N/A
0N/ADATABASE = if on_development_system?(RAILS_ROOT)
0N/A 'ontohub_development'
0N/A else
0N/A 'ontohub'
0N/A end
0N/ABACKUP_ROOT = if on_development_system?(RAILS_ROOT)
0N/A RAILS_ROOT.join('tmp', 'backup')
0N/A else
0N/A File.realpath('/home/ontohub/ontohub_data_backup')
0N/A end
0N/A
0N/Abackup = Backup::Backup.new(DATABASE, data_root(RAILS_ROOT), BACKUP_ROOT,
0N/A sql_dump_without_su: on_development_system?(RAILS_ROOT),
0N/A dry_run: false, verbose: true)
0N/A
0N/Acase ARGV.first
0N/Awhen 'create'
0N/A backup.create
0N/Awhen 'restore'
0N/A if ARGV.length == 1
0N/A puts 'To restore a backup, you need to specify one with the arguments'
0N/A puts '"restore backup_name"'
0N/A exit
0N/A end
0N/A backup_name = ARGV[1]
0N/A backup.restore(backup_name)
0N/Awhen 'prune'
0N/A Backup::Backup.prune(BACKUP_ROOT)
0N/Aelse
0N/A puts 'unknown or missing parameter'
0N/A puts 'use parameter "create" or "restore <backup_name>" or "prune"'
0N/A exit
0N/Aend
0N/A