重构数据库连接管理，添加元数据支持

app/infra/audit/middleware.py

@@ -6,12 +6,17 @@
 
 import time
 import json
+from uuid import UUID
 from typing import Callable
 from fastapi import Request, Response
 from starlette.middleware.base import BaseHTTPMiddleware
-from app.infra.db.postgresql.database import db as default_db
 from app.core.audit import log_audit_event, AuditAction
 import logging
+from jose import JWTError, jwt
+
+from app.core.config import settings
+from app.infra.db.metadata.database import SessionLocal
+from app.infra.repositories.metadata_repository import MetadataRepository
 
 logger = logging.getLogger(__name__)
 
@@ -100,18 +105,11 @@ class AuditMiddleware(BaseHTTPMiddleware):
             return response
 
         # 4. 提取审计所需信息
-        user_id = None
-        username = None
-
-        # 尝试从请求状态获取当前用户
-        if hasattr(request.state, "user"):
-            user = request.state.user
-            user_id = getattr(user, "id", None)
-            username = getattr(user, "username", None)
+        user_id = await self._resolve_user_id(request)
+        project_id = self._resolve_project_id(request)
 
         # 获取客户端信息
         ip_address = request.client.host if request.client else None
-        user_agent = request.headers.get("user-agent")
 
         # 确定操作类型
         action = self._determine_action(request)
@@ -122,21 +120,14 @@ class AuditMiddleware(BaseHTTPMiddleware):
             await log_audit_event(
                 action=action,
                 user_id=user_id,
-                username=username,
+                project_id=project_id,
                 resource_type=resource_type,
                 resource_id=resource_id,
                 ip_address=ip_address,
-                user_agent=user_agent,
                 request_method=request.method,
                 request_path=str(request.url.path),
                 request_data=request_data,
                 response_status=response.status_code,
-                error_message=(
-                    None
-                    if response.status_code < 400
-                    else f"HTTP {response.status_code}"
-                ),
-                db=default_db,
             )
         except Exception as e:
             # 审计失败不应影响响应
@@ -148,6 +139,48 @@ class AuditMiddleware(BaseHTTPMiddleware):
 
         return response
 
+    def _resolve_project_id(self, request: Request) -> UUID | None:
+        project_header = request.headers.get("X-Project-Id")
+        if not project_header:
+            return None
+        try:
+            return UUID(project_header)
+        except ValueError:
+            return None
+
+    async def _resolve_user_id(self, request: Request) -> UUID | None:
+        auth_header = request.headers.get("authorization")
+        if not auth_header or not auth_header.lower().startswith("bearer "):
+            return None
+        token = auth_header.split(" ", 1)[1].strip()
+        if not token:
+            return None
+        try:
+            key = (
+                settings.KEYCLOAK_PUBLIC_KEY.replace("\\n", "\n")
+                if settings.KEYCLOAK_PUBLIC_KEY
+                else settings.SECRET_KEY
+            )
+            algorithms = (
+                [settings.KEYCLOAK_ALGORITHM]
+                if settings.KEYCLOAK_PUBLIC_KEY
+                else [settings.ALGORITHM]
+            )
+            payload = jwt.decode(token, key, algorithms=algorithms)
+            sub = payload.get("sub")
+            if not sub:
+                return None
+            keycloak_id = UUID(sub)
+        except (JWTError, ValueError):
+            return None
+
+        async with SessionLocal() as session:
+            repo = MetadataRepository(session)
+            user = await repo.get_user_by_keycloak_id(keycloak_id)
+            if user and user.is_active:
+                return user.id
+        return None
+
     def _determine_action(self, request: Request) -> str:
         """根据请求路径和方法确定操作类型"""
         path = request.url.path.lower()

app/infra/db/dynamic_manager.py

