添加 row_factory 参数以支持字典行返回

This commit is contained in:
2026-02-24 17:02:48 +08:00
parent 2826999ddc
commit 1b1b0a3697

View File

@@ -6,6 +6,7 @@ from typing import Dict
from uuid import UUID from uuid import UUID
from psycopg_pool import AsyncConnectionPool from psycopg_pool import AsyncConnectionPool
from psycopg.rows import dict_row
from sqlalchemy.engine.url import make_url from sqlalchemy.engine.url import make_url
from sqlalchemy.ext.asyncio import ( from sqlalchemy.ext.asyncio import (
AsyncEngine, AsyncEngine,
@@ -103,6 +104,7 @@ class ProjectConnectionManager:
min_size=pool_min_size, min_size=pool_min_size,
max_size=pool_max_size, max_size=pool_max_size,
open=False, open=False,
kwargs={"row_factory": dict_row},
) )
await pool.open() await pool.open()
self._ts_cache[key] = pool self._ts_cache[key] = pool
@@ -134,6 +136,7 @@ class ProjectConnectionManager:
min_size=pool_min_size, min_size=pool_min_size,
max_size=pool_max_size, max_size=pool_max_size,
open=False, open=False,
kwargs={"row_factory": dict_row},
) )
await pool.open() await pool.open()
self._pg_raw_cache[key] = pool self._pg_raw_cache[key] = pool