/lib/backup/files.rb

https://gitlab.com/vicvega/gitlab-ce · Ruby · 40 lines · 32 code · 7 blank · 1 comment · 1 complexity · 346a5e9c689ec6068abc5281545ee068 MD5 · raw file

  1. require 'open3'
  2. module Backup
  3. class Files
  4. attr_reader :name, :app_files_dir, :backup_tarball, :files_parent_dir
  5. def initialize(name, app_files_dir)
  6. @name = name
  7. @app_files_dir = File.realpath(app_files_dir)
  8. @files_parent_dir = File.realpath(File.join(@app_files_dir, '..'))
  9. @backup_tarball = File.join(Gitlab.config.backup.path, name + '.tar.gz')
  10. end
  11. # Copy files from public/files to backup/files
  12. def dump
  13. FileUtils.mkdir_p(Gitlab.config.backup.path)
  14. FileUtils.rm_f(backup_tarball)
  15. run_pipeline!([%W(tar -C #{app_files_dir} -cf - .), %W(gzip -c -1)], out: [backup_tarball, 'w', 0600])
  16. end
  17. def restore
  18. backup_existing_files_dir
  19. create_files_dir
  20. run_pipeline!([%W(gzip -cd), %W(tar -C #{app_files_dir} -xf -)], in: backup_tarball)
  21. end
  22. def backup_existing_files_dir
  23. timestamped_files_path = File.join(files_parent_dir, "#{name}.#{Time.now.to_i}")
  24. if File.exist?(app_files_dir)
  25. FileUtils.mv(app_files_dir, File.expand_path(timestamped_files_path))
  26. end
  27. end
  28. def run_pipeline!(cmd_list, options={})
  29. status_list = Open3.pipeline(*cmd_list, options)
  30. abort 'Backup failed' unless status_list.compact.all?(&:success?)
  31. end
  32. end
  33. end