Changeset 2704 for lang/ruby/ssb/trunk/libs/ssb/time_stamper.rb

Timestamp:

12/07/07 08:53:56 (5 years ago)

Author:

coji

Message:

lang/ruby/ssb: refactored TimeStamp?

Files:

• lang/ruby/ssb/trunk/libs/ssb/time_stamper.rb (modified) (2 diffs)

lang/ruby/ssb/trunk/libs/ssb/time_stamper.rb

@@ -8 +8 @@
 module SSB
   class TimeStamper
-    def initialize
+    attr_reader :log
+
+    # { span_label => [stamp_from, stamp_to], ... }
+    def initialize(span_defs = nil)
+      @span_defs = Hash.new
+      @span_defs.update(span_defs) unless span_defs.nil?
       @log = Hash.new
     end
 
-    def stamp(label)
-      @log[label] = Time.now
-      @log[label]
+    def span_labels
+      @span_defs.keys
     end
 
-    def diff(from, to)
-      ((self[to].to_f - self[from].to_f)*1000).to_i
+    def span(label)
+      diff(@span_defs[label][0], @span_defs[label][1])
+    end
+
+    def stamp(stamp, time = Time.now)
+      @log[stamp] = time
+    end
+
+    def diff(stamp_from, stamp_to)
+      ((@log[stamp_to].to_f - @log[stamp_from].to_f)*1000).to_i
     end
 
@@ -28 +40 @@
       @log[label]
     end
+
+    def method_missing(msg, *args)
+      if span_labels.include?(msg.to_sym)
+        span(msg.to_sym)
+      else
+        #super
+      end
+    end
   end
 end