Copy context when submission

aiomisc/thread_pool.py

@@ -32,6 +32,10 @@ def context_partial(
     func: F, *args: Any,
     **kwargs: Any,
 ) -> Any:
+    warnings.warn(
+        "context_partial has been deprecated and will be removed",
+        DeprecationWarning,
+    )
     context = contextvars.copy_context()
     return partial(context.run, func, *args, **kwargs)
 
@@ -47,6 +51,7 @@ class WorkItemBase:
     kwargs: Dict[str, Any]
     future: asyncio.Future
     loop: asyncio.AbstractEventLoop
+    context: contextvars.Context
 
 
 class ThreadPoolStatistic(Statistic):
@@ -76,10 +81,11 @@ def __call__(self, statistic: ThreadPoolStatistic) -> None:
             return
 
         result, exception = None, None
-
         delta = -time.monotonic()
         try:
-            result = self.func(*self.args, **self.kwargs)
+            result = self.context.run(
+                self.func, *self.args, **self.kwargs,
+            )
             statistic.success += 1
         except BaseException as e:
             statistic.error += 1
@@ -195,6 +201,7 @@ def submit(  # type: ignore
                     args=args,
                     kwargs=kwargs,
                     future=future,
+                    context=contextvars.copy_context(),
                     loop=loop,
                 ),
             )
@@ -232,7 +239,7 @@ def run_in_executor(
     try:
         loop = asyncio.get_running_loop()
         return loop.run_in_executor(
-            executor, context_partial(func, *args, **kwargs),
+            executor, partial(func, *args, **kwargs),
         )
     except RuntimeError:
         # In case the event loop is not running right now is