# frozen_string_literal: true

module Sentry
  class BackgroundWorker
    module ActiveRecordConnectionPatch
      def _perform(&block)
        super(&block)
      ensure
        # some applications have partial or even no AR connection
        if ActiveRecord::Base.connected?
          # make sure the background worker returns AR connection if it accidentally acquire one during serialization
          ActiveRecord::Base.connection_pool.release_connection
        end
      end
    end
  end
end

Sentry::BackgroundWorker.prepend(Sentry::BackgroundWorker::ActiveRecordConnectionPatch)