Contents

# frozen_string_literal: true

module SafeRequestTimeout
  class ActiveRecordHook
    class << self
      # Add the timeout hook to the connection class.
      #
      # @param connection_class [Class] The class to add the timeout hook to.
      # @return [void]
      def add_timeout!(connection_class = nil)
        connection_class ||= ::ActiveRecord::Base.connection.class

        SafeRequestTimeout::Hooks.add_timeout!(connection_class, [:exec_query])

        SafeRequestTimeout::Hooks.clear_timeout!(connection_class, [:commit_db_transaction])
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
safe_request_timeout-1.0.1	lib/safe_request_timeout/active_record_hook.rb
safe_request_timeout-1.0.0	lib/safe_request_timeout/active_record_hook.rb