feat: http endpoint filtering for aws resources

src/lumigo_opentelemetry/instrumentations/botocore/__init__.py

@@ -1,5 +1,13 @@
+from typing import Dict, Any
+
+from lumigo_opentelemetry.libs.general_utils import lumigo_safe_execute
+from lumigo_opentelemetry.resources.span_processor import set_span_skip_export
+from opentelemetry.trace import Span, SpanKind
+
 from lumigo_opentelemetry.instrumentations import AbstractInstrumentor
 from lumigo_opentelemetry.instrumentations.botocore.parsers import AwsParser
+from lumigo_opentelemetry.libs.sampling import should_sample
+from lumigo_opentelemetry.utils.span_utils import safe_get_span_attributes
 
 
 class BotoCoreInstrumentorWrapper(AbstractInstrumentor):
@@ -21,8 +29,19 @@ def install_instrumentation(self) -> None:
 
         BotocoreInstrumentor().instrument(
             request_hook=AwsParser.request_hook,
-            response_hook=AwsParser.response_hook,
+            response_hook=filtered_resource_hook,
         )
 
 
+def filtered_resource_hook(
+    span: Span, service_name: str, operation_name: str, result: Dict[Any, Any]
+) -> None:
+    AwsParser.response_hook(span, service_name, operation_name, result)
+    with lumigo_safe_execute("aws: response_hook skip check"):
+        span_attributes = safe_get_span_attributes(span)
+        if span_attributes:
+            if not should_sample(span_attributes, SpanKind.CLIENT):
+                set_span_skip_export(span)
+
+
 instrumentor: AbstractInstrumentor = BotoCoreInstrumentorWrapper()