lib/libhoney/log_transmission.rb in libhoney-2.3.0 vs lib/libhoney/log_transmission.rb in libhoney-2.4.0

@@ -17,17 +17,20 @@
       @verbose = verbose
     end
 
     # Prints an event
     def add(event)
+      Thread.current[:libhoney_transmitting] = true
       if @verbose
         metadata = "Honeycomb dataset '#{event.dataset}' | #{event.timestamp.iso8601}"
         metadata << " (sample rate: #{event.sample_rate})" if event.sample_rate != 1
         @output.print("#{metadata} | ")
       end
       clean_data(event.data).tap do |data|
         @output.puts(data.to_json)
       end
+    ensure
+      Thread.current[:libhoney_transmitting] = false
     end
 
     # Flushes the output (but does not close it)
     def close(_drain)
       @output.flush