Skip to content

Instantly share code, notes, and snippets.

@mattscilipoti

mattscilipoti/postgres.rake

Last active May 1, 2019 18:42
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save mattscilipoti/4455341 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save mattscilipoti/4455341 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. mattscilipoti renamed this gist Jan 4, 2013. 1 changed file with 0 additions and 0 deletions.
    0 gistfile1 → postgres.rake
    View file
    File renamed without changes.
  2. mattscilipoti created this gist Jan 4, 2013.
    30 changes: 30 additions & 0 deletions gistfile1
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,30 @@
    puts "WARN: We are patching rake tasks to support postgres. You have upgraded Rails and may not need this patch any more (lib/tasks/postgres.rake)." if Rails.version > "3.2.8"

    namespace :db do
    desc 'kills connections to postgres db'
    task :kill_postgres_connections => :environment do
    env = ENV['RAILS_ENV']
    db_config = Rails.configuration.database_configuration[env]
    fail(ArgumentError, "Could not find db config entry for env (#{env})") unless db_config
    db_name = db_config['database']

    # thanks to http://stackoverflow.com/questions/12924466/capistrano-with-postgresql-error-database-is-being-accessed-by-other-users
    # previously, we were kill'ing the postgres processes: http://stackoverflow.com/questions/2369744/rails-postgres-drop-error-database-is-being-accessed-by-other-users
    cmd = %(psql -c "SELECT procpid, pg_terminate_backend(procpid) as terminated FROM pg_stat_activity WHERE procpid <> pg_backend_pid();" -d '#{db_name}')

    puts "WARN: killing connections to #{db_name}."
    #puts " Using:\n #{cmd}"

    unless system(cmd)
    fail $?.inspect
    end
    end

    desc 'Alias for :kill_postgres_connections'
    task :kill => :kill_postgres_connections
    end

    #
    # This patch ensures db:drop works for postgres db, by killing the postgres connections
    #
    task 'db:drop' => 'db:kill_postgres_connections'