Sha256: 603a4beda6850609ff6492fa61d16362a27da2e27a09243624cb7e0bec3e3e3f
Contents?: true
Size: 1.81 KB
Versions: 31
Compression:
Stored size: 1.81 KB
Contents
require 'ddtrace/contrib/registry'
module Datadog
module Contrib
# Extensions that can be added to the base library
# Adds registry, configuration access for integrations.
module Extensions
def self.extended(base)
Datadog.send(:extend, Helpers)
Datadog::Configuration::Settings.send(:include, Configuration::Settings)
end
# Helper methods for Datadog module.
module Helpers
def registry
configuration.registry
end
end
module Configuration
# Extensions for Datadog::Configuration::Settings
module Settings
InvalidIntegrationError = Class.new(StandardError)
def self.included(base)
# Add the additional options to the global configuration settings
base.instance_eval do
option :registry, default: Registry.new
end
end
def [](integration_name, configuration_name = :default)
integration = fetch_integration(integration_name)
integration.configuration(configuration_name) unless integration.nil?
end
def use(integration_name, options = {}, &block)
integration = fetch_integration(integration_name)
unless integration.nil?
configuration_name = options[:describes] || :default
filtered_options = options.reject { |k, _v| k == :describes }
integration.configure(configuration_name, filtered_options, &block)
end
integration.patch if integration.respond_to?(:patch)
end
private
def fetch_integration(name)
get_option(:registry)[name] ||
raise(InvalidIntegrationError, "'#{name}' is not a valid integration.")
end
end
end
end
end
end
Version data entries
31 entries across 31 versions & 2 rubygems