Deploy to Koyeb.

Author: aaugustin

Procfile

@@ -0,0 +1 @@
+web: python app.py

app.py

@@ -1,8 +1,11 @@
 #!/usr/bin/env python
 
 import asyncio
+import http
 import json
+import os
 import secrets
+import signal
 
 from websockets.asyncio.server import broadcast, serve
 
@@ -181,9 +184,20 @@ async def handler(websocket):
         await start(websocket)
 
 
+def health_check(connection, request):
+    if request.path == "/healthz":
+        return connection.respond(http.HTTPStatus.OK, "OK\n")
+
+
 async def main():
-    async with serve(handler, "", 8001):
-        await asyncio.get_running_loop().create_future()  # run forever
+    # Set the stop condition when receiving SIGTERM.
+    loop = asyncio.get_running_loop()
+    stop = loop.create_future()
+    loop.add_signal_handler(signal.SIGTERM, stop.set_result, None)
+
+    port = int(os.environ.get("PORT", "8001"))
+    async with serve(handler, "", port, process_request=health_check):
+        await stop
 
 
 if __name__ == "__main__":

requirements.txt

@@ -0,0 +1 @@
+websockets