fix ASGI app running error message set

abersheeran · Jun 26, 2024 · c3eaee7 · c3eaee7
1 parent 25f4d08
commit c3eaee7
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/a2wsgi/asgi.py b/a2wsgi/asgi.py
@@ -182,15 +182,18 @@ def asgi_done_callback(self, future: asyncio.Future) -> None:
             pass
         else:
             if exception is not None:
-                self.sync_event.set(
-                    {
-                        "type": "a2wsgi.error",
-                        "exception": (
-                            type(exception),
-                            exception,
-                            exception.__traceback__,
-                        ),
-                    }
+                task = asyncio.create_task(self.sync_event_set_lock.acquire())
+                task.add_done_callback(
+                    lambda _: self.sync_event.set(
+                        {
+                            "type": "a2wsgi.error",
+                            "exception": (
+                                type(exception),
+                                exception,
+                                exception.__traceback__,
+                            ),
+                        }
+                    )
                 )
         finally:
             self.asgi_done.set()