diff options
author | Stan Hu <stanhu@gmail.com> | 2019-01-29 01:46:05 +0300 |
---|---|---|
committer | Stan Hu <stanhu@gmail.com> | 2019-02-12 08:30:57 +0300 |
commit | 5488e8a4a3a2bb1493a58429d53e32c94f18114e (patch) | |
tree | d75871be74807056d0a0b206ad35cc5082f8f069 /lib | |
parent | 91b88e0be4a943aa7000c7d921e77fc200c53500 (diff) |
Kill all PostgreSQL connections for rake dev:setup
If there are any clients connected to the DB, PostgreSQL won't let you
drop the database. It's possible that Sidekiq, Unicorn, or some other
client will be hanging onto a connection, preventing the DROP DATABASE
from working. To workaround this problem, this method cancels all the
connections so that the db:reset command will work.
Note that there's still a slight possibility a client connects after its
connection is terminated. If this is an issue, we could solve it by
revoking CONNECT access, but for now it seems this works.
Closes https://gitlab.com/gitlab-org/gitlab-development-kit/issues/450
Diffstat (limited to 'lib')
-rw-r--r-- | lib/gitlab/database.rb | 22 | ||||
-rw-r--r-- | lib/tasks/gitlab/setup.rake | 5 |
2 files changed, 27 insertions, 0 deletions
diff --git a/lib/gitlab/database.rb b/lib/gitlab/database.rb index b6ca777e029..817b5b97a3b 100644 --- a/lib/gitlab/database.rb +++ b/lib/gitlab/database.rb @@ -235,6 +235,28 @@ module Gitlab connection.schema_cache.data_source_exists?(table_name) end + # WARNING: Only run this if you know what you're doing. + # + # If there are any clients connected to the DB, PostgreSQL won't let + # you drop the database. It's possible that Sidekiq, Unicorn, or + # some other client will be hanging onto a connection, preventing + # the DROP DATABASE from working. To workaround this problem, this + # method terminates all the connections so that a subsequent DROP + # will work. + def self.terminate_all_connections + return false unless Gitlab::Database.postgresql? + + cmd = <<~SQL + SELECT pg_terminate_backend(pg_stat_activity.pid) + FROM pg_stat_activity + WHERE datname = current_database() + AND pid <> pg_backend_pid(); + SQL + + connection.execute(cmd)&.result_status == PG::PGRES_TUPLES_OK + rescue ActiveRecord::NoDatabaseError + end + private_class_method :connection def self.database_version diff --git a/lib/tasks/gitlab/setup.rake b/lib/tasks/gitlab/setup.rake index f71e69987cb..dab96c5c54d 100644 --- a/lib/tasks/gitlab/setup.rake +++ b/lib/tasks/gitlab/setup.rake @@ -25,6 +25,11 @@ namespace :gitlab do puts "" end + # In production, we might want to prevent ourselves from shooting + # ourselves in the foot, so let's only do this in a test or + # development environment. + Gitlab::Database.terminate_all_connections unless Rails.env.production? + Rake::Task["db:reset"].invoke Rake::Task["add_limits_mysql"].invoke Rake::Task["setup_postgresql"].invoke |