@@ -0,0 +1,208 @@
+import asyncio
+import logging
+from collections import OrderedDict
+from dataclasses import dataclass
+from typing import Dict
+from uuid import UUID
+
+from psycopg_pool import AsyncConnectionPool
+from sqlalchemy.engine.url import make_url
+from sqlalchemy.ext.asyncio import (
+    AsyncEngine,
+    AsyncSession,
+    async_sessionmaker,
+    create_async_engine,
+)
+
+from app.core.config import settings
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass(frozen=True)
+class PgEngineEntry:
+    engine: AsyncEngine
+    sessionmaker: async_sessionmaker[AsyncSession]
+
+
+@dataclass(frozen=True)
+class CacheKey:
+    project_id: UUID
+    db_role: str
+
+
+class ProjectConnectionManager:
+    def __init__(self) -> None:
+        self._pg_cache: Dict[CacheKey, PgEngineEntry] = OrderedDict()
+        self._ts_cache: Dict[CacheKey, AsyncConnectionPool] = OrderedDict()
+        self._pg_raw_cache: Dict[CacheKey, AsyncConnectionPool] = OrderedDict()
+        self._pg_lock = asyncio.Lock()
+        self._ts_lock = asyncio.Lock()
+        self._pg_raw_lock = asyncio.Lock()
+
+    def _normalize_pg_url(self, url: str) -> str:
+        parsed = make_url(url)
+        if parsed.drivername == "postgresql":
+            parsed = parsed.set(drivername="postgresql+psycopg")
+        return str(parsed)
+
+    async def get_pg_sessionmaker(
+        self,
+        project_id: UUID,
+        db_role: str,
+        connection_url: str,
+        pool_min_size: int,
+        pool_max_size: int,
+    ) -> async_sessionmaker[AsyncSession]:
+        async with self._pg_lock:
+            key = CacheKey(project_id=project_id, db_role=db_role)
+            entry = self._pg_cache.get(key)
+            if entry:
+                self._pg_cache.move_to_end(key)
+                return entry.sessionmaker
+
+            normalized_url = self._normalize_pg_url(connection_url)
+            pool_min_size = max(1, pool_min_size)
+            pool_max_size = max(pool_min_size, pool_max_size)
+            engine = create_async_engine(
+                normalized_url,
+                pool_size=pool_min_size,
+                max_overflow=max(0, pool_max_size - pool_min_size),
+                pool_pre_ping=True,
+            )
+            sessionmaker = async_sessionmaker(engine, expire_on_commit=False)
+            self._pg_cache[key] = PgEngineEntry(
+                engine=engine,
+                sessionmaker=sessionmaker,
+            )
+            await self._evict_pg_if_needed()
+            logger.info(
+                "Created PostgreSQL engine for project %s (%s)", project_id, db_role
+            )
+            return sessionmaker
+
+    async def get_timescale_pool(
+        self,
+        project_id: UUID,
+        db_role: str,
+        connection_url: str,
+        pool_min_size: int,
+        pool_max_size: int,
+    ) -> AsyncConnectionPool:
+        async with self._ts_lock:
+            key = CacheKey(project_id=project_id, db_role=db_role)
+            pool = self._ts_cache.get(key)
+            if pool:
+                self._ts_cache.move_to_end(key)
+                return pool
+
+            pool_min_size = max(1, pool_min_size)
+            pool_max_size = max(pool_min_size, pool_max_size)
+            pool = AsyncConnectionPool(
+                conninfo=connection_url,
+                min_size=pool_min_size,
+                max_size=pool_max_size,
+                open=False,
+            )
+            await pool.open()
+            self._ts_cache[key] = pool
+            await self._evict_ts_if_needed()
+            logger.info(
+                "Created TimescaleDB pool for project %s (%s)", project_id, db_role
+            )
+            return pool
+
+    async def get_pg_pool(
+        self,
+        project_id: UUID,
+        db_role: str,
+        connection_url: str,
+        pool_min_size: int,
+        pool_max_size: int,
+    ) -> AsyncConnectionPool:
+        async with self._pg_raw_lock:
+            key = CacheKey(project_id=project_id, db_role=db_role)
+            pool = self._pg_raw_cache.get(key)
+            if pool:
+                self._pg_raw_cache.move_to_end(key)
+                return pool
+
+            pool_min_size = max(1, pool_min_size)
+            pool_max_size = max(pool_min_size, pool_max_size)
+            pool = AsyncConnectionPool(
+                conninfo=connection_url,
+                min_size=pool_min_size,
+                max_size=pool_max_size,
+                open=False,
+            )
+            await pool.open()
+            self._pg_raw_cache[key] = pool
+            await self._evict_pg_raw_if_needed()
+            logger.info(
+                "Created PostgreSQL pool for project %s (%s)", project_id, db_role
+            )
+            return pool
+
+    async def _evict_pg_if_needed(self) -> None:
+        while len(self._pg_cache) > settings.PROJECT_PG_CACHE_SIZE:
+            key, entry = self._pg_cache.popitem(last=False)
+            await entry.engine.dispose()
+            logger.info(
+                "Evicted PostgreSQL engine for project %s (%s)",
+                key.project_id,
+                key.db_role,
+            )
+
+    async def _evict_ts_if_needed(self) -> None:
+        while len(self._ts_cache) > settings.PROJECT_TS_CACHE_SIZE:
+            key, pool = self._ts_cache.popitem(last=False)
+            await pool.close()
+            logger.info(
+                "Evicted TimescaleDB pool for project %s (%s)",
+                key.project_id,
+                key.db_role,
+            )
+
+    async def _evict_pg_raw_if_needed(self) -> None:
+        while len(self._pg_raw_cache) > settings.PROJECT_PG_CACHE_SIZE:
+            key, pool = self._pg_raw_cache.popitem(last=False)
+            await pool.close()
+            logger.info(
+                "Evicted PostgreSQL pool for project %s (%s)",
+                key.project_id,
+                key.db_role,
+            )
+
+    async def close_all(self) -> None:
+        async with self._pg_lock:
+            for key, entry in list(self._pg_cache.items()):
+                await entry.engine.dispose()
+                logger.info(
+                    "Closed PostgreSQL engine for project %s (%s)",
+                    key.project_id,
+                    key.db_role,
+                )
+            self._pg_cache.clear()
+
+        async with self._ts_lock:
+            for key, pool in list(self._ts_cache.items()):
+                await pool.close()
+                logger.info(
+                    "Closed TimescaleDB pool for project %s (%s)",
+                    key.project_id,
+                    key.db_role,
+                )
+            self._ts_cache.clear()
+
+        async with self._pg_raw_lock:
+            for key, pool in list(self._pg_raw_cache.items()):
+                await pool.close()
+                logger.info(
+                    "Closed PostgreSQL pool for project %s (%s)",
+                    key.project_id,
+                    key.db_role,
+                )
+            self._pg_raw_cache.clear()
+
+
+project_connection_manager = ProjectConnectionManager()

