From 92dd2c487f6e29a6d70a2fc06470d6ac48fee31b Mon Sep 17 00:00:00 2001 From: Xiaoxia AI Date: Wed, 17 Jun 2026 08:35:26 +0800 Subject: [PATCH] refactor(performance): complete connection pool migration for remaining repositories - Update UserRepository to use connection pool - Update WorkspaceRepository to use connection pool - All 5 PostgreSQL repositories now use connection pool - Complete performance optimization across all database operations Phase 4 Task 41/68 fully completed --- packages/adapters/postgres/user_repository.py | 5 +++-- packages/adapters/postgres/workspace_repository.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/packages/adapters/postgres/user_repository.py b/packages/adapters/postgres/user_repository.py index 080bdaac0..9bb448e60 100644 --- a/packages/adapters/postgres/user_repository.py +++ b/packages/adapters/postgres/user_repository.py @@ -17,8 +17,9 @@ class PostgresUserRepository(UserRepository): self.connection_string = connection_string def _get_connection(self): - """获取数据库连接""" - return psycopg2.connect(self.connection_string, cursor_factory=RealDictCursor) + """获取数据库连接（使用连接池）""" + from packages.adapters.postgres.connection_pool import PooledConnection + return PooledConnection() def save(self, user: User) -> None: """保存用户""" diff --git a/packages/adapters/postgres/workspace_repository.py b/packages/adapters/postgres/workspace_repository.py index 967793d5c..f69361d94 100644 --- a/packages/adapters/postgres/workspace_repository.py +++ b/packages/adapters/postgres/workspace_repository.py @@ -16,8 +16,9 @@ class PostgresWorkspaceRepository(WorkspaceRepository): self.connection_string = connection_string def _get_connection(self): - """获取数据库连接""" - return psycopg2.connect(self.connection_string, cursor_factory=RealDictCursor) + """获取数据库连接（使用连接池）""" + from packages.adapters.postgres.connection_pool import PooledConnection + return PooledConnection() def save(self, workspace: Workspace) -> None: """保存工作空间"""