mirror of
https://github.com/RhenCloud/Cloud-Index.git
synced 2025-12-06 15:26:10 +08:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from config import Config
|
|
|
|
from .base import BaseStorage
|
|
from .github import GitHubStorage
|
|
from .r2 import R2Storage
|
|
|
|
|
|
class StorageFactory:
|
|
"""存储工厂类，根据配置创建对应的存储实例"""
|
|
|
|
_instance: Optional[BaseStorage] = None
|
|
|
|
@classmethod
|
|
def get_storage(cls) -> BaseStorage:
|
|
"""
|
|
获取存储实例（单例模式）
|
|
|
|
Returns:
|
|
BaseStorage: 存储实例
|
|
|
|
Raises:
|
|
RuntimeError: 当存储类型未配置或不支持时
|
|
"""
|
|
if cls._instance is not None:
|
|
return cls._instance
|
|
|
|
storage_type = Config.STORAGE_TYPE
|
|
|
|
if not storage_type:
|
|
raise RuntimeError("STORAGE_TYPE environment variable is not set. Supported types: r2, github")
|
|
|
|
if storage_type == "r2":
|
|
cls._instance = R2Storage()
|
|
elif storage_type == "github":
|
|
cls._instance = GitHubStorage()
|
|
else:
|
|
raise RuntimeError(f"Unsupported storage type: {storage_type}. Supported types: r2, github")
|
|
|
|
return cls._instance
|
|
|
|
return cls._instance
|
|
|
|
@classmethod
|
|
def reset(cls):
|
|
"""重置单例实例（主要用于测试）"""
|
|
cls._instance = None
|