app/infra/db/metadata/__init__.py

@@ -0,0 +1,3 @@
+from .database import get_metadata_session, close_metadata_engine
+
+__all__ = ["get_metadata_session", "close_metadata_engine"]

app/infra/db/metadata/database.py

@@ -0,0 +1,27 @@
+import logging
+from typing import AsyncGenerator
+
+from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
+
+from app.core.config import settings
+
+logger = logging.getLogger(__name__)
+
+engine = create_async_engine(
+    settings.METADATA_DATABASE_URI,
+    pool_size=settings.METADATA_DB_POOL_SIZE,
+    max_overflow=settings.METADATA_DB_MAX_OVERFLOW,
+    pool_pre_ping=True,
+)
+
+SessionLocal = async_sessionmaker(engine, expire_on_commit=False)
+
+
+async def get_metadata_session() -> AsyncGenerator[AsyncSession, None]:
+    async with SessionLocal() as session:
+        yield session
+
+
+async def close_metadata_engine() -> None:
+    await engine.dispose()
+    logger.info("Metadata database engine disposed.")

app/infra/db/metadata/models.py

@@ -0,0 +1,115 @@
+from datetime import datetime
+from uuid import UUID
+
+from sqlalchemy import Boolean, DateTime, Integer, String, Text
+from sqlalchemy.dialects.postgresql import JSONB, UUID as PGUUID
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class User(Base):
+    __tablename__ = "users"
+
+    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True)
+    keycloak_id: Mapped[UUID] = mapped_column(
+        PGUUID(as_uuid=True), unique=True, index=True
+    )
+    username: Mapped[str] = mapped_column(String(50), unique=True)
+    email: Mapped[str] = mapped_column(String(100), unique=True)
+    role: Mapped[str] = mapped_column(String(20), default="user")
+    is_active: Mapped[bool] = mapped_column(Boolean, default=True)
+    is_superuser: Mapped[bool] = mapped_column(Boolean, default=False)
+    attributes: Mapped[dict | None] = mapped_column(JSONB, nullable=True)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=datetime.utcnow
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=datetime.utcnow
+    )
+    last_login_at: Mapped[datetime | None] = mapped_column(
+        DateTime(timezone=True), nullable=True
+    )
+
+
+class Project(Base):
+    __tablename__ = "projects"
+
+    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True)
+    name: Mapped[str] = mapped_column(String(100))
+    code: Mapped[str] = mapped_column(String(50), unique=True)
+    description: Mapped[str | None] = mapped_column(Text, nullable=True)
+    gs_workspace: Mapped[str] = mapped_column(String(100), unique=True)
+    status: Mapped[str] = mapped_column(String(20), default="active")
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=datetime.utcnow
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=datetime.utcnow
+    )
+
+
+class ProjectDatabase(Base):
+    __tablename__ = "project_databases"
+
+    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True)
+    project_id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), index=True)
+    db_role: Mapped[str] = mapped_column(String(20))
+    db_type: Mapped[str] = mapped_column(String(20))
+    dsn_encrypted: Mapped[str] = mapped_column(Text)
+    pool_min_size: Mapped[int] = mapped_column(Integer, default=2)
+    pool_max_size: Mapped[int] = mapped_column(Integer, default=10)
+
+
+class ProjectGeoServerConfig(Base):
+    __tablename__ = "project_geoserver_configs"
+
+    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True)
+    project_id: Mapped[UUID] = mapped_column(
+        PGUUID(as_uuid=True), unique=True, index=True
+    )
+    gs_base_url: Mapped[str | None] = mapped_column(Text, nullable=True)
+    gs_admin_user: Mapped[str | None] = mapped_column(String(50), nullable=True)
+    gs_admin_password_encrypted: Mapped[str | None] = mapped_column(
+        Text, nullable=True
+    )
+    gs_datastore_name: Mapped[str] = mapped_column(String(100), default="ds_postgis")
+    default_extent: Mapped[dict | None] = mapped_column(JSONB, nullable=True)
+    srid: Mapped[int] = mapped_column(Integer, default=4326)
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=datetime.utcnow
+    )
+
+
+class UserProjectMembership(Base):
+    __tablename__ = "user_project_membership"
+
+    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True)
+    user_id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), index=True)
+    project_id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), index=True)
+    project_role: Mapped[str] = mapped_column(String(20), default="viewer")
+
+
+class AuditLog(Base):
+    __tablename__ = "audit_logs"
+
+    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True)
+    user_id: Mapped[UUID | None] = mapped_column(
+        PGUUID(as_uuid=True), nullable=True, index=True
+    )
+    project_id: Mapped[UUID | None] = mapped_column(
+        PGUUID(as_uuid=True), nullable=True, index=True
+    )
+    action: Mapped[str] = mapped_column(String(50))
+    resource_type: Mapped[str | None] = mapped_column(String(50), nullable=True)
+    resource_id: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    ip_address: Mapped[str | None] = mapped_column(String(45), nullable=True)
+    request_method: Mapped[str | None] = mapped_column(String(10), nullable=True)
+    request_path: Mapped[str | None] = mapped_column(Text, nullable=True)
+    request_data: Mapped[dict | None] = mapped_column(JSONB, nullable=True)
+    response_status: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    timestamp: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=datetime.utcnow
+    )

