diff options
Diffstat (limited to 'config/initializers/database_query_analyzers.rb')
| -rw-r--r-- | config/initializers/database_query_analyzers.rb | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/config/initializers/database_query_analyzers.rb b/config/initializers/database_query_analyzers.rb index 2e73fbb79a2..ad6ed20b94d 100644 --- a/config/initializers/database_query_analyzers.rb +++ b/config/initializers/database_query_analyzers.rb @@ -1,15 +1,19 @@ # frozen_string_literal: true # Currently we register validator only for `dev` or `test` environment -Gitlab::Database::QueryAnalyzer.instance.hook! -Gitlab::Database::QueryAnalyzer.instance.all_analyzers.append(::Gitlab::Database::QueryAnalyzers::GitlabSchemasMetrics) -Gitlab::Database::QueryAnalyzer.instance.all_analyzers.append( - ::Gitlab::Database::QueryAnalyzers::PreventCrossDatabaseModification -) +Gitlab::Database::QueryAnalyzer.instance.tap do |query_analyzer| + query_analyzer.hook! -if Gitlab.dev_or_test_env? - query_analyzer = ::Gitlab::Database::QueryAnalyzers::GitlabSchemasValidateConnection - Gitlab::Database::QueryAnalyzer.instance.all_analyzers.append(query_analyzer) + query_analyzer.all_analyzers.tap do |analyzers| + analyzers.append(::Gitlab::Database::QueryAnalyzers::GitlabSchemasMetrics) + analyzers.append(::Gitlab::Database::QueryAnalyzers::PreventCrossDatabaseModification) + analyzers.append(::Gitlab::Database::QueryAnalyzers::Ci::PartitioningRoutingAnalyzer) + + if Gitlab.dev_or_test_env? + analyzers.append(::Gitlab::Database::QueryAnalyzers::GitlabSchemasValidateConnection) + analyzers.append(::Gitlab::Database::QueryAnalyzers::QueryRecorder) + end + end end Gitlab::Application.configure do |config| |