Add customizer to auto instrumentation code.

Author: DylanRussell

opentelemetry-distro/pyproject.toml

@@ -40,6 +40,9 @@ otlp = [
 [project.entry-points.opentelemetry_configurator]
 configurator = "opentelemetry.distro:OpenTelemetryConfigurator"
 
+[project.entry-points.opentelemetry_customizer]
+customizer = "opentelemetry.distro:OpenTelemetryCustomizer"
+
 [project.entry-points.opentelemetry_distro]
 distro = "opentelemetry.distro:OpenTelemetryDistro"
 

opentelemetry-distro/src/opentelemetry/distro/__init__.py

@@ -19,6 +19,7 @@
     OTEL_METRICS_EXPORTER,
     OTEL_TRACES_EXPORTER,
 )
+from opentelemetry.sdk._config_customizer import _BaseConfiguratorCustomizer
 from opentelemetry.instrumentation.distro import BaseDistro
 from opentelemetry.sdk._configuration import _OTelSDKConfigurator
 from opentelemetry.sdk.environment_variables import OTEL_EXPORTER_OTLP_PROTOCOL
@@ -27,6 +28,8 @@
 class OpenTelemetryConfigurator(_OTelSDKConfigurator):
     pass
 
+class OpenTelemetryCustomizer(_BaseConfiguratorCustomizer):
+    pass
 
 class OpenTelemetryDistro(BaseDistro):
     """

opentelemetry-instrumentation/src/opentelemetry/instrumentation/auto_instrumentation/_load.py

@@ -23,6 +23,7 @@
 from opentelemetry.instrumentation.environment_variables import (
     OTEL_PYTHON_CONFIGURATOR,
     OTEL_PYTHON_DISABLED_INSTRUMENTATIONS,
+    OTEL_PYTHON_CUSTOMIZER,
     OTEL_PYTHON_DISTRO,
 )
 from opentelemetry.instrumentation.version import __version__
@@ -133,6 +134,14 @@ def _load_instrumentors(distro):
 
 
 def _load_configurators():
+    customizer = None
+    if environ.get(OTEL_PYTHON_CUSTOMIZER, None):
+        for entry_point in entry_points(
+            group="opentelemetry_customizer",
+            name=environ.get(OTEL_PYTHON_CUSTOMIZER),
+        ):
+            customizer = entry_point.load()()
+            break
     configurator_name = environ.get(OTEL_PYTHON_CONFIGURATOR, None)
     configured = None
     for entry_point in entry_points(group="opentelemetry_configurator"):
@@ -149,7 +158,8 @@ def _load_configurators():
                 or configurator_name == entry_point.name
             ):
                 entry_point.load()().configure(
-                    auto_instrumentation_version=__version__
+                    auto_instrumentation_version=__version__,
+                    customizer=customizer,
                 )  # type: ignore
                 configured = entry_point.name
             else:

opentelemetry-instrumentation/src/opentelemetry/instrumentation/environment_variables.py

@@ -26,3 +26,7 @@
 """
 .. envvar:: OTEL_PYTHON_CONFIGURATOR
 """
+OTEL_PYTHON_CUSTOMIZER = "OTEL_PYTHON_CUSTOMIZER"
+"""
+.. envvar:: OTEL_PYTHON_CUSTOMIZER
+"""