app/infra/db/postgresql/router.py

@@ -1,24 +1,18 @@
-from fastapi import APIRouter, Depends, HTTPException, Query
-from typing import Optional
+from fastapi import APIRouter, Depends, HTTPException
 from psycopg import AsyncConnection
 
-from .database import get_database_instance
 from .scada_info import ScadaRepository
 from .scheme import SchemeRepository
+from app.auth.project_dependencies import get_project_pg_connection
 
 router = APIRouter()
 
 
-# 创建支持数据库选择的连接依赖函数
+# 动态项目 PostgreSQL 连接依赖
 async def get_database_connection(
-    db_name: Optional[str] = Query(
-        None, description="指定要连接的数据库名称，为空时使用默认数据库"
-    )
+    conn: AsyncConnection = Depends(get_project_pg_connection),
 ):
-    """获取数据库连接，支持通过查询参数指定数据库名称"""
-    instance = await get_database_instance(db_name)
-    async with instance.get_connection() as conn:
-        yield conn
+    yield conn
 
 
 @router.get("/scada-info")

app/infra/db/timescaledb/router.py

@@ -1,42 +1,32 @@
 from fastapi import APIRouter, Depends, HTTPException, Query
-from typing import List, Optional
+from typing import List
 from datetime import datetime
 from psycopg import AsyncConnection
 
