ygrene_istio_tracing 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8361cbf8ec454a057b43180a718ba352d20a83658b36cd4efd92a24320ca853e
-  data.tar.gz: a052baad266616035ece1adbbb4d2c2fd6c4950c8cf379b47c92e4cbd7afef88
+  metadata.gz: 130a10777cc833fd87deccd676ec488a8b851fd9e875603e8aca682355ecf29a
+  data.tar.gz: ce364a10d73bc0008010cdd6b848e4759e793a5e1fa3423ca39d2b9169cd7a24
 SHA512:
-  metadata.gz: dd6558df91d5c1c8f69ea7e71d9da9f01bd4dc5548f23cca204a15f6b86ab456d49ccf9abdffa90120161d7902c346da8e3e4d5cb9eb0722a2f293c12c182854
-  data.tar.gz: 5b4cc52c8e68f37e5c4e2bf00fe10f3b661df4c70eefaae4da92247d8b7afb68d2a37e947030c457028e305251d4c448c539c15409e64b52b482aa5aa0d182ac
+  metadata.gz: a86ed190b5a3556d02dba560899e723856e950fdb3079ce0662502a3f86e41cd7e23726ad402120bf17b9cfaeafaa02871710325fd12c6b982d2ce5de8a8ed7b
+  data.tar.gz: 34e5a7cc8bffff5b33985f9d426a5cfb1b510448859f35f3abe13a3ca1f5694f1943c45f5c5085ccfa80935036f1f92bdf8d86e3449b71b0c8385624fa8d775a

data/.ruby-gemset

@@ -0,0 +1 @@
+ygrene-istio-tracing

data/.ruby-version

@@ -0,0 +1 @@
+ruby-2.5.1

data/lib/ygrene_istio_tracing.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require 'ygrene_istio_tracing/version'
+require 'ygrene_istio_tracing/tracing'

data/lib/ygrene_istio_tracing/tracing.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'ygrene_istio_tracing/tracing/context'
+require 'ygrene_istio_tracing/tracing/header'
+require 'ygrene_istio_tracing/tracing/http'
+require 'ygrene_istio_tracing/tracing/rack'
+
+begin
+  require 'rails/railtie'
+  require 'ygrene_istio_tracing/tracing/railtie'
+rescue LoadError
+end

data/lib/ygrene_istio_tracing/tracing/context.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module YgreneIstioTracing
+  module Tracing
+    class NotSetError < ::StandardError
+      def initialize
+        super('Context is not set for this thread')
+      end
+    end
+
+    class Context
+      VAR_NAME = :_istio_tracing_context_
+
+      class << self
+        def current
+          Thread.current.thread_variable_get(VAR_NAME) || raise(NotSetError)
+        end
+
+        def with_given_context(ctx)
+          Thread.current.thread_variable_set(VAR_NAME, ctx)
+          yield
+        ensure
+          remove_current
+        end
+
+        def exists_current?
+          !!Thread.current.thread_variable_get(VAR_NAME)
+        end
+
+        def build_current(headers)
+          Thread.current.thread_variable_set(VAR_NAME, Context.new(headers))
+        end
+
+        def remove_current
+          Thread.current.thread_variable_set(VAR_NAME, nil)
+        end
+      end
+
+      attr_reader :headers
+
+      def initialize(headers)
+        @headers = headers
+      end
+    end
+  end
+end

data/lib/ygrene_istio_tracing/tracing/header.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module YgreneIstioTracing
+  module Tracing
+    PROPAGATION_HEADERS = %w[
+      x-request-id
+      x-b3-traceid
+      x-b3-spanid
+      x-b3-parentspanid
+      x-b3-sampled
+      x-b3-flags
+      x-ot-span-context
+      X-REQUEST-ID
+      X-B3-TRACEID
+      X-B3-SPANID
+      X-B3-PARENTSPANID
+      X-B3-SAMPLED
+      X-B3-FLAGS
+      X-OT-SPAN-CONTEXT
+    ].freeze
+  end
+end

