basic streaming test

Author: coilysiren

.gitignore

@@ -1,3 +1,4 @@
+bunny.webm
 .build
 Pulumi.*.yaml
 

Makefile

@@ -50,6 +50,7 @@ build-native: .build
 build-docker: .build .build-docker
 
 ## retrieve the bunny video from the server
+# via https://en.wikipedia.org/wiki/Big_Buck_Bunny
 retrieve-bunny:
 	scp kai@kai-server:/home/kai/bunny.webm .
 

src/main.py

@@ -10,7 +10,7 @@
 import structlog
 import structlog.processors
 
-from . import application, bsky, cache, worker
+from . import application, bsky, cache, worker, streaming
 
 dotenv.load_dotenv()
 (app, limiter) = application.init()
@@ -45,6 +45,18 @@ async def cache_clear(request: fastapi.Request, suffix: str):
     return {"status": "ok"}
 
 
+@app.get("/streaming")
+@app.get("/streaming/")
+async def streaming_test():
+    return fastapi.responses.StreamingResponse(streaming.testing(), media_type="text/plain")
+
+
+# @app.get("/video")
+# @app.get("/video/")
+# async def video():
+#     return fastapi.responses.FileResponse("bunny.webm")
+
+
 @app.get("/bsky/{handle}/followers")
 @app.get("/bsky/{handle}/followers/")
 @limiter.limit("10/second")

src/streaming.py

@@ -0,0 +1,8 @@
+import asyncio
+import typing
+
+
+async def testing() -> typing.AsyncGenerator[str, None]:
+    for i in range(100):
+        await asyncio.sleep(1)
+        yield f"Hello, world! {i}\n"