fix(db): validate cached project connections

2026-07-21 11:35:18 +08:00
parent 9f225374de
commit 50d823ca58
7 changed files with 318 additions and 105 deletions
@@ -1,5 +1,7 @@
 from .database import *
+from .connection import project_connection
 from .s0_base import get_link_nodes
+from psycopg.rows import dict_row

 def sql_update_coord(node: str, x: float, y: float) -> str:
    coord = f"st_geomfromtext('point({x} {y})')"
@@ -49,10 +51,11 @@ def get_links_in_extent(name: str, x1: float, y1: float, x2: float, y2: float) -
    node_ids = set([s.split(':')[0] for s in get_nodes_in_extent(name, x1, y1, x2, y2)])

    all_link_ids = []
-    with conn[name].cursor(row_factory=dict_row) as cur:
-        cur.execute(f"select id from pipes")
-        for record in cur:
-            all_link_ids.append(record['id'])
+    with project_connection(name) as conn:
+        with conn.cursor(row_factory=dict_row) as cur:
+            cur.execute(f"select id from pipes")
+            for record in cur:
+                all_link_ids.append(record['id'])

    links = []
    for link_id in all_link_ids: