opencensus-jaeger 0.2.1

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "opencensus/jaeger"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/examples/.byebug_history

@@ -0,0 +1,97 @@
+c
+continue
+next
+step
+next
+a.running?
+a=OpenCensus.config.trace.exporter.instance_variable_get('@executor')
+a.running?
+next
+a.running?
+a=OpenCensus.config.trace.exporter.instance_variable_get('@executor')
+a = _
+OpenCensus.config.trace.exporter.instance_variable_get('@executor')
+OpenCensus.config.trace.exporter.instance_variable_get('executor')
+OpenCensus.config.trace.exporter.instance_variable_get(:executor)
+OpenCensus.config.trace.exporter.running?
+OpenCensus.config.trace.exporter
+OpenCensus.config.trace.methods
+OpenCensus.config.trace
+OpenCensus.config.trcae
+OpenCensus.config.methods
+OpenCensus.config
+next
+value
+field.value
+next
+field
+field.exporter
+field
+key
+next
+key
+@fields
+field
+n
+'n
+n
+next
+field
+key
+next
+key
+field.exporter.methods
+field.exporter
+field.methods
+field
+field.is_a? Config
+field.value
+@fields
+field
+fiel
+next
+backtrace
+step
+puts "#{c}"
+step
+next
+step
+c
+step
+next
+step
+continue
+require 'jaeger/client'
+continue
+Jaeger.constants
+Jaeger::Thrift
+$LOAD_PATH
+require_relative 'jaeger/client/../../thrift/gen-rb'
+require 'jaeger/client/../../thrift/gen-rb'
+require 'jaeger/client'
+require 'jaeger/client/../../thrift/gen-rb'
+require_relative 'jaeger/client/../../thrift/gen-rb'
+require 'jaeger/client/../../thrift/gen-rb'
+require 'jaeger/client'
+require_relative 'jaeger/client/../../thrift/gen-rb'
+__FILE__
+File.dirname(__FILE__)
+require 'jaeger'
+$LOAD_PATH.push(File.dirname(__FILE__)+ '/../../thrift/gen-rb')
+'
+require 'jaeger/thrift/gen-rb/'
+require 'jaeger/thrift/agent'
+$LOAD_PATH.push(File.dirname(__FILE__)+ '/../../thrift/gen-rb')
+require 'opentracing'
+require 'jaeger/thrift/agent'
+Jaeger::Thrift
+Jaeger::Span.constants
+Jaeger::Span.constant
+Jaeger::Span.class
+Jaeger::Span
+Jaeger.constants
+Jaeger.class
+Jaeger.type
+Jaeger.tyoe
+Jaeger.methods
+Jaeger

data/examples/Gemfile

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+gem 'opencensus'
+gem 'sinatra'
+gem 'thin'
+gem 'faraday'
+gem 'opencensus-jaeger', path: "../"

data/examples/Gemfile.lock

@@ -0,0 +1,46 @@
+PATH
+  remote: ..
+  specs:
+    opencensus-jaeger (0.2.0)
+      opencensus (~> 0.4.0)
+      thrift (~> 0.11.0)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    daemons (1.3.1)
+    eventmachine (1.2.7)
+    faraday (0.15.4)
+      multipart-post (>= 1.2, < 3)
+    multipart-post (2.0.0)
+    mustermann (1.0.3)
+    opencensus (0.4.0)
+    rack (2.0.6)
+    rack-protection (2.0.5)
+      rack
+    sinatra (2.0.5)
+      mustermann (~> 1.0)
+      rack (~> 2.0)
+      rack-protection (= 2.0.5)
+      tilt (~> 2.0)
+    thin (1.7.2)
+      daemons (~> 1.0, >= 1.0.9)
+      eventmachine (~> 1.0, >= 1.0.4)
+      rack (>= 1, < 3)
+    thrift (0.11.0.0)
+    tilt (2.0.9)
+
+PLATFORMS
+  ruby
+  x86_64-darwin-17
+  x86_64-darwin-18
+
+DEPENDENCIES
+  faraday
+  opencensus
+  opencensus-jaeger!
+  sinatra
+  thin
+
+BUNDLED WITH
+   1.17.3

data/examples/config.ru