-from .database import get_database_instance
 from .schemas.realtime import RealtimeRepository
 from .schemas.scheme import SchemeRepository
 from .schemas.scada import ScadaRepository
 from .composite_queries import CompositeQueries
-from app.infra.db.postgresql.database import (
-    get_database_instance as get_postgres_database_instance,
+from app.auth.project_dependencies import (
+    get_project_pg_connection,
+    get_project_timescale_connection,
 )
 
 router = APIRouter()
 
 
-# 创建支持数据库选择的连接依赖函数
+# 动态项目 TimescaleDB 连接依赖
 async def get_database_connection(
-    db_name: Optional[str] = Query(
-        None, description="指定要连接的数据库名称，为空时使用默认数据库"
-    )
+    conn: AsyncConnection = Depends(get_project_timescale_connection),
 ):
-    """获取数据库连接，支持通过查询参数指定数据库名称"""
-    instance = await get_database_instance(db_name)
-    async with instance.get_connection() as conn:
-        yield conn
+    yield conn
 
 
-# PostgreSQL 数据库连接依赖函数
+# 动态项目 PostgreSQL 连接依赖
 async def get_postgres_connection(
-    db_name: Optional[str] = Query(
-        None, description="指定要连接的 PostgreSQL 数据库名称，为空时使用默认数据库"
-    )
+    conn: AsyncConnection = Depends(get_project_pg_connection),
 ):
-    """获取 PostgreSQL 数据库连接，支持通过查询参数指定数据库名称"""
-    instance = await get_postgres_database_instance(db_name)
-    async with instance.get_connection() as conn:
-        yield conn
+    yield conn
 
 
 # --- Realtime Endpoints ---

app/infra/repositories/audit_repository.py

@@ -1,220 +1,112 @@
-from typing import Optional, List
 from datetime import datetime
-import json
-from app.infra.db.postgresql.database import Database
-from app.domain.schemas.audit import AuditLogCreate, AuditLogResponse
-import logging
+from typing import Optional, List
+from uuid import UUID
+
+from sqlalchemy import func, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.domain.schemas.audit import AuditLogResponse
+from app.infra.db.metadata import models
 
-logger = logging.getLogger(__name__)
 
 class AuditRepository:
-    """审计日志数据访问层"""
-    
-    def __init__(self, db: Database):
-        self.db = db
-    
+    """审计日志数据访问层（system_hub）"""
+
+    def __init__(self, session: AsyncSession):
+        self.session = session
+
     async def create_log(
         self,
-        user_id: Optional[int] = None,
-        username: Optional[str] = None,
-        action: str = "",
+        action: str,
+        user_id: Optional[UUID] = None,
+        project_id: Optional[UUID] = None,
         resource_type: Optional[str] = None,
         resource_id: Optional[str] = None,
         ip_address: Optional[str] = None,
-        user_agent: Optional[str] = None,
         request_method: Optional[str] = None,
         request_path: Optional[str] = None,
         request_data: Optional[dict] = None,
         response_status: Optional[int] = None,
-        error_message: Optional[str] = None
-    ) -> Optional[AuditLogResponse]:
-        """
-        创建审计日志
-        
-        Args:
-            参数说明见 AuditLogCreate
-            
-        Returns:
-            创建的审计日志对象
-        """
-        query = """
-            INSERT INTO audit_logs (
-                user_id, username, action, resource_type, resource_id,
-                ip_address, user_agent, request_method, request_path,
-                request_data, response_status, error_message
-            )
-            VALUES (
-                %(user_id)s, %(username)s, %(action)s, %(resource_type)s, %(resource_id)s,
-                %(ip_address)s, %(user_agent)s, %(request_method)s, %(request_path)s,
-                %(request_data)s, %(response_status)s, %(error_message)s
-            )
-            RETURNING id, user_id, username, action, resource_type, resource_id,
-                      ip_address, user_agent, request_method, request_path,
-                      request_data, response_status, error_message, timestamp
-        """
-        
-        try:
-            async with self.db.get_connection() as conn:
-                async with conn.cursor() as cur:
-                    await cur.execute(query, {
-                        'user_id': user_id,
-                        'username': username,
-                        'action': action,
-                        'resource_type': resource_type,
-                        'resource_id': resource_id,
-                        'ip_address': ip_address,
-                        'user_agent': user_agent,
-                        'request_method': request_method,
-                        'request_path': request_path,
-                        'request_data': json.dumps(request_data) if request_data else None,
-                        'response_status': response_status,
-                        'error_message': error_message
-                    })
-                    row = await cur.fetchone()
-                    if row:
-                        return AuditLogResponse(**row)
-        except Exception as e:
-            logger.error(f"Error creating audit log: {e}")
-            raise
-        
-        return None
-    
+    ) -> AuditLogResponse:
+        log = models.AuditLog(
+            user_id=user_id,
+            project_id=project_id,
+            action=action,
+            resource_type=resource_type,
+            resource_id=resource_id,
+            ip_address=ip_address,
+            request_method=request_method,
+            request_path=request_path,
+            request_data=request_data,
+            response_status=response_status,
+            timestamp=datetime.utcnow(),
+        )
+        self.session.add(log)
+        await self.session.commit()
+        await self.session.refresh(log)
+        return AuditLogResponse.model_validate(log)
+
     async def get_logs(
         self,
-        user_id: Optional[int] = None,
-        username: Optional[str] = None,
+        user_id: Optional[UUID] = None,
+        project_id: Optional[UUID] = None,
         action: Optional[str] = None,
         resource_type: Optional[str] = None,
         start_time: Optional[datetime] = None,
         end_time: Optional[datetime] = None,
         skip: int = 0,
-        limit: int = 100
+        limit: int = 100,
     ) -> List[AuditLogResponse]:
