time blocks

lib/span.rb

@@ -19,6 +19,15 @@ def initialize(tracer, name, options={})
       @end_time = nil
     end
 
+
+    def trace()
+      if block_given?
+        yield(self)
+      end
+      self.finish()
+      return self
+    end
+
     # Mark the span finished at the current time and submit it.
     def finish()
       self.finish_at(Time.now.utc)

test/helper.rb

@@ -7,3 +7,8 @@
 def get_test_tracer()
   return Datadog::Tracer.new
 end
+
+
+class Writer
+
+end

test/test_span.rb

@@ -28,4 +28,16 @@ def test_span_with_parent()
     assert_equal(span.name, 'my.op')
   end
 
+  def test_span_block()
+    start = Time.now.utc
+    span = Datadog::Span.new(nil, 'my.op').trace do |s|
+      assert_equal(s.name, 'my.op')
+      assert_equal(s.end_time, nil)
+    end
+
+    assert span.end_time != nil
+    assert span.end_time < Time.now.utc
+    assert span.end_time > start
+  end
+
 end