Added "addMessage" method
This commit is contained in:
parent
a6dfc299b8
commit
8d3284f261
|
|
@ -55,10 +55,31 @@ class SseStream():
|
|||
""" Method which is called regular to send out the event """
|
||||
while True:
|
||||
msg = self.queue.get(block=True, timeout=None)
|
||||
if isinstance( msg, sse ):
|
||||
if isinstance( msg, sse.Sse ):
|
||||
for data in msg:
|
||||
yield data
|
||||
|
||||
def addMessage(self, event_name, data, id = 0):
|
||||
"""
|
||||
Adds a event to the queue so it can be sent out
|
||||
---
|
||||
Parameters:
|
||||
------
|
||||
event_name: str
|
||||
Name of the event
|
||||
data: string or list of strings
|
||||
Datat to be send
|
||||
id: int
|
||||
id of the event, if set to 0 no ID will be sent """
|
||||
event = sse.Sse()
|
||||
if id > 0:
|
||||
event.set_event_id(id)
|
||||
else:
|
||||
event.reset_event_id()
|
||||
event.add_message( event_name, data )
|
||||
|
||||
self._queue.put(event)
|
||||
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue