From e00e67db42ea3e4b994160dbdc288a8effa14713 Mon Sep 17 00:00:00 2001
From: Jacob Vosmaer <contact@jacobvosmaer.nl>
Date: Tue, 28 Oct 2014 18:52:21 +0100
Subject: [PATCH] Drop all Postgres sequences during backup restore
---
lib/backup/database.rb | 1 +
lib/tasks/gitlab/db/drop_all_postgres_sequences.rake | 10 ++++++++++
2 files changed, 11 insertions(+)
create mode 100644 lib/tasks/gitlab/db/drop_all_postgres_sequences.rake
diff --git a/lib/backup/database.rb b/lib/backup/database.rb
index d12d30a9110..ea659e3b605 100644
--- a/lib/backup/database.rb
+++ b/lib/backup/database.rb
@@ -34,6 +34,7 @@ module Backup
# Drop all tables because PostgreSQL DB dumps do not contain DROP TABLE
# statements like MySQL.
Rake::Task["gitlab:db:drop_all_tables"].invoke
+ Rake::Task["gitlab:db:drop_all_postgres_sequences"].invoke
pg_env
system('psql', config['database'], '-f', db_file_name)
end
diff --git a/lib/tasks/gitlab/db/drop_all_postgres_sequences.rake b/lib/tasks/gitlab/db/drop_all_postgres_sequences.rake
new file mode 100644
index 00000000000..e9cf0a9b5e8
--- /dev/null
+++ b/lib/tasks/gitlab/db/drop_all_postgres_sequences.rake
@@ -0,0 +1,10 @@
+namespace :gitlab do
+ namespace :db do
+ task drop_all_postgres_sequences: :environment do
+ connection = ActiveRecord::Base.connection
+ connection.execute("SELECT c.relname FROM pg_class c WHERE c.relkind = 'S';").each do |sequence|
+ connection.execute("DROP SEQUENCE #{sequence['relname']}")
+ end
+ end
+ end
+end
--
GitLab