summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/tools/third_party/websockets/example/deployment/haproxy/app.py
diff options
context:
space:
mode:
Diffstat (limited to 'testing/web-platform/tests/tools/third_party/websockets/example/deployment/haproxy/app.py')
-rw-r--r--testing/web-platform/tests/tools/third_party/websockets/example/deployment/haproxy/app.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/testing/web-platform/tests/tools/third_party/websockets/example/deployment/haproxy/app.py b/testing/web-platform/tests/tools/third_party/websockets/example/deployment/haproxy/app.py
new file mode 100644
index 0000000000..360479b8eb
--- /dev/null
+++ b/testing/web-platform/tests/tools/third_party/websockets/example/deployment/haproxy/app.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+
+import asyncio
+import os
+import signal
+
+import websockets
+
+
+async def echo(websocket):
+ async for message in websocket:
+ await websocket.send(message)
+
+
+async def main():
+ # 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)
+
+ async with websockets.serve(
+ echo,
+ host="localhost",
+ port=8000 + int(os.environ["SUPERVISOR_PROCESS_NAME"][-2:]),
+ ):
+ await stop
+
+
+if __name__ == "__main__":
+ asyncio.run(main())