helios-opentelemetry-sdk 0.1.20 → 0.1.22

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a91efee8919de8c1d6ba5535e9bd3e347f3402545739261b272e33e1a64e036e
-  data.tar.gz: 1d0c7e40082bcb8153aacd08b36ed919589d32836a17cd65ba16ef68db13391c
+  metadata.gz: 5f067b2b75fcf692e420c9ac84ac7107e09d736174a855d7fce9ca6fb4198990
+  data.tar.gz: 137c32fdde515fbc4a00b191672a6a9188e3434820b8042a287aa715c7b15ad3
 SHA512:
-  metadata.gz: 62f535badbe0f1c481f15dd17e110209233c99025b8d544988e264e78ff23d5b70e12bc10579748f33a0cc23fcf0716b8b4bcade9e2ab3509a1150b1795ef88d
-  data.tar.gz: a935beb8953928cbd407dc71ddad09a6f1134f8d39c5d385e40eafdbf273f1976657f72a5fc114c64595a6ad53b52a2899f8878c8117c699458be3f8867e2568
+  metadata.gz: 797b434ff07072ae9c0094f71f889692ff56c6aba9d5c6668bd0ad1c151503cfaf25344f6c6115fe9b9023a3036b4fb5c0299aae83c391405bd7084d1847b1f0
+  data.tar.gz: e2939e688efce8f83e971a7dc34752fb5559b7202f0c99db257c13b533f66f99ee8ac150f8f4e064c5daa1de7c8440df3add138816d8769284cd1b8ca1493a2b

data/.github/workflows/test.yml

@@ -20,7 +20,7 @@ jobs:
         strategy:
             fail-fast: false
             matrix:
-                ruby-version: ['2.6', '2.7', '3.1']
+                ruby-version: ['2.7', '3.1']
         services:
             mongodb:
                 image: mongo@sha256:845fd775d7a58f564fae7c17a4eec7ff038e1c4004ebe05cc1cb5fc3767cf6cc
@@ -33,14 +33,15 @@ jobs:
                     --health-retries=3
         steps:
             - uses: actions/checkout@v2
-            - uses: actions/setup-ruby@v1
+            - uses: ruby/setup-ruby@v1
               with:
                   ruby-version: ${{ matrix.ruby-version }}
+                  bundler-cache: true
             - name: Install bundler
               run: gem install bundler -v 2.3.16
             - name: bundle install
               run: bundle install
             - name: Rubocop
-              run: rubocop lib spec
+              run: bundle exec rubocop lib spec
             - name: Run specs
-              run: rspec spec
+              run: bundle exec rspec spec

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    helios-opentelemetry-sdk (0.1.20)
+    helios-opentelemetry-sdk (0.1.22)
       opentelemetry-exporter-otlp (~> 0.22.0)
       opentelemetry-instrumentation-all (~> 0.25.0)
       opentelemetry-instrumentation-rspec (~> 0.2.0)
@@ -17,7 +17,7 @@ GEM
       faraday-net_http (~> 2.0)
       ruby2_keywords (>= 0.0.4)
     faraday-net_http (2.0.3)
-    google-protobuf (3.21.1)
+    google-protobuf (3.21.7)
     googleapis-common-protos-types (1.3.1)
       google-protobuf (~> 3.14)
     mongo (2.18.1)

data/lib/helios/opentelemetry/sdk/helios_processor.rb

@@ -0,0 +1,37 @@
+require 'opentelemetry/sdk'
+require_relative '../semantic_attributes'
+
+HELIOS_TEST_TRIGGERED_TRACE = 'hs-triggered-test'.freeze
+
+ATTRIBUTES_TO_DROP = [
+  Helios::OpenTelemetry::SemanticAttributes::DB_QUERY_RESULT,
+  Helios::OpenTelemetry::SemanticAttributes::HTTP_RESPONSE_BODY,
+  Helios::OpenTelemetry::SemanticAttributes::HTTP_RESPONSE_HEADERS,
+  Helios::OpenTelemetry::SemanticAttributes::HTTP_REQUEST_BODY,
+  Helios::OpenTelemetry::SemanticAttributes::HTTP_REQUEST_HEADERS
+]
+
+class HeliosProcessor < ::OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor
+  def initialize(exporter, metadata_only = False)
+    @metadata_only = metadata_only
+    super(exporter)
+  end
+
+  def on_start(span, parent_context)
+    baggage = ::OpenTelemetry::Baggage.values(context: parent_context)
+    return unless baggage.key?(HELIOS_TEST_TRIGGERED_TRACE)
+
+    span.set_attribute(HELIOS_TEST_TRIGGERED_TRACE, 'true')
+
+    super(span, parent_context)
+  end
+
+  def on_finish(span)
+    if @metadata_only
+      new_attributes = span.attributes.dup.reject { |k, _| ATTRIBUTES_TO_DROP.include?(k) }
+      span.instance_variable_set(:@attributes, new_attributes)
+    end
+
+    super(span)
+  end
+end

data/lib/helios/opentelemetry/sdk/version.rb

@@ -1,7 +1,7 @@
 module Helios
   module OpenTelemetry
     module SDK
-      VERSION = '0.1.20'
+      VERSION = '0.1.22'
     end
   end
 end

data/lib/helios/opentelemetry/sdk.rb

@@ -3,6 +3,7 @@ require 'opentelemetry/exporter/otlp'
 require 'opentelemetry/instrumentation/all'
 require_relative 'sdk/version'
 require_relative 'sdk/instrumentations'
+require_relative 'sdk/helios_processor'
 
 module Helios
   module OpenTelemetry
@@ -37,11 +38,11 @@ module Helios
           end
           c.resource = ::OpenTelemetry::SDK::Resources::Resource.create(resource_attrs)
           apply_instrumentations(c)
-          c.add_span_processor(
-            ::OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor.new(
-              get_span_exporter(helios_config)
-            )
-          )
+          c.add_span_processor(HeliosProcessor.new(get_span_exporter(helios_config), helios_config[:metadata_only]))
+          puts 'Helios tracing initialized (' \
+            "service: #{service_name}, " \
+            "token: #{api_token[0, 3]}*****, " \
+            "environment: #{environment})"
         end
 
         send_debug_span
@@ -88,9 +89,11 @@ if auto_tracing_mode?
   api_token = ENV['HS_TOKEN']
   service_name = ENV['HS_SERVICE_NAME']
   environment = ENV['HS_ENVIRONMENT']
+  metadata_only = ENV['HS_METADATA_ONLY'].lowercase == 'true'
   enable = true
   Helios::OpenTelemetry::SDK.initialize(api_token: api_token,
                                         service_name: service_name,
                                         environment: environment,
+                                        metadata_only: metadata_only,
                                         enable: enable)
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: helios-opentelemetry-sdk
 version: !ruby/object:Gem::Version
-  version: 0.1.20
+  version: 0.1.22
 platform: ruby
 authors:
 - Team Helios
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-09-06 00:00:00.000000000 Z
+date: 2023-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -129,6 +129,7 @@ files:
 - bin/setup
 - helios-opentelemetry-sdk.gemspec
 - lib/helios/opentelemetry/sdk.rb
+- lib/helios/opentelemetry/sdk/helios_processor.rb
 - lib/helios/opentelemetry/sdk/instrumentations.rb
 - lib/helios/opentelemetry/sdk/patches/faraday_patch.rb
 - lib/helios/opentelemetry/sdk/patches/mongo_patch.rb