add asyncio_webockets benchmark

Author: kumaraditya303

pyperformance/data-files/benchmarks/MANIFEST

@@ -21,6 +21,7 @@ async_tree_eager_io_tg	<local:async_tree>
 async_tree_eager_memoization_tg	<local:async_tree>
 asyncio_tcp	<local>
 asyncio_tcp_ssl	<local:asyncio_tcp>
+asyncio_websockets	<local>
 concurrent_imap	<local>
 coroutines	<local>
 coverage	<local>

pyperformance/data-files/benchmarks/bm_asyncio_websockets/pyproject.toml

@@ -0,0 +1,9 @@
+[project]
+name = "pyperformance_bm_asyncio_websockets"
+requires-python = ">=3.8"
+dependencies = ["pyperf", "websockets"]
+urls = {repository = "https://github.com/python/pyperformance"}
+dynamic = ["version"]
+
+[tool.pyperformance]
+name = "asyncio_websockets"

pyperformance/data-files/benchmarks/bm_asyncio_websockets/requirements.txt

@@ -0,0 +1 @@
+websockets==11.0.3

pyperformance/data-files/benchmarks/bm_asyncio_websockets/run_benchmark.py

@@ -0,0 +1,39 @@
+"""
+Benchmark for asyncio websocket server and client performance
+transferring 10MB of data.
+"""
+
+import pyperf
+import websockets.server
+import websockets.client
+import asyncio
+
+CHUNK_SIZE = 1024
+DATA = b"x" * CHUNK_SIZE
+
+stop: asyncio.Event
+
+
+async def handler(websocket) -> None:
+    for _ in range(100):
+        data = await websocket.recv()
+        assert data == DATA
+
+    stop.set()
+
+
+async def main() -> None:
+    global stop
+    t0 = pyperf.perf_counter()
+    stop = asyncio.Event()
+    async with websockets.server.serve(handler, "", 8001):
+        async with websockets.client.connect("ws://localhost:8001") as ws:
+            await asyncio.gather(*[ws.send(DATA) for _ in range(100)])
+        await stop.wait()
+    return pyperf.perf_counter() - t0
+
+
+if __name__ == "__main__":
+    runner = pyperf.Runner()
+    runner.metadata['description'] = "Benchmark asyncio websockets"
+    runner.bench_async_func('asyncio_websockets', main)