|
| 1 | +# frozen_string_literal: true |
| 2 | + |
| 3 | +# Copyright The OpenTelemetry Authors |
| 4 | +# |
| 5 | +# SPDX-License-Identifier: Apache-2.0 |
| 6 | + |
| 7 | +begin |
| 8 | + # mitigate potential load order issues by always requiring |
| 9 | + # the metrics-api gem if it is in the bundle. This module is |
| 10 | + # intended to function with or without it, but loading it after |
| 11 | + # any of the definition queries have been made may cause false negatives. |
| 12 | + require 'opentelemetry-metrics-api' |
| 13 | +rescue LoadError # rubocop: disable Lint/SuppressedException |
| 14 | +end |
| 15 | + |
| 16 | +module OpenTelemetry |
| 17 | + module Instrumentation |
| 18 | + # Extensions to Instrumentation::Base that handle metrics instruments. |
| 19 | + # The goal here is to allow metrics to be added gradually to instrumentation libraries, |
| 20 | + # without requiring that the metrics-sdk or metrics-api gems are present in the bundle |
| 21 | + # (if they are not, or if the metrics-api gem does not meet the minimum version requirement, |
| 22 | + # the no-op edition is installed.) |
| 23 | + module Metrics |
| 24 | + METER_TYPES = %i[ |
| 25 | + counter |
| 26 | + observable_counter |
| 27 | + histogram |
| 28 | + gauge |
| 29 | + observable_gauge |
| 30 | + up_down_counter |
| 31 | + observable_up_down_counter |
| 32 | + ].freeze |
| 33 | + |
| 34 | + def self.prepended(base) |
| 35 | + base.prepend(Compatibility) |
| 36 | + base.extend(Compatibility) |
| 37 | + base.extend(Registration) |
| 38 | + |
| 39 | + if base.metrics_compatible? |
| 40 | + base.prepend(Extensions) |
| 41 | + else |
| 42 | + base.prepend(NoopExtensions) |
| 43 | + end |
| 44 | + end |
| 45 | + |
| 46 | + # Methods to check whether the metrics API is defined |
| 47 | + # and is a compatible version |
| 48 | + module Compatibility |
| 49 | + METRICS_API_MINIMUM_GEM_VERSION = Gem::Version.new('0.2.0') |
| 50 | + |
| 51 | + def metrics_defined? |
| 52 | + defined?(OpenTelemetry::Metrics) |
| 53 | + end |
| 54 | + |
| 55 | + def metrics_compatible? |
| 56 | + metrics_defined? && Gem.loaded_specs['opentelemetry-metrics-api'].version >= METRICS_API_MINIMUM_GEM_VERSION |
| 57 | + end |
| 58 | + |
| 59 | + extend(self) |
| 60 | + end |
| 61 | + |
| 62 | + # class-level methods to declare and register metrics instruments. |
| 63 | + # This can be extended even if metrics is not active or present. |
| 64 | + module Registration |
| 65 | + METER_TYPES.each do |instrument_kind| |
| 66 | + define_method(instrument_kind) do |name, **opts, &block| |
| 67 | + opts[:callback] ||= block if block |
| 68 | + register_instrument(instrument_kind, name, **opts) |
| 69 | + end |
| 70 | + end |
| 71 | + |
| 72 | + def register_instrument(kind, name, **opts) |
| 73 | + key = [kind, name] |
| 74 | + if instrument_configs.key?(key) |
| 75 | + warn("Duplicate instrument configured for #{self}: #{key.inspect}") |
| 76 | + else |
| 77 | + instrument_configs[key] = opts |
| 78 | + end |
| 79 | + end |
| 80 | + |
| 81 | + def instrument_configs |
| 82 | + @instrument_configs ||= {} |
| 83 | + end |
| 84 | + end |
| 85 | + |
| 86 | + # No-op instance methods for metrics instruments. |
| 87 | + module NoopExtensions |
| 88 | + METER_TYPES.each do |kind| |
| 89 | + define_method(kind) { |*, **| } # rubocop: disable Lint/EmptyBlock |
| 90 | + end |
| 91 | + |
| 92 | + def with_meter; end |
| 93 | + |
| 94 | + def metrics_enabled? |
| 95 | + false |
| 96 | + end |
| 97 | + end |
| 98 | + |
| 99 | + # Instance methods for metrics instruments. |
| 100 | + module Extensions |
| 101 | + %i[ |
| 102 | + counter |
| 103 | + observable_counter |
| 104 | + histogram |
| 105 | + gauge |
| 106 | + observable_gauge |
| 107 | + up_down_counter |
| 108 | + observable_up_down_counter |
| 109 | + ].each do |kind| |
| 110 | + define_method(kind) do |name| |
| 111 | + get_metrics_instrument(kind, name) |
| 112 | + end |
| 113 | + end |
| 114 | + |
| 115 | + # This is based on a variety of factors, and should be invalidated when @config changes. |
| 116 | + # It should be explicitly set in `prepare_install` for now. |
| 117 | + def metrics_enabled? |
| 118 | + !!@metrics_enabled |
| 119 | + end |
| 120 | + |
| 121 | + # @api private |
| 122 | + # ONLY yields if the meter is enabled. |
| 123 | + def with_meter |
| 124 | + yield @meter if metrics_enabled? |
| 125 | + end |
| 126 | + |
| 127 | + private |
| 128 | + |
| 129 | + def compute_metrics_enabled |
| 130 | + return false unless metrics_compatible? |
| 131 | + return false if metrics_disabled_by_env_var? |
| 132 | + |
| 133 | + !!@config[:metrics] || metrics_enabled_by_env_var? |
| 134 | + end |
| 135 | + |
| 136 | + # Checks if this instrumentation's metrics are enabled by env var. |
| 137 | + # This follows the conventions as outlined above, using `_METRICS_ENABLED` as a suffix. |
| 138 | + # Unlike INSTRUMENTATION_*_ENABLED variables, these are explicitly opt-in (i.e. |
| 139 | + # if the variable is unset, and `metrics: true` is not in the instrumentation's config, |
| 140 | + # the metrics will not be enabled) |
| 141 | + def metrics_enabled_by_env_var? |
| 142 | + ENV.key?(metrics_env_var_name) && ENV[metrics_env_var_name] != 'false' |
| 143 | + end |
| 144 | + |
| 145 | + def metrics_disabled_by_env_var? |
| 146 | + ENV[metrics_env_var_name] == 'false' |
| 147 | + end |
| 148 | + |
| 149 | + def metrics_env_var_name |
| 150 | + @metrics_env_var_name ||= |
| 151 | + begin |
| 152 | + var_name = name.dup |
| 153 | + var_name.upcase! |
| 154 | + var_name.gsub!('::', '_') |
| 155 | + var_name.gsub!('OPENTELEMETRY_', 'OTEL_RUBY_') |
| 156 | + var_name << '_METRICS_ENABLED' |
| 157 | + var_name |
| 158 | + end |
| 159 | + end |
| 160 | + |
| 161 | + def prepare_install |
| 162 | + @metrics_enabled = compute_metrics_enabled |
| 163 | + if metrics_defined? |
| 164 | + @metrics_instruments = {} |
| 165 | + @instrument_mutex = Mutex.new |
| 166 | + end |
| 167 | + |
| 168 | + @meter = OpenTelemetry.meter_provider.meter(name, version: version) if metrics_enabled? |
| 169 | + |
| 170 | + super |
| 171 | + end |
| 172 | + |
| 173 | + def get_metrics_instrument(kind, name) |
| 174 | + # TODO: we should probably return *something* |
| 175 | + # if metrics is not enabled, but if the api is undefined, |
| 176 | + # it's unclear exactly what would be suitable. |
| 177 | + # For now, there are no public methods that call this |
| 178 | + # if metrics isn't defined. |
| 179 | + return unless metrics_defined? |
| 180 | + |
| 181 | + @metrics_instruments.fetch([kind, name]) do |key| |
| 182 | + @instrument_mutex.synchronize do |
| 183 | + @metrics_instruments[key] ||= create_configured_instrument(kind, name) |
| 184 | + end |
| 185 | + end |
| 186 | + end |
| 187 | + |
| 188 | + def create_configured_instrument(kind, name) |
| 189 | + config = self.class.instrument_configs[[kind, name]] |
| 190 | + |
| 191 | + if config.nil? |
| 192 | + Kernel.warn("unconfigured instrument requested: #{kind} of '#{name}'") |
| 193 | + return |
| 194 | + end |
| 195 | + |
| 196 | + meter.public_send(:"create_#{kind}", name, **config) |
| 197 | + end |
| 198 | + end |
| 199 | + end |
| 200 | + end |
| 201 | +end |
0 commit comments