support message sse

This commit is contained in:
jxxghp
2023-07-04 20:14:31 +08:00
parent 609c820584
commit 01c88c7dc8
2 changed files with 50 additions and 2 deletions

18
app/helper/message.py Normal file
View File

@ -0,0 +1,18 @@
import queue
class MessageHelper:
"""
消息队列管理器
"""
def __init__(self):
self.queue = queue.Queue()
def put(self, message: str):
self.queue.put(message)
def get(self):
if not self.queue.empty():
return self.queue.get(block=True)
else:
return None