Pass contextvars to threadpool executor

abersheeran · Dec 23, 2022 · 6347403 · 6347403
1 parent f6d86c7
commit 6347403
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/a2wsgi/wsgi.py b/a2wsgi/wsgi.py
@@ -184,8 +184,10 @@ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
         sender = None
         try:
             sender = self.loop.create_task(self.sender(send))
+            context = contextvars.copy_context()
+            func = functools.partial(context.run, self.wsgi)
             await self.loop.run_in_executor(
-                self.executor, self.wsgi, environ, self.start_response
+                self.executor, func, environ, self.start_response
             )
             self.send_queue.append(None)
             self.send_event.set()