Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # Directly copied from eycap-0.5.2 (thanks!)
- #
- # With these tasks you can:
- # - dump your production database and save it in current_path/db_backups
- # - dump your production into your local database (clone_to_local)
- #
- # Tested and fixed by fjguzman, prosenstein
- namespace :db do
- task :backup_name, :roles => :db, :only => { :primary => true } do
- now = Time.now
- run "mkdir -p #{current_path}/db_backups"
- backup_time = [now.year,now.month,now.day,now.hour,now.min,now.sec].join('-')
- set :backup_file, "#{current_path}/db_backups/#{@environment_database}-snapshot-#{backup_time}.sql"
- end
- desc "Backup your MySQL or PostgreSQL database to current_path+/db_backups"
- task :dump, :roles => :db, :only => {:primary => true} do
- run("cat #{current_path}/config/database.yml") { |channel, stream, data| @environment_info = YAML.load(data)[rails_env] }
- dbuser = @environment_info['username']
- dbpass = @environment_info['password']
- @environment_database = @environment_info['database']
- dbhost = @environment_info['host']
- dbhost_switch = "-h #{dbhost}" if dbhost && !dbhost.blank?
- backup_name
- if @environment_info['adapter'] == 'mysql'
- #dbhost = environment_dbhost.sub('-master', '') + '-replica' if dbhost != 'localhost' # added for Solo offering, which uses localhost
- run "mysqldump --add-drop-table -u #{dbuser} #{dbhost_switch} -p #{@environment_database} | bzip2 -c > #{backup_file}.bz2" do |ch, stream, out |
- ch.send_data "#{dbpass}\n" if out=~ /^Enter password:/
- end
- else
- run "pg_dump -W -c -U #{dbuser} #{dbhost_switch} #{@environment_database} | bzip2 -c > #{backup_file}.bz2" do |ch, stream, out |
- ch.send_data "#{dbpass}\n" if out=~ /^Password:/
- end
- end
- end
- desc "Sync your production database to your local workstation"
- task :clone_to_local, :roles => :db, :only => {:primary => true} do
- dump
- get "#{backup_file}.bz2", "/tmp/#{application}.sql.bz2"
- development_info = YAML.load_file("config/database.yml")['development']
- dbhost = development_info['host']
- dbhost_switch = "-h #{dbhost}" if dbhost && !dbhost.blank?
- if development_info['adapter'] == 'mysql'
- run_str = "bzcat /tmp/#{application}.sql.bz2 | mysql -u #{development_info['username']} --password='#{development_info['password']}' #{dbhost_switch} #{development_info['database']}"
- else
- run_str = "PGPASSWORD=#{development_info['password']} bzcat /tmp/#{application}.sql.bz2 | psql -U #{development_info['username']} #{dbhost_switch} #{development_info['database']}"
- end
- %x!#{run_str}!
- end
- end
Add Comment
Please, Sign In to add comment