ActiveRecord:除了Rails日志中的总时间外,是否可以获得执行的DB查询数量?

时间:2012-11-07 14:17:21

标签: ruby-on-rails-3 performance activerecord

对于每个请求,我都会在日志中找到它:

Completed 200 OK in 854ms (Views: 1.0ms | ActiveRecord: 17.0ms)

是否可以让它也包含查询数量?
类似的东西:

Completed 200 OK in 854ms (Views: 1.0ms | ActiveRecord: 17.0ms | Queries: 10)

理想情况下,我希望所有“缓存”的内容也会出现在那个数字中。即,即使“缓存”从命中数据库中保存我的“N + 1”查询,我仍然想知道我有问题。

我很喜欢monkeypatching /手动编辑的东西,因为我真的想要这只是我的开发盒。

(如果这可以文明化,那么我可以在制作中使用它,那就更好了,但如果没有,我可以在我自己的机器上手动修改Rails)

谢谢!
丹尼尔

1 个答案:

答案 0 :(得分:2)

我知道ThinkingSphinx gem做了类似的事情,将运行Sphinx查询所花费的时间添加到日志中的摘要中。您可以做类似的事情(可能通过制作自己的宝石,因为我打赌其他人会喜欢这个功能)来显示查询数量。 我并没有真正看起来它是如何工作的,但看起来对ActionController和LogSubscriber的修改是负责的:

LIB / thinking_sphinx / action_controller.rb:

module ThinkingSphinx
  module ActionController
    extend ActiveSupport::Concern

    protected

    attr_internal :query_runtime

    def cleanup_view_runtime
      log_subscriber = ThinkingSphinx::ActiveRecord::LogSubscriber
      query_runtime_pre_render = log_subscriber.reset_runtime
      runtime = super
      query_runtime_post_render = log_subscriber.reset_runtime
      self.query_runtime = query_runtime_pre_render + query_runtime_post_render
      runtime - query_runtime_post_render
    end

    def append_info_to_payload(payload)
      super
      payload[:query_runtime] = query_runtime
    end

    module ClassMethods
      def log_process_action(payload)
        messages, query_runtime = super, payload[:query_runtime]
        messages << ("Sphinx: %.1fms" % query_runtime.to_f) if query_runtime
        messages
      end
    end
  end
end

LIB / thinking_sphinx / active_record / log_subscriber.rb:

require 'active_support/log_subscriber'

module ThinkingSphinx
  module ActiveRecord
    class LogSubscriber < ActiveSupport::LogSubscriber
      def self.runtime=(value)
        Thread.current['thinking_sphinx_query_runtime'] = value
      end

      def self.runtime
        Thread.current['thinking_sphinx_query_runtime'] ||= 0
      end

      def self.reset_runtime
        rt, self.runtime = runtime, 0
        rt
      end

      def initialize
        super
        @odd_or_even = false
      end

      def query(event)
        self.class.runtime += event.duration
        return unless logger.debug?

        identifier = color('Sphinx Query (%.1fms)' % event.duration, GREEN, true)
        query = event.payload[:query]
        query = color query, nil, true if odd?

        debug "  #{identifier}  #{query}"
      end

      def message(event)
        return unless logger.debug?

        identifier = color 'Sphinx', GREEN, true
        message = event.payload[:message]
        message = color message, nil, true if odd?

        debug "  #{identifier}  #{message}"
      end

      def odd?
        @odd_or_even = !@odd_or_even
      end

      def logger
        return @logger if defined? @logger
        self.logger = ::ActiveRecord::Base.logger
      end

      def logger=(logger)
        @logger = logger
      end

      attach_to :thinking_sphinx
    end
  end
end

我希望这会有所帮助。