add functional xray provider

Author: roger-zhangg

aws_lambda_powertools/tracing/provider/xray_tracer.py

@@ -2,7 +2,7 @@
 
 from contextlib import contextmanager
 from numbers import Number
-from typing import Generator
+from typing import Generator, Sequence
 
 from ...shared import constants
 from ...shared.lazy_import import LazyLoader
@@ -50,12 +50,18 @@ def trace(self, name: str) -> Generator[XraySpan, None, None]:
             yield XraySpan(subsegment=sub_segment)
 
     @contextmanager
-    def trace_async(self, name: str) -> Generator[XraySpan, None, None]:
-        with self.in_subsegment_async(name=name) as sub_segment:
+    async def trace_async(self, name: str) -> Generator[XraySpan, None, None]:
+        async with self.in_subsegment_async(name=name) as sub_segment:
             yield XraySpan(subsegment=sub_segment)
 
     def set_attribute(self, key: str, value: str | Number | bool, **kwargs) -> None:
         if kwargs.get("namespace", "") != "":
             self.put_metadata(key=key, value=value, namespace=kwargs["namespace"])
         else:
             self.put_annotation(key=key, value=value)
+
+    def patch(self, modules: Sequence[str]) -> None:
+        pass
+
+    def patch_all(self) -> None:
+        pass

aws_lambda_powertools/tracing/tracer.py

@@ -308,7 +308,7 @@ def handler(event, context):
 
         @functools.wraps(lambda_handler)
         def decorate(event, context, **kwargs):
-            with self.provider.in_subsegment(name=f"## {lambda_handler_name}") as subsegment:
+            with self.provider.trace(name=f"## {lambda_handler_name}") as subsegment:
                 try:
                     logger.debug("Calling lambda handler")
                     response = lambda_handler(event, context, **kwargs)
@@ -573,7 +573,7 @@ def _decorate_async_function(
     ):
         @functools.wraps(method)
         async def decorate(*args, **kwargs):
-            async with self.provider.in_subsegment_async(name=f"## {method_name}") as subsegment:
+            async with self.provider.trace_async(name=f"## {method_name}") as subsegment:
                 try:
                     logger.debug(f"Calling method: {method_name}")
                     response = await method(*args, **kwargs)
@@ -606,7 +606,7 @@ def _decorate_generator_function(
     ):
         @functools.wraps(method)
         def decorate(*args, **kwargs):
-            with self.provider.in_subsegment(name=f"## {method_name}") as subsegment:
+            with self.provider.trace(name=f"## {method_name}") as subsegment:
                 try:
                     logger.debug(f"Calling method: {method_name}")
                     result = yield from method(*args, **kwargs)
@@ -640,7 +640,7 @@ def _decorate_generator_function_with_context_manager(
         @functools.wraps(method)
         @contextlib.contextmanager
         def decorate(*args, **kwargs):
-            with self.provider.in_subsegment(name=f"## {method_name}") as subsegment:
+            with self.provider.trace(name=f"## {method_name}") as subsegment:
                 try:
                     logger.debug(f"Calling method: {method_name}")
                     with method(*args, **kwargs) as return_val:
@@ -673,7 +673,7 @@ def _decorate_sync_function(
     ) -> AnyCallableT:
         @functools.wraps(method)
         def decorate(*args, **kwargs):
-            with self.provider.in_subsegment(name=f"## {method_name}") as subsegment:
+            with self.provider.trace(name=f"## {method_name}") as subsegment:
                 try:
                     logger.debug(f"Calling method: {method_name}")
                     response = method(*args, **kwargs)