diff options
Diffstat (limited to 'config/initializers/active_record_transaction_observer.rb')
-rw-r--r-- | config/initializers/active_record_transaction_observer.rb | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/config/initializers/active_record_transaction_observer.rb b/config/initializers/active_record_transaction_observer.rb new file mode 100644 index 00000000000..fc9b73d656e --- /dev/null +++ b/config/initializers/active_record_transaction_observer.rb @@ -0,0 +1,18 @@ +# frozen_string_literal: true + +return unless Gitlab.com? || Gitlab.dev_or_test_env? + +def feature_flags_available? + # When the DBMS is not available, an exception (e.g. PG::ConnectionBad) is raised + active_db_connection = ActiveRecord::Base.connection.active? rescue false + + active_db_connection && Feature::FlipperFeature.table_exists? +rescue ActiveRecord::NoDatabaseError + false +end + +Gitlab::Application.configure do + if feature_flags_available? && ::Feature.enabled?(:active_record_transactions_tracking, type: :ops, default_enabled: :yaml) + Gitlab::Database::Transaction::Observer.register! + end +end |