Contents

require 'securerandom'

module Honeycomb
  module Span
  end

  class << self
    def with_trace_id(trace_id = SecureRandom.uuid)
      Thread.current[:honeycomb_trace_id] = trace_id
      yield trace_id
    ensure
      Thread.current[:honeycomb_trace_id] = nil
    end

    def trace_id
      Thread.current[:honeycomb_trace_id]
    end

    def with_span_id(span_id)
      parent_span_id = Thread.current[:honeycomb_span_id]
      Thread.current[:honeycomb_span_id] = span_id
      yield parent_span_id
    ensure
      Thread.current[:honeycomb_span_id] = parent_span_id
    end

    def span(service_name:, name:, span_id: SecureRandom.uuid)
      event = client.event

      event.add_field :traceId, trace_id if trace_id
      event.add_field :serviceName, service_name
      event.add_field :name, name
      event.add_field :id, span_id

      start = Time.now
      with_span_id(span_id) do |parent_span_id|
        event.add_field :parentId, parent_span_id if parent_span_id
        yield
      end
    rescue Exception => e
      if event
        event.add_field :exception_class, e.class.name
        event.add_field :exception_message, e.message
      end
      raise
    ensure
      if start && event
        finish = Time.now
        duration = finish - start
        event.add_field :durationMs, duration * 1000
        event.send
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
honeycomb-beeline-0.0.2	lib/honeycomb/span.rb
honeycomb-beeline-0.0.1	lib/honeycomb/span.rb