profiler.rb in speed_gun-1.0.0.rc1

- old
+ new
@@ -1,128 +1,21 @@
 require 'speed_gun'
-require 'speed_gun/store'
-require 'speed_gun/browser'
-require 'speed_gun/hook'
-require 'securerandom'
-require 'msgpack'
-require 'multi_json'
 
+# @abstract
 class SpeedGun::Profiler
-  PROFILERS = {}
-
-  def self.load(id)
-    src = SpeedGun.store[id]
-
-    return nil unless src
-
-    restore(src)
+  def self.profile(*args, &block)
+    new.profile(*args, &block)
   end
 
-  def self.restore(src)
-    data = src.kind_of?(String) ? MessagePack.unpack(src) : src
+  def profile(name = self.class.name, payload = {}, &block)
+    starts_at = Time.now
 
-    profiler = new({})
-    profiler.restore_by_hash(data)
+    ret = yield
 
-    profiler
-  end
+    event = SpeedGun::Event.new(
+      name, payload, starts_at, Time.now
+    )
+    SpeedGun.current_profile.record!(event)
 
-  def initialize(env)
-    @id = SecureRandom.uuid
-    @path = env['PATH_INFO']
-    @query = env['QUERY_STRING']
-    @env = env
-    @requested_at = Time.now
-    @profiles = []
-    @browser = nil
-    @active = true
-    @now_profile = nil
+    return ret
   end
-  attr_reader :id, :path, :query, :env, :requested_at, :profiles, :browser
-  attr_accessor :now_profile
-
-  def profile(type, *args, &block)
-    profiler = PROFILERS[type]
-
-    if profiler
-      profiler.profile(self, *args, &block)
-    else
-      yield
-    end
-  end
-
-  def skip?
-    SpeedGun.config.skip_paths.any? { |prefix| prefix.match(@path) }
-  end
-
-  def active?
-    @active && !skip?
-  end
-
-  def activate!
-    @active = true
-  end
-
-  def deactivate!
-    @active = false
-  end
-
-  def dump
-    SpeedGun.store[id] = to_msgpack
-
-    SpeedGun::Hook.invoke_all(self)
-  end
-
-  def browser=(hash)
-    @browser = SpeedGun::Browser.new(hash)
-  end
-
-  def as_msgpack(*args)
-    {
-      id: @id,
-      path: @path,
-      query: @query,
-      env: msgpackable_env,
-      requested_at: @requested_at.to_i,
-      profiles: @profiles.map { |profile| profile.as_msgpack(*args) },
-      browser: @browser ? @browser.as_msgpack(*args) : nil,
-    }
-  end
-
-  def to_msgpack(*args)
-    as_msgpack(*args).to_msgpack(*args)
-  end
-
-  def to_json(*args)
-    MultiJson.dump(as_msgpack(*args))
-  end
-
-  def restore_by_hash(hash)
-    hash.each_pair do |key, val|
-      instance_variable_set(:"@#{key}", restore_attribute(key, val))
-    end
-  end
-
-  def restore_attribute(key, val)
-    case key
-    when 'requested_at'
-      Time.at(val)
-    when 'profiles'
-      val.map { |profile| SpeedGun::Profiler::Base.load(profile) }
-    when 'browser'
-      val ? SpeedGun::Browser.new(val) : val
-    else
-      val
-    end
-  end
-
-  private
-
-  def msgpackable_env
-    env = {}
-    @env.each_pair { |key, val| env[key] = val if key[0] =~ /[A-Z]/ }
-    env
-  end
 end
-
-require 'speed_gun/profiler/manual'
-require 'speed_gun/profiler/js'