gitlab-foss/lib/gitlab/database/migrations/pg_backend_pid.rb

# frozen_string_literal: true

module Gitlab
  module Database
    module Migrations
      module PgBackendPid
        module MigratorPgBackendPid
          extend ::Gitlab::Utils::Override

          override :with_advisory_lock
          def with_advisory_lock
            Gitlab::Database::Migrations::PgBackendPid.say(connection)

            super
          ensure
            Gitlab::Database::Migrations::PgBackendPid.say(connection)
          end
        end

        module OldMigratorPgBackendPid
          extend ::Gitlab::Utils::Override

          override :with_advisory_lock_connection
          def with_advisory_lock_connection
            super do |conn|
              Gitlab::Database::Migrations::PgBackendPid.say(conn)

              yield(conn)
            ensure
              Gitlab::Database::Migrations::PgBackendPid.say(conn)
            end
          end
        end

        def self.patch!
          if ::Gitlab.next_rails?
            ActiveRecord::Migrator.prepend(MigratorPgBackendPid)
          else
            ActiveRecord::Migrator.prepend(OldMigratorPgBackendPid)
          end
        end

        def self.say(conn)
          return unless ActiveRecord::Migration.verbose

          pg_backend_pid = conn.select_value('SELECT pg_backend_pid()')
          db_name = Gitlab::Database.db_config_name(conn)

          # rubocop:disable Rails/Output
          puts "#{db_name}: == [advisory_lock_connection] " \
               "object_id: #{conn.object_id}, pg_backend_pid: #{pg_backend_pid}"
          # rubocop:enable Rails/Output
        end
      end
    end
  end
end