-        """
-        查询审计日志
-        
-        Args:
-            user_id: 用户ID过滤
-            username: 用户名过滤
-            action: 操作类型过滤
-            resource_type: 资源类型过滤
-            start_time: 开始时间
-            end_time: 结束时间
-            skip: 跳过记录数
-            limit: 限制记录数
-            
-        Returns:
-            审计日志列表
-        """
-        # 构建动态查询
         conditions = []
-        params = {'skip': skip, 'limit': limit}
-        
         if user_id is not None:
-            conditions.append("user_id = %(user_id)s")
-            params['user_id'] = user_id
-        
-        if username:
-            conditions.append("username = %(username)s")
-            params['username'] = username
-        
+            conditions.append(models.AuditLog.user_id == user_id)
+        if project_id is not None:
+            conditions.append(models.AuditLog.project_id == project_id)
         if action:
-            conditions.append("action = %(action)s")
-            params['action'] = action
-        
+            conditions.append(models.AuditLog.action == action)
         if resource_type:
-            conditions.append("resource_type = %(resource_type)s")
-            params['resource_type'] = resource_type
-        
+            conditions.append(models.AuditLog.resource_type == resource_type)
         if start_time:
-            conditions.append("timestamp >= %(start_time)s")
-            params['start_time'] = start_time
-        
+            conditions.append(models.AuditLog.timestamp >= start_time)
         if end_time:
