Contents

module Timeasure
  module ClassMethods
    def tracked_instance_methods(*method_names)
      method_names.each do |method_name|
        instance_interceptor = const_get("#{timeasure_name}InstanceInterceptor")
        add_method_to_interceptor(instance_interceptor, method_name)
      end
    end

    def tracked_class_methods(*method_names)
      method_names.each do |method_name|
        class_interceptor = const_get("#{timeasure_name}ClassInterceptor")
        add_method_to_interceptor(class_interceptor, method_name)
      end
    end

    def timeasure_name
      name.gsub('::', '_')
    end

    private

    def add_method_to_interceptor(interceptor, method_name)
      interceptor.class_eval do
        define_method method_name do |*args, &block|
          Timeasure.measure(klass_name: interceptor.klass_name.to_s, method_name: method_name.to_s) do
            super(*args, &block)
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
timeasure-0.1.0	lib/timeasure/class_methods.rb