From e4a4c45f652b82b207007e1384efd23269c934c5 Mon Sep 17 00:00:00 2001 From: nocturn9x Date: Thu, 26 Aug 2021 16:54:13 +0200 Subject: [PATCH] Updated server example with context manager to close the server socket automatically --- tests/server.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/server.py b/tests/server.py index a8f7235..71adfba 100644 --- a/tests/server.py +++ b/tests/server.py @@ -19,10 +19,11 @@ async def serve(bind_address: tuple): await sock.listen(5) logging.info(f"Serving asynchronously at {bind_address[0]}:{bind_address[1]}") async with giambio.create_pool() as pool: - while True: - conn, address_tuple = await sock.accept() - logging.info(f"{address_tuple[0]}:{address_tuple[1]} connected") - await pool.spawn(handler, conn, address_tuple) + async with sock: + while True: + conn, address_tuple = await sock.accept() + logging.info(f"{address_tuple[0]}:{address_tuple[1]} connected") + await pool.spawn(handler, conn, address_tuple) async def handler(sock: AsyncSocket, client_address: tuple):