summaryrefslogtreecommitdiff
path: root/lib/gitlab/database/postgres_foreign_key.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/gitlab/database/postgres_foreign_key.rb')
-rw-r--r--lib/gitlab/database/postgres_foreign_key.rb6
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/gitlab/database/postgres_foreign_key.rb b/lib/gitlab/database/postgres_foreign_key.rb
index 72640f8785d..241b6f009f7 100644
--- a/lib/gitlab/database/postgres_foreign_key.rb
+++ b/lib/gitlab/database/postgres_foreign_key.rb
@@ -10,6 +10,12 @@ module Gitlab
where(referenced_table_identifier: identifier)
end
+
+ scope :by_constrained_table_identifier, ->(identifier) do
+ raise ArgumentError, "Constrained table name is not fully qualified with a schema: #{identifier}" unless identifier =~ /^\w+\.\w+$/
+
+ where(constrained_table_identifier: identifier)
+ end
end
end
end