initial commit

Author: bachuv

azure/durable_functions/models/DurableOrchestrationClient.py

@@ -47,7 +47,9 @@ def __init__(self, context: str):
     async def start_new(self,
                         orchestration_function_name: str,
                         instance_id: Optional[str] = None,
-                        client_input: Optional[Any] = None) -> str:
+                        client_input: Optional[Any] = None,
+                        trace_parent: str = None,
+                        trace_state: str = None) -> str:
         """Start a new instance of the specified orchestrator function.
 
         If an orchestration instance with the specified ID already exists, the
@@ -72,7 +74,10 @@ async def start_new(self,
             instance_id=instance_id, orchestration_function_name=orchestration_function_name)
 
         response: List[Any] = await self._post_async_request(
-            request_url, self._get_json_input(client_input))
+            request_url,
+            self._get_json_input(client_input),
+            trace_parent,
+            trace_state)
 
         status_code: int = response[0]
         if status_code <= 202 and response[1]:

azure/durable_functions/models/utils/http_utils.py

@@ -3,7 +3,7 @@
 import aiohttp
 
 
-async def post_async_request(url: str, data: Any = None) -> List[Union[int, Any]]:
+async def post_async_request(url: str, data: Any = None, trace_parent: str = None, trace_state: str = None) -> List[Union[int, Any]]:
     """Post request with the data provided to the url provided.
 
     Parameters
@@ -20,7 +20,11 @@ async def post_async_request(url: str, data: Any = None) -> List[Union[int, Any]
     """
     async with aiohttp.ClientSession() as session:
         async with session.post(url,
-                                json=data) as response:
+                                json=data,
+                                headers={
+                                    "traceparent": trace_parent,
+                                    "tracestate": trace_state
+                                }) as response:
             # We disable aiohttp's input type validation
             # as the server may respond with alternative
             # data encodings. This is potentially unsafe.