@@ -0,0 +1,15 @@
+require 'opencensus/jaeger'
+require 'opencensus/trace/integrations/rack_middleware'
+require 'opencensus/trace/integrations/faraday_middleware'
+require_relative './example.rb'
+
+OpenCensus.configure do |c|
+  c.trace.exporter = OpenCensus::Trace::Exporters::JaegerExporter.new \
+    service_name: 'test_service',
+    host: 'localhost',
+    port: 6831,
+    tags: { 'example_tag': 'hello_world' }
+end
+
+use OpenCensus::Trace::Integrations::RackMiddleware
+run Example

data/examples/example.rb

@@ -0,0 +1,25 @@
+require 'sinatra/base'
+
+class Example < Sinatra::Base
+  get '/' do
+    'Hello world'
+  end
+
+  get "/lengthy" do
+    # Configure this Faraday connection with a middleware to trace outgoing
+    # requests.
+    conn = Faraday.new(url: "http://www.google.com") do |c|
+      c.use OpenCensus::Trace::Integrations::FaradayMiddleware
+      c.adapter Faraday.default_adapter
+    end
+    conn.get "/"
+
+    # You may instrument your code to create custom spans for long-running
+    # operations.
+    OpenCensus::Trace.in_span "long task" do
+      sleep rand
+    end
+
+    "Done!"
+  end
+end

data/lib/opencensus-jaeger.rb

@@ -0,0 +1,4 @@
+module OpenCensus
+end
+
+require 'opencensus/jaeger'

data/lib/opencensus/jaeger.rb

@@ -0,0 +1,12 @@
+require 'opencensus'
+
+module OpenCensus
+end
+
+thrift_folder = File.expand_path(__dir__ + '../../../thrift/gen-rb')
+$LOAD_PATH.unshift(thrift_folder) unless $LOAD_PATH.include?(thrift_folder)
+
+require 'jaeger/thrift/agent'
+require 'opencensus/jaeger/version'
+require 'opencensus/logging'
+require 'opencensus/trace/exporters/jaeger_exporter'

data/lib/opencensus/jaeger/version.rb

@@ -0,0 +1,5 @@
+module OpenCensus
+  module Jaeger
+    VERSION = '0.2.1'.freeze
+  end
+end

data/lib/opencensus/logging.rb

@@ -0,0 +1,24 @@
+require 'logger'
+
+module Logging
+  def default_logger
+    Logging.logger
+  end
+
+  def self.logger
+    @logger ||= Logger.new(STDOUT)
+    @logger.level = ENV['LOG_LEVEL'].to_i || Logger::INFO
+    at_exit do
+      @logger.close
+    end
+    @logger
+  end
+
+  def self.included(base)
+    class << base
+      def logger
+        Logging.logger
+      end
+    end
+  end
+end

data/lib/opencensus/trace/exporters/jaeger_driver.rb