data/lib/ygrene_istio_tracing/tracing/http.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'net/http'
+require 'typhoeus'
+
+module YgreneIstioTracing
+  module NetHttp
+    module ClassInterceptor
+      def new(*options)
+        o = super(*options)
+        o
+      end
+    end
+    module InstanceInterceptor
+      def initialize(*options)
+        super(*options)
+      end
+
+      def request(req, *args, &block)
+        return super unless Context.exists_current?
+
+        headers = YgreneIstioTracing::Tracing::Context.current.headers
+        YgreneIstioTracing::Tracing::PROPAGATION_HEADERS.each do |h|
+          req[h] = headers[h] if headers[h].present?
+        end
+        super
+      end
+    end
+  end
+end
+
+::Net::HTTP.singleton_class.prepend YgreneIstioTracing::NetHttp::ClassInterceptor
+::Net::HTTP.prepend YgreneIstioTracing::NetHttp::InstanceInterceptor
+
+module Typhoeus
+  Typhoeus.before do |req|
+    return req unless YgreneIstioTracing::Tracing::Context.exists_current?
+
+    headers = YgreneIstioTracing::Tracing::Context.current.headers
+    YgreneIstioTracing::Tracing::PROPAGATION_HEADERS.each do |h|
+      req.options[:headers][h] = headers[h] unless headers[h]&.nil?
+    end
+    req
+  end
+end

data/lib/ygrene_istio_tracing/tracing/rack.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module YgreneIstioTracing
+  module Tracing
+    class RackMiddleware
+
+      def initialize(app)
+        @app = app
+      end
+
+      def call(env)
+        Context.build_current(extract_http_headers(env))
+        @app.call(env)
+      ensure
+        Context.remove_current
+      end
+
+      def extract_http_headers(env)
+        env.select { |k, _v| PROPAGATION_HEADERS.include?(k.downcase) }
+      end
+    end
+  end
+end

data/lib/ygrene_istio_tracing/tracing/railtie.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require 'rails/railtie'
+
+module YgreneIstioTracing
+  module Tracing
+    class Railtie < ::Rails::Railtie
+      initializer 'ygrene_istio_tracing.tracing.rack_middleware' do |app|
+        puts 'Tracing Loaded'
+        app.middleware.use(::Istio::Tracing::RackMiddleware)
+      end
+    end
+  end
+end

data/lib/ygrene_istio_tracing/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module YgreneIstioTracing
+  VERSION = '1.0.3'
+end

data/ygrene_istio_tracing.gemspec

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+lib = File.expand_path('lib', __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ygrene_istio_tracing/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'ygrene_istio_tracing'
+  spec.version       = YgreneIstioTracing::VERSION
+  spec.authors       = ['Austin Adams']
+  spec.email         = ['me@austbot.com']
+
+  spec.summary       = 'Ruby gem for Istio service mesh (https://istio.io/)'
+  spec.description   = 'This gem lets your rails app propagate http headers for distributed tracing with Istio.'
+  spec.homepage      = 'https://github.com/ygrene/ygrene_istio_tracing'
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  if spec.respond_to?(:metadata)
+    spec.metadata['allowed_push_host'] = "https://rubygems.org"
+  else
+    raise 'RubyGems 2.0 or newer is required to protect against ' \
+      'public gem pushes.'
+  end
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'typhoeus', '~> 1.3.1'
+
+  spec.add_development_dependency 'bundler', '~> 1.16'
+  spec.add_development_dependency 'rack'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rspec-rails'
+
+  spec.add_development_dependency 'webmock', ['~>3.3.0']
+  spec.add_development_dependency 'httparty', ['~>0.16.3']
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ygrene_istio_tracing
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - Austin Adams
@@ -118,6 +118,8 @@ extra_rdoc_files: []
 files:
 - ".gitignore"
 - ".rspec"
+- ".ruby-gemset"
+- ".ruby-version"
 - Gemfile
 - Gemfile.lock
 - LICENSE
@@ -125,6 +127,15 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
+- lib/ygrene_istio_tracing.rb
+- lib/ygrene_istio_tracing/tracing.rb
+- lib/ygrene_istio_tracing/tracing/context.rb
+- lib/ygrene_istio_tracing/tracing/header.rb
+- lib/ygrene_istio_tracing/tracing/http.rb
+- lib/ygrene_istio_tracing/tracing/rack.rb
+- lib/ygrene_istio_tracing/tracing/railtie.rb
+- lib/ygrene_istio_tracing/version.rb
+- ygrene_istio_tracing.gemspec
 homepage: https://github.com/ygrene/ygrene_istio_tracing
 licenses: []
 metadata: