Class StorageAdapter

src/storage_adapter.py:75–319

存储适配器，根据配置选择存储后端

class StorageAdapter:
    """存储适配器，根据配置选择存储后端"""

    def __init__(self):
        self._backend: Optional["StorageBackend"] = None
        self._initialized = False
        self._lock = asyncio.Lock()

    async def initialize(self) -> None:
        """初始化存储适配器"""
        async with self._lock:
            if self._initialized:
                return

            # 按优先级检查存储后端：PostgreSQL > MongoDB > SQLite
            postgresql_uri = os.getenv("POSTGRESQL_URI", "")
            mongodb_uri = os.getenv("MONGODB_URI", "")

            if postgresql_uri:
                # 使用 PostgreSQL
                try:
                    from .storage.psql_manager import PSQLManager

                    self._backend = PSQLManager()
                    await self._backend.initialize()
                    log.info("Using PostgreSQL storage backend")
                except Exception as e:
                    log.error(f"Failed to initialize PostgreSQL backend: {e}")
                    # 尝试降级到 SQLite
                    log.info("Falling back to SQLite storage backend")
                    try:
                        from .storage.sqlite_manager import SQLiteManager

                        self._backend = SQLiteManager()
                        await self._backend.initialize()
                        log.info("Using SQLite storage backend (fallback)")
                    except Exception as e2:
                        log.error(f"Failed to initialize SQLite backend: {e2}")
                        raise RuntimeError("No storage backend available") from e2
            elif not mongodb_uri:
                # 优先使用 SQLite（默认启用，无需环境变量）
                try:
                    from .storage.sqlite_manager import SQLiteManager

                    self._backend = SQLiteManager()
                    await self._backend.initialize()
                    log.info("Using SQLite storage backend")
                except Exception as e:
                    log.error(f"Failed to initialize SQLite backend: {e}")
                    raise RuntimeError("No storage backend available") from e
            else:
                # 使用 MongoDB
                try:
                    from .storage.mongodb_manager import MongoDBManager

                    self._backend = MongoDBManager()
                    await self._backend.initialize()
                    log.info("Using MongoDB storage backend")