避免退出时，output阻塞退出

2022-06-06 17:22:57 +08:00
parent 5d2b86d8e7
commit 2444545dc1
2 changed files with 15 additions and 2 deletions
--- a/proxy/queues.py
+++ b/proxy/queues.py
@ -1,4 +1,8 @@
 from queue import SimpleQueue
-from proxy.common import MessagePayload
+from typing import TYPE_CHECKING

-MESSAGE_QUEUE: "SimpleQueue[MessagePayload]" = SimpleQueue()
+if TYPE_CHECKING:
+    from typing import Optional
+    from proxy.common import MessagePayload
+
+MESSAGE_QUEUE: "SimpleQueue[Optional[MessagePayload]]" = SimpleQueue()