aboutsummaryrefslogtreecommitdiffstats
path: root/activesupport/lib/active_support/benchmarkable.rb
blob: 1b391697e60ff863c63916fb388f324a8668070b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
require 'active_support/core_ext/benchmark'
require 'active_support/core_ext/hash/keys'

module ActiveSupport
  module Benchmarkable
    # Allows you to measure the execution time of a block in a template and records the result to
    # the log. Wrap this block around expensive operations or possible bottlenecks to get a time
    # reading for the operation. For example, let's say you thought your file processing method
    # was taking too long; you could wrap it in a benchmark block.
    #
    #  <% benchmark "Process data files" do %>
    #    <%= expensive_files_operation %>
    #  <% end %>
    #
    # That would add something like "Process data files (345.2ms)" to the log, which you can then
    # use to compare timings when optimizing your code.
    #
    # You may give an optional logger level (:debug, :info, :warn, :error) as the :level option.
    # The default logger level value is :info.
    #
    #  <% benchmark "Low-level files", :level => :debug do %>
    #    <%= lowlevel_files_operation %>
    #  <% end %>
    #
    # Finally, you can pass true as the third argument to silence all log activity (other than the
    # timing information) from inside the block. This is great for boiling down a noisy block to 
    # just a single statement that produces one log line:
    #
    #  <% benchmark "Process data files", :level => :info, :silence => true do %>
    #    <%= expensive_and_chatty_files_operation %>
    #  <% end %>
    def benchmark(message = "Benchmarking", options = {})
      if logger
        options.assert_valid_keys(:level, :silence)
        options[:level] ||= :info

        result = nil
        ms = Benchmark.ms { result = options[:silence] ? silence { yield } : yield }
        logger.send(options[:level], '%s (%.1fms)' % [ message, ms ])
        result
      else
        yield
      end
    end

    # Silence the logger during the execution of the block.
    #
    def silence
      ActiveSupport::Deprecation.warn "ActiveSupport::Benchmarkable#silence is deprecated. It will be removed from Rails 4.1."
      old_logger_level, logger.level = logger.level, ::Logger::ERROR if logger
      yield
    ensure
      logger.level = old_logger_level if logger
    end
  end
end