backup/restore script

Author: dorner

lib/tasks/backup_db_rds.rake

@@ -1,3 +1,5 @@
+require 'aws-sdk-s3'
+
 desc "Update the development db to what is being used in prod"
 task :backup_db_rds => :environment do
   logger = Logger.new(STDOUT)
@@ -6,18 +8,12 @@ task :backup_db_rds => :environment do
   current_time = Time.current.strftime("%Y%m%d%H%M%S")
 
   logger.info("Copying the database...")
-  backup_filename = "#{current_time}.rds.dump"
+  backup_filename = "#{Rails.env}-#{current_time}.rds.dump"
   system("PGPASSWORD='#{ENV["DIAPER_DB_PASSWORD"]}' pg_dump -Fc -v --host=#{ENV["DIAPER_DB_HOST"]} --username=#{ENV["DIAPER_DB_USERNAME"]} --dbname=#{ENV["DIAPER_DB_DATABASE"]} -f #{backup_filename}")
 
-  account_name = ENV["AZURE_STORAGE_ACCOUNT_NAME"]
-  account_key = ENV["AZURE_STORAGE_ACCESS_KEY"]
-
-  blob_client = Azure::Storage::Blob::BlobService.create(
-    storage_account_name: account_name,
-    storage_access_key: account_key
-  )
+  client = Aws::S3::Client.new
 
   logger.info("Uploading #{backup_filename}")
-  blob_client.create_block_blob("backups", backup_filename, File.read(backup_filename))
+  client.put_object(bucket: "human-essentials-backups", key: "backups/#{backup_filename}", body: File.read(backup_filename))
   File.delete(backup_filename)
 end

lib/tasks/fetch_latest_db.rake

@@ -1,5 +1,7 @@
+require 'aws-sdk-s3'
+
 desc "Update the development db to what is being used in prod"
-BACKUP_CONTAINER_NAME = 'backups'
+BUCKET_NAME = "human-essentials-backups"
 PASSWORD_REPLACEMENT = 'password'
 
 task :fetch_latest_db do
@@ -51,44 +53,31 @@ end
 private
 
 def fetch_latest_backups
-  backups = blob_client.list_blobs(BACKUP_CONTAINER_NAME)
+  backups = blob_client.list_objects_v2(bucket: BUCKET_NAME)
 
   #
   # Retrieve the most up to date version of the DB dump
   #
   backup = backups.select { |b| b.name.match?(".rds.dump") }.sort do |a,b|
-    Time.parse(a.properties[:last_modified]) <=> Time.parse(b.properties[:last_modified])
+    Time.parse(a.last_modified) <=> Time.parse(b.last_modified)
   end.reverse.first
 
   #
   # Download each of the backups onto the local disk in tmp
   #
   filepath = fetch_file_path(backup)
   puts "\nDownloading blob #{backup.name} to #{filepath}"
-  blob, content = blob_client.get_blob(BACKUP_CONTAINER_NAME, backup.name)
-  File.open(filepath, "wb") { |f| f.write(content)  }
+  blob_client.get_object(bucket: BUCKET_NAME, key: backup.name, response_target: filepath)
 
   #
   # At this point, the dumps should be stored on the local
   # machine of the user under tmp.
   #
-  return backup
+  backup
 end
 
 def blob_client
-  return @blob_client if @blob_client
-
-  account_name = ENV["AZURE_STORAGE_ACCOUNT_NAME"]
-  account_key = ENV["AZURE_STORAGE_ACCESS_KEY"]
-
-  if account_name.blank? || account_key.blank?
-    raise "You must have the correct azure credentials in your ENV"
-  end
-
-  @blob_client = Azure::Storage::Blob::BlobService.create(
-    storage_account_name: account_name,
-    storage_access_key: account_key
-  )
+  Aws::S3::Client.new
 end
 
 def fetch_file_path(backup)