gitlab-org--gitlab-foss/lib/backup/manager.rb

196 lines
6.3 KiB
Ruby
Raw Normal View History

module Backup
class Manager
2016-05-24 23:52:31 -04:00
ARCHIVES_TO_BACKUP = %w[uploads builds artifacts lfs registry]
FOLDERS_TO_BACKUP = %w[repositories db]
2016-05-23 11:17:07 -04:00
def pack
# Make sure there is a connection
ActiveRecord::Base.connection.reconnect!
# saving additional informations
s = {}
s[:db_version] = "#{ActiveRecord::Migrator.current_version}"
s[:backup_created_at] = Time.now
s[:gitlab_version] = Gitlab::VERSION
2014-02-25 06:12:09 -05:00
s[:tar_version] = tar_version
2015-03-27 09:18:59 -04:00
s[:skipped] = ENV["SKIP"]
tar_file = "#{s[:backup_created_at].to_i}_gitlab_backup.tar"
2015-03-19 21:24:57 -04:00
Dir.chdir(Gitlab.config.backup.path) do
File.open("#{Gitlab.config.backup.path}/backup_information.yml",
"w+") do |file|
file << s.to_yaml.gsub(/^---\n/,'')
end
# create archive
$progress.print "Creating backup archive: #{tar_file} ... "
# Set file permissions on open to prevent chmod races.
tar_system_options = {out: [tar_file, 'w', Gitlab.config.backup.archive_permissions]}
if Kernel.system('tar', '-cf', '-', *backup_contents, tar_system_options)
$progress.puts "done".color(:green)
2015-03-19 21:24:57 -04:00
else
puts "creating archive #{tar_file} failed".color(:red)
2015-03-19 21:24:57 -04:00
abort 'Backup failed'
end
upload(tar_file)
end
end
def upload(tar_file)
$progress.print "Uploading backup archive to remote storage #{remote_directory} ... "
connection_settings = Gitlab.config.backup.upload.connection
if connection_settings.blank?
$progress.puts "skipped".color(:yellow)
return
end
directory = connect_to_remote_directory(connection_settings)
2015-03-27 09:18:59 -04:00
if directory.files.create(key: tar_file, body: File.open(tar_file), public: false,
multipart_chunk_size: Gitlab.config.backup.upload.multipart_chunk_size,
encryption: Gitlab.config.backup.upload.encryption)
$progress.puts "done".color(:green)
else
puts "uploading backup to #{remote_directory} failed".color(:red)
abort 'Backup failed'
end
end
def cleanup
$progress.print "Deleting tmp directories ... "
2015-03-27 09:18:59 -04:00
backup_contents.each do |dir|
next unless File.exist?(File.join(Gitlab.config.backup.path, dir))
if FileUtils.rm_rf(File.join(Gitlab.config.backup.path, dir))
$progress.puts "done".color(:green)
else
puts "deleting tmp directory '#{dir}' failed".color(:red)
abort 'Backup failed'
end
end
end
def remove_old
# delete backups
$progress.print "Deleting old backups ... "
keep_time = Gitlab.config.backup.keep_time.to_i
if keep_time > 0
removed = 0
Dir.chdir(Gitlab.config.backup.path) do
file_list = Dir.glob('*_gitlab_backup.tar')
file_list.map! { |f| $1.to_i if f =~ /(\d+)_gitlab_backup.tar/ }
file_list.sort.each do |timestamp|
if Time.at(timestamp) < (Time.now - keep_time)
if Kernel.system(*%W(rm #{timestamp}_gitlab_backup.tar))
removed += 1
end
end
end
end
2015-03-27 09:18:59 -04:00
$progress.puts "done. (#{removed} removed)".color(:green)
else
$progress.puts "skipping".color(:yellow)
end
end
def unpack
Dir.chdir(Gitlab.config.backup.path)
# check for existing backups in the backup dir
file_list = Dir.glob("*_gitlab_backup.tar").each.map { |f| f.split(/_/).first.to_i }
puts "no backups found" if file_list.count == 0
2015-03-27 09:18:59 -04:00
if file_list.count > 1 && ENV["BACKUP"].nil?
puts "Found more than one backup, please specify which one you want to restore:"
puts "rake gitlab:backup:restore BACKUP=timestamp_of_backup"
exit 1
end
tar_file = ENV["BACKUP"].nil? ? File.join("#{file_list.first}_gitlab_backup.tar") : File.join(ENV["BACKUP"] + "_gitlab_backup.tar")
unless File.exists?(tar_file)
puts "The specified backup doesn't exist!"
exit 1
end
$progress.print "Unpacking backup ... "
2015-03-27 09:18:59 -04:00
2013-11-05 09:00:48 -05:00
unless Kernel.system(*%W(tar -xf #{tar_file}))
puts "unpacking backup failed".color(:red)
exit 1
else
$progress.puts "done".color(:green)
end
ENV["VERSION"] = "#{settings[:db_version]}" if settings[:db_version].to_i > 0
# restoring mismatching backups can lead to unexpected problems
if settings[:gitlab_version] != Gitlab::VERSION
puts "GitLab version mismatch:".color(:red)
puts " Your current GitLab version (#{Gitlab::VERSION}) differs from the GitLab version in the backup!".color(:red)
puts " Please switch to the following version and try again:".color(:red)
puts " version: #{settings[:gitlab_version]}".color(:red)
puts
puts "Hint: git checkout v#{settings[:gitlab_version]}"
exit 1
end
end
2014-02-25 06:12:09 -05:00
def tar_version
tar_version, _ = Gitlab::Popen.popen(%W(tar --version))
tar_version.force_encoding('locale').split("\n").first
2014-02-25 06:12:09 -05:00
end
2015-03-27 09:18:59 -04:00
def skipped?(item)
2016-05-24 23:52:31 -04:00
settings[:skipped] && settings[:skipped].include?(item) || disabled_features.include?(item)
2015-03-27 09:18:59 -04:00
end
private
def connect_to_remote_directory(connection_settings)
connection = ::Fog::Storage.new(connection_settings)
# We only attempt to create the directory for local backups. For AWS
# and other cloud providers, we cannot guarantee the user will have
# permission to create the bucket.
if connection.service == ::Fog::Storage::Local
connection.directories.create(key: remote_directory)
else
connection.directories.get(remote_directory)
end
end
def remote_directory
Gitlab.config.backup.upload.remote_directory
end
2015-03-27 09:18:59 -04:00
def backup_contents
folders_to_backup + archives_to_backup + ["backup_information.yml"]
2015-03-27 09:18:59 -04:00
end
def archives_to_backup
2016-05-24 23:52:31 -04:00
ARCHIVES_TO_BACKUP.map{ |name| (name + ".tar.gz") unless skipped?(name) }.compact
end
2015-03-27 09:18:59 -04:00
def folders_to_backup
2016-05-24 23:52:31 -04:00
FOLDERS_TO_BACKUP.reject{ |name| skipped?(name) }
2016-05-23 11:17:07 -04:00
end
def disabled_features
features = []
features << 'registry' unless Gitlab.config.registry.enabled
features
2015-03-27 09:18:59 -04:00
end
def settings
@settings ||= YAML.load_file("backup_information.yml")
end
end
end