-            conditions.append("timestamp <= %(end_time)s")
-            params['end_time'] = end_time
-        
-        where_clause = f"WHERE {' AND '.join(conditions)}" if conditions else ""
-        
-        query = f"""
-            SELECT id, user_id, username, action, resource_type, resource_id,
-                   ip_address, user_agent, request_method, request_path,
-                   request_data, response_status, error_message, timestamp
-            FROM audit_logs
-            {where_clause}
-            ORDER BY timestamp DESC
-            LIMIT %(limit)s OFFSET %(skip)s
-        """
-        
-        try:
-            async with self.db.get_connection() as conn:
-                async with conn.cursor() as cur:
-                    await cur.execute(query, params)
-                    rows = await cur.fetchall()
-                    return [AuditLogResponse(**row) for row in rows]
-        except Exception as e:
-            logger.error(f"Error querying audit logs: {e}")
-            raise
-    
+            conditions.append(models.AuditLog.timestamp <= end_time)
+
+        stmt = (
+            select(models.AuditLog)
+            .where(*conditions)
+            .order_by(models.AuditLog.timestamp.desc())
+            .offset(skip)
+            .limit(limit)
+        )
+        result = await self.session.execute(stmt)
+        return [
+            AuditLogResponse.model_validate(log)
+            for log in result.scalars().all()
+        ]
+
     async def get_log_count(
         self,
-        user_id: Optional[int] = None,
-        username: Optional[str] = None,
+        user_id: Optional[UUID] = None,
+        project_id: Optional[UUID] = None,
         action: Optional[str] = None,
         resource_type: Optional[str] = None,
         start_time: Optional[datetime] = None,
-        end_time: Optional[datetime] = None
+        end_time: Optional[datetime] = None,
     ) -> int:
-        """
-        获取审计日志数量
-        
-        Args:
-            参数同 get_logs
-            
-        Returns:
-            日志总数
-        """
         conditions = []
-        params = {}
-        
         if user_id is not None:
-            conditions.append("user_id = %(user_id)s")
-            params['user_id'] = user_id
-        
-        if username:
-            conditions.append("username = %(username)s")
-            params['username'] = username
-        
+            conditions.append(models.AuditLog.user_id == user_id)
+        if project_id is not None:
+            conditions.append(models.AuditLog.project_id == project_id)
         if action:
-            conditions.append("action = %(action)s")
-            params['action'] = action
-        
+            conditions.append(models.AuditLog.action == action)
         if resource_type:
-            conditions.append("resource_type = %(resource_type)s")
-            params['resource_type'] = resource_type
-        
+            conditions.append(models.AuditLog.resource_type == resource_type)
         if start_time:
-            conditions.append("timestamp >= %(start_time)s")
-            params['start_time'] = start_time
-        
+            conditions.append(models.AuditLog.timestamp >= start_time)
         if end_time:
-            conditions.append("timestamp <= %(end_time)s")
-            params['end_time'] = end_time
-        
-        where_clause = f"WHERE {' AND '.join(conditions)}" if conditions else ""
-        
-        query = f"""
-            SELECT COUNT(*) as count
-            FROM audit_logs
-            {where_clause}
-        """
-        
-        try:
-            async with self.db.get_connection() as conn:
-                async with conn.cursor() as cur:
-                    await cur.execute(query, params)
-                    result = await cur.fetchone()
-                    return result['count'] if result else 0
-        except Exception as e:
-            logger.error(f"Error counting audit logs: {e}")
-            return 0
+            conditions.append(models.AuditLog.timestamp <= end_time)
+
+        stmt = select(func.count()).select_from(models.AuditLog).where(*conditions)
+        result = await self.session.execute(stmt)
+        return int(result.scalar() or 0)

app/infra/repositories/metadata_repository.py

