SSE plugin for Starlette
Implements the Server-Sent Events specification.
Background: https://sysid.github.io/server-sent-events/
Installation:
pip install sse-starlette
Usage:
import asyncio
import uvicorn
from starlette.applications import Starlette
from starlette.routing import Route
from sse_starlette.sse import EventSourceResponse
async def numbers(minimum, maximum):
for i in range(minimum, maximum + 1):
await asyncio.sleep(0.9)
yield dict(data=i)
async def sse(request):
generator = numbers(1, 5)
return EventSourceResponse(generator)
routes = [
Route("/", endpoint=sse)
]
app = Starlette(debug=True, routes=routes)
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000, log_level='info')
Output:
Caveat: SSE streaming does not work in combination with GZipMiddleware.
Be aware that for proper server shutdown your application must stop all
running tasks (generators). Otherwise you might experience the following warnings
at shutdown: Waiting for background tasks to complete. (CTRL+C to force quit)
.
Client disconnects need to be handled in your Request handler (see example.py):
async def endless(req: Request):
async def event_publisher():
i = 0
try:
while True:
i += 1
yield dict(data=i)
await asyncio.sleep(0.2)
except asyncio.CancelledError as e:
_log.info(f"Disconnected from client (via refresh/close) {req.client}")
# Do any other cleanup, if any
raise e
return EventSourceResponse(event_publisher())
By default, the server sends a ping every 15 seconds. You can customize this by:
ping
parameterping
event to a comment event so that it is not visible to the client@router.get("")
async def handle():
generator = numbers(1, 100)
return EventSourceResponse(
generator,
headers={"Server": "nini"},
ping=5,
ping_message_factory=lambda: ServerSentEvent(**{"comment": "You can't see\r\nthis ping"}),
)
Fan out proxies usually rely on response being cacheable. To support that, you can set the value of Cache-Control
.
For example:
return EventSourceResponse(
generator(), headers={"Cache-Control": "public, max-age=29"}
)
See example: examples/error_handling.py
Async generators can expose tricky error and cleanup behavior especially when they are interrupted.
Background: Cleanup in async generators.
Example no_async_generators.py
shows an alternative implementation
that does not rely on async generators but instead uses memory channels (examples/no_async_generators.py
).
pip install pipenv
pipenv install --dev -e .
pipenv run pytest
pipenv shell
Makefile
for available commands and development support, e.g. run the unit tests:make test
For integration testing you can use the provided examples in tests
and examples
.
If you are using Postman, please see: https://github.com/sysid/sse-starlette/issues/47#issuecomment-1445953826