add enable_telemetry setting to the BaseFactory

Author: hrodmn

src/titiler/application/titiler/application/main.py

@@ -118,6 +118,7 @@ def validate_access_token(access_token: str = Security(api_key_query)):
             cogViewerExtension(),
             stacExtension(),
         ],
+        enable_telemtry=api_settings.telemetry_enabled,
     )
 
     app.include_router(
@@ -138,6 +139,7 @@ def validate_access_token(access_token: str = Security(api_key_query)):
             stacViewerExtension(),
             stacRenderExtension(),
         ],
+        enable_telemtry=api_settings.telemetry_enabled,
     )
 
     app.include_router(
@@ -151,7 +153,10 @@ def validate_access_token(access_token: str = Security(api_key_query)):
 ###############################################################################
 # Mosaic endpoints
 if not api_settings.disable_mosaic:
-    mosaic = MosaicTilerFactory(router_prefix="/mosaicjson")
+    mosaic = MosaicTilerFactory(
+        router_prefix="/mosaicjson",
+        enable_telemtry=api_settings.telemetry_enabled,
+    )
     app.include_router(
         mosaic.router,
         prefix="/mosaicjson",

src/titiler/core/titiler/core/factory.py

@@ -158,6 +158,8 @@ class BaseFactory(metaclass=abc.ABCMeta):
 
     conforms_to: Set[str] = field(factory=set)
 
+    enable_telemtry: bool = field(default=False)
+
     def __attrs_post_init__(self):
         """Post Init: register route and configure specific options."""
         # prefix for endpoint's operationId
@@ -175,7 +177,8 @@ def __attrs_post_init__(self):
         for scopes, dependencies in self.route_dependencies:
             self.add_route_dependencies(scopes=scopes, dependencies=dependencies)
 
-        self.add_telemetry()
+        if self.enable_telemtry:
+            self.add_telemetry()
 
     @abc.abstractmethod
     def register_routes(self):
@@ -241,6 +244,9 @@ def add_telemetry(self):
         of each APIRoute to ensure consistent OpenTelemetry tracing.
         """
         if not factory_trace.decorator_enabled:
+            logger.warning(
+                "telemetry enabled for the factory class but tracing is not available"
+            )
             return
 
         for route in self.router.routes:

src/titiler/core/titiler/core/telemetry.py

@@ -35,7 +35,7 @@ def add_span_attributes(attributes: Dict[str, Any]) -> None:
 
 
 def flatten_dict(d: dict, parent_key: str = "", sep: str = ".") -> dict:
-    """Flattens a nested dictionary."""
+    """Flattens a nested dictionary for adding span attributes."""
     items = {}
     for k, v in d.items():
         new_key = parent_key + sep + k if parent_key else k
@@ -101,20 +101,27 @@ def factory_trace(
     *,
     factory_instance: Optional[Any] = None,
 ) -> Any:
-    """A decorator for Factory methods that automatically handles tracing."""
+    """A decorator for Factory methods that automatically handles tracing for factory methods"""
 
     def decorator(func: Callable[P, Any]) -> Callable[P, Any]:
         if not tracer:
             return func
 
         op_name = func.__name__
 
+        attributes = {}
+        if factory_instance:
+            if hasattr(factory_instance, "reader"):
+                attributes["reader"] = str(factory_instance.reader)
+            if hasattr(factory_instance, "backend"):
+                attributes["backend"] = str(factory_instance.backend)
+
         if inspect.iscoroutinefunction(func):
 
             @functools.wraps(func)
             async def async_wrapper(*args: P.args, **kwargs: P.kwargs) -> Any:
                 span_name = _get_span_name(op_name, factory_instance)
-                with operation_tracer(span_name):
+                with operation_tracer(span_name, attributes=attributes):
                     return await func(*args, **kwargs)
 
             return async_wrapper
@@ -123,7 +130,7 @@ async def async_wrapper(*args: P.args, **kwargs: P.kwargs) -> Any:
             @functools.wraps(func)
             def sync_wrapper(*args: P.args, **kwargs: P.kwargs) -> Any:
                 span_name = _get_span_name(op_name, factory_instance)
-                with operation_tracer(span_name):
+                with operation_tracer(span_name, attributes=attributes):
                     return func(*args, **kwargs)
 
             return sync_wrapper