refactor(storage): route project DSNs and remove legacy backends

app/native/wndb/connection.py

@@ -4,9 +4,10 @@ from threading import RLock
 
 import psycopg as pg
 
-from app.core.config import get_pgconn_string
+from app.infra.db.project_routing import get_project_pgconn_string
 
 g_conn_dict: dict[str, pg.Connection] = {}
+g_conninfo_dict: dict[str, str] = {}
 _registry_lock = RLock()
 _project_locks: dict[str, RLock] = {}
 
@@ -42,14 +43,18 @@ def _get_project_lock(name: str) -> RLock:
 
 def open_connection(name: str) -> pg.Connection:
     with _get_project_lock(name):
+        conninfo = get_project_pgconn_string(db_name=name)
         connection = g_conn_dict.get(name)
-        if connection is None or not _is_healthy(connection):
+        if (
+            connection is None
+            or g_conninfo_dict.get(name) != conninfo
+            or not _is_healthy(connection)
+        ):
             if connection is not None:
                 _close_connection(connection)
-            connection = pg.connect(
-                conninfo=get_pgconn_string(db_name=name), autocommit=True
-            )
+            connection = pg.connect(conninfo=conninfo, autocommit=True)
             g_conn_dict[name] = connection
+            g_conninfo_dict[name] = conninfo
         return connection
 
 
@@ -60,6 +65,12 @@ def is_connection_open(name: str) -> bool:
             return False
         if not _is_healthy(connection):
             del g_conn_dict[name]
+            g_conninfo_dict.pop(name, None)
+            _close_connection(connection)
+            return False
+        if g_conninfo_dict.get(name) != get_project_pgconn_string(db_name=name):
+            del g_conn_dict[name]
+            g_conninfo_dict.pop(name, None)
             _close_connection(connection)
             return False
         return True
@@ -68,6 +79,7 @@ def is_connection_open(name: str) -> bool:
 def close_connection(name: str) -> None:
     with _get_project_lock(name):
         connection = g_conn_dict.pop(name, None)
+        g_conninfo_dict.pop(name, None)
         if connection is not None:
             _close_connection(connection)
 

app/native/wndb/project.py

@@ -7,7 +7,8 @@ from .connection import (
     is_connection_open,
     open_connection,
 )
-from app.core.config import get_pgconn_string, get_pg_config, get_pg_password
+from app.core.config import get_pg_config, get_pg_password
+from app.infra.db.project_routing import get_project_pgconn_string
 
 # no undo/redo
 
@@ -16,7 +17,7 @@ _server_databases = ["template0", "template1", "postgres", "project"]
 
 def list_project() -> list[str]:
     ps = []
-    with pg.connect(conninfo=get_pgconn_string(), autocommit=True) as conn:
+    with pg.connect(conninfo=get_project_pgconn_string(), autocommit=True) as conn:
         with conn.cursor(row_factory=dict_row) as cur:
             for p in cur.execute(
                 f"select datname from pg_database where datname <> 'postgres' and datname <> 'template0' and datname <> 'template1' and datname <> 'project'"
@@ -27,7 +28,7 @@ def list_project() -> list[str]:
 
 def have_project(name: str) -> bool:
     with pg.connect(
-        conninfo=get_pgconn_string(db_name="postgres"), autocommit=True
+        conninfo=get_project_pgconn_string(db_name="postgres"), autocommit=True
     ) as conn:
         with conn.cursor() as cur:
             cur.execute("select 1 from pg_database where datname = %s", (name,))
@@ -38,7 +39,7 @@ def copy_project(source: str, new: str) -> None:
     close_connection(source)
 
     with pg.connect(
-        conninfo=get_pgconn_string(db_name="postgres"), autocommit=True
+        conninfo=get_project_pgconn_string(db_name="postgres"), autocommit=True
     ) as admin_conn:
         with admin_conn.cursor() as cur:
             cur.execute(
@@ -131,7 +132,9 @@ class CopyProjectEx:
         connection.commit()
 
     def __call__(self, source: str, new_db: str, excluded_tables: [str] = None) -> None:
-        source_connection = pg.connect(conninfo=get_pgconn_string(), autocommit=True)
+        source_connection = pg.connect(
+            conninfo=get_project_pgconn_string(), autocommit=True
+        )
 
         self.create_database(source_connection, new_db)
 
@@ -140,7 +143,7 @@ class CopyProjectEx:
         source_connection.close()
 
         new_db_connection = pg.connect(
-            conninfo=get_pgconn_string(db_name=new_db), autocommit=True
+            conninfo=get_project_pgconn_string(db_name=new_db), autocommit=True
         )
         self.init_operation_table(new_db_connection, excluded_tables)
         new_db_connection.close()
@@ -151,7 +154,7 @@ def create_project(name: str) -> None:
 
 
 def delete_project(name: str) -> None:
-    with pg.connect(conninfo=get_pgconn_string(), autocommit=True) as conn:
+    with pg.connect(conninfo=get_project_pgconn_string(), autocommit=True) as conn:
         with conn.cursor() as cur:
             cur.execute(
                 f"select pg_terminate_backend(pid) from pg_stat_activity where datname = '{name}'"
@@ -161,7 +164,7 @@ def delete_project(name: str) -> None:
 
 def clean_project(excluded: list[str] = []) -> None:
     projects = list_project()
-    with pg.connect(conninfo=get_pgconn_string(), autocommit=True) as conn:
+    with pg.connect(conninfo=get_project_pgconn_string(), autocommit=True) as conn:
         with conn.cursor(row_factory=dict_row) as cur:
             row = cur.execute(f"select current_database()").fetchone()
             if row != None: