opentelemetry-ruby/index.rb at 9ea8988228cd404a0b5d4bb763d0b8c2fe3468e4 · odigos-io/opentelemetry-ruby · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# refs:
#  - https://github.com/newrelic/newrelic-ruby-agent/blob/dev/lib/boot/strap.rb
#  - https://github.com/open-telemetry/opentelemetry-operator/blob/b4e22034a1cdc498e3f9e0cb95f7c6204972f42c/autoinstrumentation/ruby/src/autoinstrumentation.rb

module OTelBundlerPatch
  def require(*_groups)
    super
    require_otel
  end

  # def require_resources
  #   additional_resource = ::OpenTelemetry::SDK::Resources::Resource.create({})
  #   additional_resource = additional_resource.merge(::OpenTelemetry::Resource::Detector::Container.detect) if defined? ::OpenTelemetry::Resource::Detector::Container
  #   additional_resource = additional_resource.merge(::OpenTelemetry::Resource::Detector::GoogleCloudPlatform.detect) if defined? ::OpenTelemetry::Resource::Detector::GoogleCloudPlatform
  #   additional_resource = additional_resource.merge(::OpenTelemetry::Resource::Detector::Azure.detect) if defined? ::OpenTelemetry::Resource::Detector::Azure
  #   additional_resource
  # end

  def require_otel
    lib = File.expand_path('..', __dir__)
    $LOAD_PATH.reject! { |path| path.include?('zero-code-instrumentation') }
    $LOAD_PATH.unshift(lib)

    begin
      OpenTelemetry::SDK.configure do |c|
        c.service_name = ENV['OTEL_SERVICE_NAME']
        # c.resource = require_resources
        c.use_all # enables all instrumentation!
      end
      OpenTelemetry.logger.info { 'OpenTelemetry initialized' }
    rescue StandardError => e
      OpenTelemetry.logger.info { "OpenTelemetry failed to initialize. Error: #{e.message}" }
    end
  end
end

gem_path = ENV['ODIGOS_GEM_PATH'] || Gem.dir
Dir.glob("#{gem_path}/gems/*").each do |file|
  $LOAD_PATH.unshift("#{file}/lib")
end

require 'bundler'

require 'opentelemetry-api'
require 'opentelemetry-sdk'
require 'opentelemetry-exporter-otlp'
require 'opentelemetry-instrumentation-all'

# require 'opentelemetry-resource_detectors'
# require 'opentelemetry-resource-detector-container'
# require 'opentelemetry-resource-detector-google_cloud_platform'
# require 'opentelemetry-resource-detector-azure'

require 'opentelemetry-helpers-mysql'
require 'opentelemetry-helpers-sql'
require 'opentelemetry-helpers-sql-obfuscation'

Bundler::Runtime.prepend(OTelBundlerPatch)
Bundler.require if ENV['REQUIRE_BUNDLER'].to_s == 'true'