@@ -0,0 +1,125 @@
+require 'socket'
+require 'thrift'
+require 'opencensus/trace/exporters/jaeger_driver/intermediate_transport'
+require 'opencensus/trace/exporters/jaeger_driver/udp_sender'
+
+module OpenCensus
+  module Trace
+    module Exporters
+      module JaegerDriver
+        extend self
+
+        MAX_64BIT_SIGNED_INT = (1 << 63) - 1
+        MAX_64BIT_UNSIGNED_INT = (1 << 64) - 1
+        TRACE_ID_UPPER_BOUND = MAX_64BIT_UNSIGNED_INT + 1
+
+        FIELDS = ::Jaeger::Thrift::Tag::FIELDS
+        KEY = FIELDS[::Jaeger::Thrift::Tag::KEY].fetch(:name)
+        VTYPE = FIELDS[::Jaeger::Thrift::Tag::VTYPE].fetch(:name)
+        VLONG = FIELDS[::Jaeger::Thrift::Tag::VLONG].fetch(:name)
+        VDOUBLE = FIELDS[::Jaeger::Thrift::Tag::VDOUBLE].fetch(:name)
+        VBOOL = FIELDS[::Jaeger::Thrift::Tag::VBOOL].fetch(:name)
+        VSTR = FIELDS[::Jaeger::Thrift::Tag::VSTR].fetch(:name)
+
+        def encode_span(span)
+          tags = build_thrift_tags(span.attributes)
+          trace_id_high = base16_hex_to_int64(
+            span.trace_id.slice(0, 16)
+          )
+          trace_id_low = base16_hex_to_int64(
+            span.trace_id.slice(16)
+          )
+          span_id = base16_hex_to_int64(
+            span.span_id
+          )
+          parent_span_id = base16_hex_to_int64(
+            span.parent_span_id
+          )
+          operation_name = span.name.value
+          references = []
+          flags = 0x01
+          start_time = span.start_time.to_f * 1_000_000
+          end_time = span.end_time.to_f * 1_000_000
+          duration = end_time - start_time
+
+          ::Jaeger::Thrift::Span.new(
+            'traceIdLow': trace_id_low,
+            'traceIdHigh': trace_id_high,
+            'spanId': span_id,
+            'parentSpanId': parent_span_id,
+            'operationName': operation_name,
+            'references': references,
+            'flags': flags,
+            'startTime': start_time,
+            'duration': duration,
+            'tags': tags
+          )
+        rescue StandardError => e
+          puts "convert error #{e}"
+        end
+
+        def ip_v4
+          ip_v4 = Socket.ip_address_list.find { |ai| ai.ipv4? && !ai.ipv4_loopback? }
+          ip_v4.nil? ? nil : ip_v4.ip_address
+        end
+
+        def base16_hex_id_to_uint64(id)
+          return nil unless id
+          value = id.to_i(16)
+          value > MAX_64BIT_UNSIGNED_INT || value < 0 ? 0 : value
+        end
+
+        # Thrift defines ID fields as i64, which is signed, therefore we convert
+        # large IDs (> 2^63) to negative longs
+        def uint64_id_to_int64(id)
+          id > MAX_64BIT_SIGNED_INT ? id - MAX_64BIT_UNSIGNED_INT - 1 : id
+        end
+
+        def base16_hex_to_int64(id)
+          uint64_id = base16_hex_id_to_uint64(id)
+          uint64_id_to_int64(uint64_id)
+        end
+
+        def build_thrift_tags(attributes)
+          attributes.collect do |key, value|
+            begin
+              if value.is_a?(OpenCensus::Trace::TruncatableString)
+                ::Jaeger::Thrift::Tag.new(
+                  KEY => key.to_s,
+                  VTYPE => ::Jaeger::Thrift::TagType::STRING,
+                  VSTR => value.value
+                )
+              elsif value.is_a?(Integer)
+                ::Jaeger::Thrift::Tag.new(
+                  KEY => key.to_s,
+                  VTYPE => ::Jaeger::Thrift::TagType::LONG,
+                  VLONG => value
+                )
+              elsif value.is_a?(Float)
+                ::Jaeger::Thrift::Tag.new(
+                  KEY => key.to_s,
+                  VTYPE => ::Jaeger::Thrift::TagType::DOUBLE,
+                  VDOUBLE => value
+                )
+              elsif value.is_a?(TrueClass) || value.is_a?(FalseClass)
+                ::Jaeger::Thrift::Tag.new(
+                  KEY => key.to_s,
+                  VTYPE => ::Jaeger::Thrift::TagType::BOOL,
+                  VBOOL => value
+                )
+              else
+                ::Jaeger::Thrift::Tag.new(
+                  KEY => key.to_s,
+                  VTYPE => ::Jaeger::Thrift::TagType::STRING,
+                  VSTR => value.to_s
+                )
+              end
+            rescue StandardError => error
+              logger.error "Cannot build thrift tags for #{key}:#{value}, error: #{error}"
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/opencensus/trace/exporters/jaeger_driver/intermediate_transport.rb

@@ -0,0 +1,25 @@
+module OpenCensus
+  module Trace
+    module Exporters
+      module JaegerDriver
+        class IntermediateTransport
+          attr_accessor :size
+
+          def initialize
+            @size = 0
+          end
+
+          def write(buf)
+            @size += buf.size
+          end
+
+          def flush
+            @size = 0
+          end
+
+          def close; end
+        end
+      end
+    end
+  end
+end

data/lib/opencensus/trace/exporters/jaeger_driver/udp_sender.rb

@@ -0,0 +1,28 @@
+require 'opencensus/trace/exporters/jaeger_driver/udp_transport'
+
+module OpenCensus
+  module Trace
+    module Exporters
+      module JaegerDriver
+        class UDPSender
+          include ::Logging
+
+          def initialize(host, port, logger, protocol_class)
+            @logger = logger || default_logger
+            @host = host
+            @port = port
+            @transport = UDPTransport.new(host, port, logger)
+            @protocol = protocol_class.new(@transport)
+            @client = ::Jaeger::Thrift::Agent::Client.new(@protocol)
+          end
+
+          def send_spans(spans)
+            @client.emitBatch(spans)
+          rescue StandardError => error
+            @logger.error "Failure while sending a batch of spans: #{error}"
+          end
+        end
+      end
+    end
+  end
+end