lib/trifle/logger.rb in trifle-logger-0.0.1 vs lib/trifle/logger.rb in trifle-logger-0.0.2

@@ -30,10 +30,12 @@
     def self.tracer
       Thread.current[:trifle_tracer]
     end
 
     def self.trace(*args, **keywords, &block)
-      return unless tracer
+      if tracer.nil?
+        return block_given? ? yield : nil
+      end
 
       tracer.trace(*args, **keywords, &block)
     end
 
     def self.tag(tag)