From ced0f50420388e444507585e74eb1597124482e9 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Wed, 1 Jan 2025 13:14:51 +0800 Subject: [PATCH] Add time checker --- main.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 4bf5636..30c8fe6 100644 --- a/main.py +++ b/main.py @@ -1413,8 +1413,11 @@ async def fastapi_get_network_coords(network: str) -> list[str] | None: # type: junction, reservoir, tank @app.get("/getmajornodecoords/") async def fastapi_get_major_node_coords(network: str, diameter: int) -> list[str] | None: + start_time = time.time() coords = get_major_node_coords(network, diameter) - logger.info("get_major_node_coords: %s", coords) + end_time = time.time() + logger.info("get_major_node_coords: %s, time: %s", coords, end_time - start_time) + result = [] for node_id, coord in coords.items(): result.append(f"{node_id}:{coord['type']}:{coord['x']}:{coord['y']}") @@ -1430,7 +1433,11 @@ async def fastapi_get_network_link_nodes(network: str) -> list[str] | None: # 获取直径大于800的管道 @app.get("/getmajorpipenodes/") async def fastapi_get_major_pipe_nodes(network: str, diameter: int) -> list[str] | None: - return get_major_pipe_nodes(network, diameter) + start_time = time.time() + result = get_major_pipe_nodes(network, diameter) + end_time = time.time() + logger.info("get_major_pipe_nodes: %s, time: %s", result, end_time - start_time) + return result ############################################################ # vertex 25.[VERTICES]