codecov_opentelem 0.1.3 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6914eed45d30e64ce8c2982e130fc182539e953e463ec31eb5924ebac9347002
-  data.tar.gz: bdc8110f1234339151b7a8bbc51aead42603faf263f1fc75d1fc89f73ff07993
+  metadata.gz: a837d9537c3dccbc0d656e4bd7c7a196073c95e5686a02252a69f267b49be89b
+  data.tar.gz: 6feed841349acebe12bbc6afc3f1f87f9b897d86867e9fc91a4a442f3930b10f
 SHA512:
-  metadata.gz: 66932b7701993e76e43798ec280718293333ab88520cb1bd881f1a9f0ce653dc4680b727fb22288e082296f6479bf8fe7161d3e834c3c562c7ccce4870d8b2f2
-  data.tar.gz: a727af20f23a98ca148612f0cc792da9733f5c8e37290a4e532ba8396a145b6129e77f1bf3e14b919407a82140ff39227a536082bd6092b1dd7bbb76032d005c
+  metadata.gz: c0823f094eb916153de0aae3a29496bd9f1245648159435e6c070ef86e371b9015f399efc3f54e3cc9fe45081357c524ea9c5fbf01f216fb78f3f41b3e9853f7
+  data.tar.gz: 7fd37d56e8ca61aac7ee18537a486a865efe4a834a77eabeb8b3dab17a002c4430cfca3a6af4b603dbbed0e598b803405cd61045b6291aaa692b5b9955adf591

data/Gemfile.lock

