From 729e66ee8e5790eefb3771040839155c499faab3 Mon Sep 17 00:00:00 2001 From: GitLab Bot Date: Fri, 27 Jan 2023 03:11:24 +0000 Subject: Add latest changes from gitlab-org/gitlab@master --- spec/support/database/prevent_cross_joins.rb | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'spec/support/database') diff --git a/spec/support/database/prevent_cross_joins.rb b/spec/support/database/prevent_cross_joins.rb index 42c69a26788..8e08824c464 100644 --- a/spec/support/database/prevent_cross_joins.rb +++ b/spec/support/database/prevent_cross_joins.rb @@ -23,6 +23,7 @@ module Database ALLOW_THREAD_KEY = :allow_cross_joins_across_databases ALLOW_ANNOTATE_KEY = ALLOW_THREAD_KEY.to_s.freeze + IGNORED_SCHEMAS = %i[gitlab_shared gitlab_internal].freeze def self.validate_cross_joins!(sql) return if Thread.current[ALLOW_THREAD_KEY] || sql.include?(ALLOW_ANNOTATE_KEY) @@ -40,8 +41,9 @@ module Database end schemas = ::Gitlab::Database::GitlabSchema.table_schemas(tables) + schemas.subtract(IGNORED_SCHEMAS) - if schemas.include?(:gitlab_ci) && schemas.include?(:gitlab_main) + if schemas.many? Thread.current[:has_cross_join_exception] = true raise CrossJoinAcrossUnsupportedTablesError, "Unsupported cross-join across '#{tables.join(", ")}' querying '#{schemas.to_a.join(", ")}' discovered " \ -- cgit v1.2.3