active_support_subscriber.rb 310 Bytes
module Gitlab
  module QueryLimiting
    class ActiveSupportSubscriber < ActiveSupport::Subscriber
      attach_to :active_record

      def sql(event)
        unless event.payload.fetch(:cached, event.payload[:name] == 'CACHE')
          Transaction.current&.increment
        end
      end
    end
  end
end