Contents

# frozen_string_literal: true

require 'http/tracer/version'
require 'opentracing'

module HTTP
  module Tracer
    class << self
      attr_accessor :ignore_request, :tracer

      IngoreRequest = ->(_verb, _uri, _opts) { false }

      def instrument(tracer: OpenTracing.global_tracer, ignore_request: IngoreRequest)
        @ignore_request = ignore_request
        @tracer = tracer
        patch_request
      end

      def remove
        return unless ::HTTP::Client.method_defined?(:request_original)

        ::HTTP::Client.class_eval do
          remove_method :request
          alias_method :request, :request_original
          remove_method :request_original
        end
      end

      def patch_request
        ::HTTP::Client.class_eval do
          alias_method :request_original, :request

          def request(verb, uri, opts = {})
            options = HTTP::Options.new.merge(opts)
            parsed_uri = uri.is_a?(String) ? URI(uri) : uri

            if ::HTTP::Tracer.ignore_request.call(verb, uri, options)
              res = request_original(verb, uri, options)
            else
              tags = {
                'component' => 'ruby-httprb',
                'span.kind' => 'client',
                'http.method' => verb,
                'http.url' => parsed_uri.path,
                'peer.host' => parsed_uri.host,
                'peer.port' => parsed_uri.port
              }

              tracer = ::HTTP::Tracer.tracer

              tracer.start_active_span('http.request', tags: tags) do |scope|
                OpenTracing.inject(scope.span.context, OpenTracing::FORMAT_RACK, options.headers)

                res = request_original(verb, uri, options)

                scope.span.set_tag('http.status_code', res.status)
                scope.span.set_tag('error', true) if res.is_a?(StandardError)
              end
            end

            res
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
httprb-opentracing-0.1.0	lib/http/tracer.rb