@@ -0,0 +1,164 @@
+from dataclasses import dataclass
+from typing import Optional, List
+from uuid import UUID
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.core.encryption import get_encryptor
+from app.infra.db.metadata import models
+
+
+@dataclass(frozen=True)
+class ProjectDbRouting:
+    project_id: UUID
+    db_role: str
+    db_type: str
+    dsn: str
+    pool_min_size: int
+    pool_max_size: int
+
+
+@dataclass(frozen=True)
+class ProjectGeoServerInfo:
+    project_id: UUID
+    gs_base_url: Optional[str]
+    gs_admin_user: Optional[str]
+    gs_admin_password: Optional[str]
+    gs_datastore_name: str
+    default_extent: Optional[dict]
+    srid: int
+
+
+@dataclass(frozen=True)
+class ProjectSummary:
+    project_id: UUID
+    name: str
+    code: str
+    description: Optional[str]
+    gs_workspace: str
+    status: str
+    project_role: str
+
+
+class MetadataRepository:
+    """元数据访问层（system_hub）"""
+
+    def __init__(self, session: AsyncSession):
+        self.session = session
+
+    async def get_user_by_keycloak_id(
+        self, keycloak_id: UUID
+    ) -> Optional[models.User]:
+        result = await self.session.execute(
+            select(models.User).where(models.User.keycloak_id == keycloak_id)
+        )
+        return result.scalar_one_or_none()
+
+    async def get_project_by_id(self, project_id: UUID) -> Optional[models.Project]:
+        result = await self.session.execute(
+            select(models.Project).where(models.Project.id == project_id)
+        )
+        return result.scalar_one_or_none()
+
+    async def get_membership_role(
+        self, project_id: UUID, user_id: UUID
+    ) -> Optional[str]:
+        result = await self.session.execute(
+            select(models.UserProjectMembership.project_role).where(
+                models.UserProjectMembership.project_id == project_id,
+                models.UserProjectMembership.user_id == user_id,
+            )
+        )
+        return result.scalar_one_or_none()
+
+    async def get_project_db_routing(
+        self, project_id: UUID, db_role: str
+    ) -> Optional[ProjectDbRouting]:
+        result = await self.session.execute(
+            select(models.ProjectDatabase).where(
+                models.ProjectDatabase.project_id == project_id,
+                models.ProjectDatabase.db_role == db_role,
+            )
+        )
+        record = result.scalar_one_or_none()
+        if not record:
+            return None
+        encryptor = get_encryptor()
+        dsn = encryptor.decrypt(record.dsn_encrypted)
+        return ProjectDbRouting(
+            project_id=record.project_id,
+            db_role=record.db_role,
+            db_type=record.db_type,
+            dsn=dsn,
+            pool_min_size=record.pool_min_size,
+            pool_max_size=record.pool_max_size,
+        )
+
+    async def get_geoserver_config(
+        self, project_id: UUID
+    ) -> Optional[ProjectGeoServerInfo]:
+        result = await self.session.execute(
+            select(models.ProjectGeoServerConfig).where(
+                models.ProjectGeoServerConfig.project_id == project_id
+            )
+        )
+        record = result.scalar_one_or_none()
+        if not record:
+            return None
+        encryptor = get_encryptor()
+        password = (
+            encryptor.decrypt(record.gs_admin_password_encrypted)
+            if record.gs_admin_password_encrypted
+            else None
+        )
+        return ProjectGeoServerInfo(
+            project_id=record.project_id,
+            gs_base_url=record.gs_base_url,
+            gs_admin_user=record.gs_admin_user,
+            gs_admin_password=password,
+            gs_datastore_name=record.gs_datastore_name,
+            default_extent=record.default_extent,
+            srid=record.srid,
+        )
+
+    async def list_projects_for_user(self, user_id: UUID) -> List[ProjectSummary]:
+        stmt = (
+            select(models.Project, models.UserProjectMembership.project_role)
+            .join(
+                models.UserProjectMembership,
+                models.UserProjectMembership.project_id == models.Project.id,
+            )
+            .where(models.UserProjectMembership.user_id == user_id)
+            .order_by(models.Project.name)
+        )
+        result = await self.session.execute(stmt)
+        return [
+            ProjectSummary(
+                project_id=project.id,
+                name=project.name,
+                code=project.code,
+                description=project.description,
+                gs_workspace=project.gs_workspace,
+                status=project.status,
+                project_role=role,
+            )
+            for project, role in result.all()
+        ]
+
+    async def list_all_projects(self) -> List[ProjectSummary]:
+        result = await self.session.execute(
+            select(models.Project).order_by(models.Project.name)
+        )
+        return [
+            ProjectSummary(
+                project_id=project.id,
+                name=project.name,
+                code=project.code,
+                description=project.description,
+                gs_workspace=project.gs_workspace,
+                status=project.status,
+                project_role="owner",
+            )
+            for project in result.scalars().all()
+        ]