OrgTJWater/TJWaterServer
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add time checker

Browse Source
This commit is contained in:
DingZQ
2025-01-01 13:14:51 +08:00
parent 6f0ad1baee
commit ced0f50420
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.py
View File

@@ -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]