debian-mirror-gitlab/lib/gitlab/query_limiting/active_support_subscriber.rb
2021-04-17 20:07:23 +05:30

16 lines
416 B
Ruby

# frozen_string_literal: true
module Gitlab
module QueryLimiting
class ActiveSupportSubscriber < ActiveSupport::Subscriber
attach_to :active_record
def sql(event)
return if !Transaction.current || event.payload.fetch(:cached, event.payload[:name] == 'CACHE')
Transaction.current.increment
Transaction.current.executed_sql(event.payload[:sql])
end
end
end
end