Sha256: 45e29b75d49b1999ec07cc57b02ad850669dfc5eb39d01731f858b452f9e689d
Contents?: true
Size: 1.45 KB
Versions: 6
Compression:
Stored size: 1.45 KB
Contents
require "heroku/client"
require "tmpdir"
class Heroku::Client::PgbackupsArchive
attr_reader :client, :pgbackup
def self.perform
backup = new
backup.capture
backup.download
backup.archive
backup.delete
end
def initialize(attrs={})
Heroku::Command.load
@client = Heroku::Client::Pgbackups.new pgbackups_url
@pgbackup = nil
end
def archive
PgbackupsArchive::Storage.new(key, file).store
end
def capture
@pgbackup = @client.create_transfer(database_url, database_url, nil,
"BACKUP", :expire => true)
until @pgbackup["finished_at"]
print "."
sleep 1
@pgbackup = @client.get_transfer @pgbackup["id"]
end
end
def delete
File.delete temp_file
end
def download
File.open(temp_file, "wb") do |output|
streamer = lambda do |chunk, remaining_bytes, total_bytes|
output.write chunk
end
Excon.get(@pgbackup["public_url"], :response_block => streamer)
end
end
private
def database_url
ENV["PGBACKUPS_DATABASE_URL"] || ENV["DATABASE_URL"]
end
def environment
defined?(Rails) ? Rails.env : nil
end
def file
File.open temp_file, "r"
end
def key
["pgbackups", environment, @pgbackup["finished_at"]
.gsub(/\/|\:|\.|\s/, "-").concat(".dump")].compact.join("/")
end
def pgbackups_url
ENV["PGBACKUPS_URL"]
end
def temp_file
"#{Dir.tmpdir}/#{URI(@pgbackup['public_url']).path.split('/').last}"
end
end
Version data entries
6 entries across 6 versions & 1 rubygems