@@ -0,0 +1,221 @@
+PATH
+  remote: .
+  specs:
+    codecov_opentelem (0.1.5)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    diff-lcs (1.5.0)
+    google-protobuf (3.21.1-x86_64-darwin)
+    googleapis-common-protos-types (1.3.1)
+      google-protobuf (~> 3.14)
+    opentelemetry-api (1.0.2)
+    opentelemetry-common (0.19.6)
+      opentelemetry-api (~> 1.0)
+    opentelemetry-exporter-otlp (0.22.0)
+      google-protobuf (~> 3.19)
+      googleapis-common-protos-types (~> 1.3)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.6)
+      opentelemetry-sdk (~> 1.0)
+      opentelemetry-semantic_conventions
+    opentelemetry-instrumentation-action_pack (0.2.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+      opentelemetry-instrumentation-rack (~> 0.21.0)
+    opentelemetry-instrumentation-action_view (0.3.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-active_support (~> 0.1)
+      opentelemetry-instrumentation-base (~> 0.20)
+    opentelemetry-instrumentation-active_job (0.3.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-active_model_serializers (0.19.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-active_record (0.4.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+      ruby2_keywords
+    opentelemetry-instrumentation-active_support (0.2.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-all (0.25.0)
+      opentelemetry-instrumentation-action_pack (~> 0.2.0)
+      opentelemetry-instrumentation-action_view (~> 0.3.0)
+      opentelemetry-instrumentation-active_job (~> 0.3.0)
+      opentelemetry-instrumentation-active_model_serializers (~> 0.19.0)
+      opentelemetry-instrumentation-active_record (~> 0.4.0)
+      opentelemetry-instrumentation-active_support (~> 0.2.0)
+      opentelemetry-instrumentation-aws_sdk (~> 0.3.0)
+      opentelemetry-instrumentation-bunny (~> 0.19.0)
+      opentelemetry-instrumentation-concurrent_ruby (~> 0.20.0)
+      opentelemetry-instrumentation-dalli (~> 0.22.0)
+      opentelemetry-instrumentation-delayed_job (~> 0.19.0)
+      opentelemetry-instrumentation-ethon (~> 0.20.0)
+      opentelemetry-instrumentation-excon (~> 0.20.0)
+      opentelemetry-instrumentation-faraday (~> 0.21.0)
+      opentelemetry-instrumentation-graphql (~> 0.20.0)
+      opentelemetry-instrumentation-http (~> 0.20.0)
+      opentelemetry-instrumentation-http_client (~> 0.20.0)
+      opentelemetry-instrumentation-koala (~> 0.19.0)
+      opentelemetry-instrumentation-lmdb (~> 0.21.0)
+      opentelemetry-instrumentation-mongo (~> 0.20.0)
+      opentelemetry-instrumentation-mysql2 (~> 0.21.0)
+      opentelemetry-instrumentation-net_http (~> 0.20.0)
+      opentelemetry-instrumentation-pg (~> 0.21.0)
+      opentelemetry-instrumentation-que (~> 0.4.0)
+      opentelemetry-instrumentation-rack (~> 0.21.0)
+      opentelemetry-instrumentation-rails (~> 0.22.0)
+      opentelemetry-instrumentation-rdkafka (~> 0.2.0)
+      opentelemetry-instrumentation-redis (~> 0.23.0)
+      opentelemetry-instrumentation-resque (~> 0.3.0)
+      opentelemetry-instrumentation-restclient (~> 0.20.0)
+      opentelemetry-instrumentation-ruby_kafka (~> 0.19.0)
+      opentelemetry-instrumentation-sidekiq (~> 0.22.0)
+      opentelemetry-instrumentation-sinatra (~> 0.20.0)
+      opentelemetry-instrumentation-trilogy (~> 0.51.0)
+    opentelemetry-instrumentation-aws_sdk (0.3.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-base (0.21.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-registry (~> 0.1)
+    opentelemetry-instrumentation-bunny (0.19.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-concurrent_ruby (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-dalli (0.22.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-delayed_job (0.19.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-ethon (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-excon (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-faraday (0.21.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-graphql (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-http (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-http_client (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-koala (0.19.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-lmdb (0.21.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-mongo (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-mysql2 (0.21.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-net_http (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-pg (0.21.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-que (0.4.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-rack (0.21.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-rails (0.22.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-action_pack (~> 0.2.0)
+      opentelemetry-instrumentation-action_view (~> 0.3.0)
+      opentelemetry-instrumentation-active_record (~> 0.4.0)
+      opentelemetry-instrumentation-active_support (~> 0.2.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-rdkafka (0.2.0)
+      opentelemetry-api (~> 1.0.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-redis (0.23.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-resque (0.3.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-restclient (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-ruby_kafka (0.19.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-sidekiq (0.22.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-sinatra (0.20.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+    opentelemetry-instrumentation-trilogy (0.51.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-instrumentation-base (~> 0.21.0)
+      opentelemetry-semantic_conventions (>= 1.8.0)
+    opentelemetry-registry (0.1.0)
+      opentelemetry-api (~> 1.0.1)
+    opentelemetry-sdk (1.1.0)
+      opentelemetry-api (~> 1.0)
+      opentelemetry-common (~> 0.19.3)
+      opentelemetry-registry (~> 0.1)
+      opentelemetry-semantic_conventions
+    opentelemetry-semantic_conventions (1.8.0)
+      opentelemetry-api (~> 1.0)
+    rake (13.0.6)
+    rspec (3.11.0)
+      rspec-core (~> 3.11.0)
+      rspec-expectations (~> 3.11.0)
+      rspec-mocks (~> 3.11.0)
+    rspec-core (3.11.0)
+      rspec-support (~> 3.11.0)
+    rspec-expectations (3.11.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.11.0)
+    rspec-mocks (3.11.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.11.0)
+    rspec-support (3.11.0)
+    ruby2_keywords (0.0.5)
+
+PLATFORMS
+  x86_64-darwin-21
+
+DEPENDENCIES
+  codecov_opentelem!
+  opentelemetry-exporter-otlp
+  opentelemetry-instrumentation-all
+  opentelemetry-sdk
+  rake (~> 13.0)
+  rspec
+
+BUNDLED WITH
+   2.3.16

data/lib/codecov_opentelem/coverage_exporter.rb

@@ -1,11 +1,12 @@
 # frozen_string_literal: true
 
+require 'base64'
 require 'json'
-require 'uri'
 require 'net/http'
+require 'uri'
 
-require 'opentelemetry/sdk'
 require 'coverage'
+require 'opentelemetry/sdk'
 
 module CoverageSpanFilter
   REGEX_NAME = 'name_regex'
@@ -14,28 +15,27 @@ end
 
 class CodecovCoverageStorageManager
   def initialize(filters)
-    Coverage.start
     @filters = filters
     @inner = {}
   end
 
   def start_coverage_for_span(span)
     if !@filters[CoverageSpanFilter::REGEX_NAME].nil? && @filters[CoverageSpanFilter::REGEX_NAME].match(span.name)
+      puts 'returned false'
       return false
     end
 
-    Coverage.resume unless Coverage.running?
+    Coverage.start unless Coverage.running?
     true
   end
 
-  def stop_coverage_for_span(_span)
+  def stop_coverage_for_span(span)
     span_id = span.context.span_id
-    Coverage.suspend
     @inner[span_id] = Coverage.result(stop: false, clear: true)
   end
 
-  def pop_coverage_for_span(_span)
-    span_id = span.context.span_id
+  def pop_coverage_for_span(span)
+    span_id = span.span_id
     @inner.delete(span_id)
   end
 end
@@ -48,10 +48,10 @@ class CodecovCoverageGenerator < OpenTelemetry::SDK::Trace::SpanProcessor
   end
 
   def on_start(span, _parent_context = nil)
-    @cov_storage.start_coverage_for_span(span) if rand < @sample_rate
+    @cov_storage.start_coverage_for_span(span) if rand() < @sample_rate
   end
 
-  def on_end(span)
+  def on_finish(span)
     @cov_storage.stop_coverage_for_span(span)
   end
 end
@@ -66,32 +66,59 @@ class CoverageExporter < OpenTelemetry::SDK::Trace::Export::SpanExporter
     @untracked_export_rate = untracked_export_rate
   end
 
-  def export(spans)
+  def export(spans, timeout)
     tracked_spans = []
     untracked_spans = []
+
     spans.each do |span|
-      cov = cov_storage.pop_coverage_for_span(span)
-      s = JSON.parse(span)
+      cov = @cov_storage.pop_coverage_for_span(span)
+
+      span_hash = Hash.new
+      span.to_h.each do |k, v|
+        v = v.dup.force_encoding("ISO-8859-1").encode("UTF-8") if v.is_a? String
+        span_hash[k] = v
+      end
+
+      cov_string = cov.map{ |k, v| "\"#{k}\":[#{v.map{ |i| i.nil? ? "null" : i}.join(',')}]"}.join(',')
+      cov_string = "{\"coverage\":{#{cov_string}}}"
+
       if !cov.nil?
-        s['codecov'] = { 'type' => 'bytes', 'coverage' => cov }
-        tracked_spans.append(s)
-      elsif rand < @untracked_export_rate
-        untracked_spans.append(s)
+        span_hash['codecov'] = {
+          'coverage': Base64.strict_encode64(cov_string),
+          'type': 'bytes',
+        }
+        tracked_spans.append(span_hash)
+      elsif rand() < @untracked_export_rate
+        untracked_spans.append(span_hash)
       end
     end
-    return SpanExportResult.SUCCESS if !tracked_spans && !untracked_spans
-
-    res = Net::HTTP.post(URI.join(@codecov_url, '/profiling/uploads'),
-                         { 'code' => @code }.to_json,
-                         'Content-Type' => 'application/json',
-                         'Authorization' => "repotoken #{@repository_token}")
-    return SpanExportResult.FAILURE if res.is_a?(Net::HTTPError)
-
-    location = res.body.raw_upload_location
-    Net::HTTP.put(URI(location),
-                  { 'spans' => tracked_spans,
-                    'untracked' => untracked_spans }.to_json,
-                  'Content-Type' => 'text/plain')
-    SpanExportResult.SUCCESS
+
+    return OpenTelemetry::SDK::Trace::Export::SUCCESS if !tracked_spans && !untracked_spans
+
+    body = { 'profiling' => @code }.to_json
+    res = Net::HTTP.post(
+      URI.join(@codecov_url, '/profiling/uploads'),
+      { 'profiling' => @code }.to_json,
+      'Content-Type' => 'application/json',
+      'Authorization' => "repotoken #{@repository_token}"
+    )
+    return OpenTelemetry::SDK::Trace::Export::FAILURE if res.is_a?(Net::HTTPError)
+
+    location = JSON(res.body)['raw_upload_location']
+    uri = URI(location)
+    https = Net::HTTP.new(uri.host, uri.port)
+    https.use_ssl = true
+    req = Net::HTTP::Put.new(
+      location,
+      {
+        'Content-Type' => 'text/plain'
+      }
+    )
+    req.body = {
+      'spans' => tracked_spans,
+      'untracked' => untracked_spans
+    }.to_json
+    https.request(req)
+    OpenTelemetry::SDK::Trace::Export::SUCCESS
   end
 end

data/lib/codecov_opentelem/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CodecovOpentelem
-  VERSION = "0.1.3"
+  VERSION = "0.1.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: codecov_opentelem
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.6
 platform: ruby
 authors:
 - Ahmed-Elghrbawy
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-06-17 00:00:00.000000000 Z
+date: 2022-06-21 00:00:00.000000000 Z
 dependencies: []
 description: 
 email:
@@ -19,6 +19,7 @@ extra_rdoc_files: []
 files:
 - CHANGELOG.md
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile