Contents

require "benchmeth/version"
require "benchmark"

# :( Global
$benchmeth_main = self

module Benchmeth
  DEFAULT_BLOCK = lambda do |method, realtime|
    puts "%s : %d ms" % [method, realtime * 1000]
  end

  def self.included(base)
    base.send :extend, ClassMethods
    base.send :include, InstanceMethods
  end

  def self.on_benchmark(&block)
    if block_given?
      @on_benchmark_block = block
    else
      @on_benchmark_block || DEFAULT_BLOCK
    end
  end

  module ClassMethods

    def benchmark(*method_names)
      method_names.each do |method_name|
        method_name = method_name.to_sym
        self.send :alias_method, :"#{method_name}_without_benchmark", method_name
        self.send :define_method, method_name do |*args|
          result = nil
          method_prefix = nil
          realtime = Benchmark.realtime do
            result = self.send(:"#{method_name}_without_benchmark", *args)
            method_prefix =
            case self
            when $benchmeth_main
              ""
            when Class
              "#{self.name}."
            else
              "#{self.class.name}#"
            end
          end
          Benchmeth.on_benchmark.call("#{method_prefix}#{method_name}", realtime)
          result
        end
      end
    end

  end

  module InstanceMethods

    def benchmark(*method_names, &block)
      self.class.benchmark(*method_names, &block)
    end

  end

end

Object.send :include, Benchmeth

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
benchmeth-0.1.0	lib/benchmeth.rb