debian-mirror-gitlab/lib/backup/files.rb

86 lines
2.8 KiB
Ruby
Raw Normal View History

2018-12-05 23:21:45 +05:30
# frozen_string_literal: true
2015-11-26 14:37:03 +05:30
require 'open3'
2018-05-09 12:01:36 +05:30
require_relative 'helper'
2015-11-26 14:37:03 +05:30
module Backup
class Files
2018-05-09 12:01:36 +05:30
include Backup::Helper
2015-11-26 14:37:03 +05:30
attr_reader :name, :app_files_dir, :backup_tarball, :files_parent_dir
def initialize(name, app_files_dir)
@name = name
@app_files_dir = File.realpath(app_files_dir)
@files_parent_dir = File.realpath(File.join(@app_files_dir, '..'))
2017-08-17 22:00:37 +05:30
@backup_files_dir = File.join(Gitlab.config.backup.path, File.basename(@app_files_dir) )
2015-11-26 14:37:03 +05:30
@backup_tarball = File.join(Gitlab.config.backup.path, name + '.tar.gz')
end
# Copy files from public/files to backup/files
def dump
FileUtils.mkdir_p(Gitlab.config.backup.path)
FileUtils.rm_f(backup_tarball)
2017-08-17 22:00:37 +05:30
if ENV['STRATEGY'] == 'copy'
2018-03-17 18:26:18 +05:30
cmd = %W(rsync -a --exclude=lost+found #{app_files_dir} #{Gitlab.config.backup.path})
2017-08-17 22:00:37 +05:30
output, status = Gitlab::Popen.popen(cmd)
unless status.zero?
puts output
2018-11-08 19:23:39 +05:30
raise Backup::Error, 'Backup failed'
2017-08-17 22:00:37 +05:30
end
2019-07-07 11:18:12 +05:30
run_pipeline!([%W(#{tar} --exclude=lost+found -C #{@backup_files_dir} -cf - .), gzip_cmd], out: [backup_tarball, 'w', 0600])
2017-08-17 22:00:37 +05:30
FileUtils.rm_rf(@backup_files_dir)
else
2019-07-07 11:18:12 +05:30
run_pipeline!([%W(#{tar} --exclude=lost+found -C #{app_files_dir} -cf - .), gzip_cmd], out: [backup_tarball, 'w', 0600])
2017-08-17 22:00:37 +05:30
end
2015-11-26 14:37:03 +05:30
end
def restore
backup_existing_files_dir
2018-11-08 19:23:39 +05:30
run_pipeline!([%w(gzip -cd), %W(#{tar} --unlink-first --recursive-unlink -C #{app_files_dir} -xf -)], in: backup_tarball)
end
def tar
if system(*%w[gtar --version], out: '/dev/null')
# It looks like we can get GNU tar by running 'gtar'
'gtar'
else
'tar'
end
2015-11-26 14:37:03 +05:30
end
def backup_existing_files_dir
2018-05-09 12:01:36 +05:30
timestamped_files_path = File.join(Gitlab.config.backup.path, "tmp", "#{name}.#{Time.now.to_i}")
2016-09-13 17:45:13 +05:30
if File.exist?(app_files_dir)
2018-05-09 12:01:36 +05:30
# Move all files in the existing repos directory except . and .. to
# repositories.old.<timestamp> directory
FileUtils.mkdir_p(timestamped_files_path, mode: 0700)
files = Dir.glob(File.join(app_files_dir, "*"), File::FNM_DOTMATCH) - [File.join(app_files_dir, "."), File.join(app_files_dir, "..")]
begin
FileUtils.mv(files, timestamped_files_path)
rescue Errno::EACCES
access_denied_error(app_files_dir)
2018-10-15 14:42:47 +05:30
rescue Errno::EBUSY
resource_busy_error(app_files_dir)
2018-05-09 12:01:36 +05:30
end
2015-11-26 14:37:03 +05:30
end
end
2017-08-17 22:00:37 +05:30
def run_pipeline!(cmd_list, options = {})
2019-03-02 22:35:43 +05:30
err_r, err_w = IO.pipe
options[:err] = err_w
status = Open3.pipeline(*cmd_list, options)
err_w.close
return if status.compact.all?(&:success?)
regex = /^g?tar: \.: Cannot mkdir: No such file or directory$/
2020-06-23 00:09:42 +05:30
error = err_r.read
raise Backup::Error, "Backup failed. #{error}" unless error =~ regex
2015-11-26 14:37:03 +05:30
end
end
end