config/initializers/database_query_analyzers.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

# frozen_string_literal: true

# Currently we register validator only for `dev` or `test` environment
if Gitlab.dev_or_test_env? || Gitlab::Utils.to_boolean(ENV['GITLAB_ENABLE_QUERY_ANALYZERS'], default: false)
  Gitlab::Database::QueryAnalyzer.instance.hook!
  Gitlab::Database::QueryAnalyzer.instance.all_analyzers.append(::Gitlab::Database::QueryAnalyzers::GitlabSchemasMetrics)

  if Rails.env.test? || Gitlab::Utils.to_boolean(ENV['ENABLE_CROSS_DATABASE_MODIFICATION_DETECTION'], default: false)
    Gitlab::Database::QueryAnalyzer.instance.all_analyzers.append(::Gitlab::Database::QueryAnalyzers::PreventCrossDatabaseModification)
  end

  Gitlab::Application.configure do |config|
    config.middleware.use(Gitlab::Middleware::QueryAnalyzer)
  end
end