Contents

# frozen_string_literal: true

module Minato
  module Trace
    module Middleware
      class DistributedTraceContext
        TRACE_HEADER_KEY = "X-Cloud-Trace-Context"

        def initialize(app)
          @app = app
        end

        def call(env)
          add_trace_context_header(env)

          @app.call(env)
        end

        private

        def add_trace_context_header(env)
          return unless ::Rails.env.production? && Minato::Trace.enabled?

          return if current_cloud_trace.nil?

          env.request_headers.merge!({ "#{TRACE_HEADER_KEY}": "#{current_cloud_trace.trace.trace_id};o=1" })
        end

        def current_cloud_trace
          @current_cloud_trace ||= current_trace_from_google_cloud
        end

        def current_trace_from_google_cloud
          _trace_client = Google::Cloud::Trace.new
          Google::Cloud::Trace.get
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 2 rubygems

Version	Path
minato_ruby_api_client-0.2.2	vendor/bundle/ruby/3.2.0/gems/minato-trace-0.2.0/lib/minato/trace/middleware/distributed_trace_context.rb
minato-trace-0.2.0	lib/minato/trace/middleware/distributed_trace_context.rb
minato-trace-0.1.7	lib/minato/trace/middleware/distributed_trace_context.rb