fix: 统一存储

python_core/services/base.py

@@ -0,0 +1,220 @@
+#!/usr/bin/env python3
+"""
+服务基类
+"""
+
+from abc import ABC, abstractmethod
+from typing import Any, Dict, Optional, Callable
+from python_core.storage import StorageInterface, get_storage
+from python_core.utils.logger import logger
+
+class ServiceBase(ABC):
+    """服务基类"""
+    
+    def __init__(self, storage: Optional[StorageInterface] = None):
+        """初始化服务
+        
+        Args:
+            storage: 存储接口实例，如果为None则使用默认存储
+        """
+        self.storage = storage or get_storage(self.get_service_name())
+        logger.info(f"Service {self.get_service_name()} initialized with storage: {type(self.storage).__name__}")
+    
+    @abstractmethod
+    def get_service_name(self) -> str:
+        """获取服务名称
+        
+        Returns:
+            str: 服务名称，用作存储键前缀
+        """
+        pass
+    
+    def get_collection_name(self, collection_type: str) -> str:
+        """获取集合名称
+        
+        Args:
+            collection_type: 集合类型
+            
+        Returns:
+            str: 完整的集合名称
+        """
+        return f"{self.get_service_name()}_{collection_type}"
+    
+    def save_data(self, collection_type: str, key: str, data: Any) -> bool:
+        """保存数据
+        
+        Args:
+            collection_type: 集合类型
+            key: 数据键
+            data: 要保存的数据
+            
+        Returns:
+            bool: 保存是否成功
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.save(collection, key, data)
+    
+    def load_data(self, collection_type: str, key: str) -> Any:
+        """加载数据
+        
+        Args:
+            collection_type: 集合类型
+            key: 数据键
+            
+        Returns:
+            Any: 加载的数据，不存在时返回None
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.load(collection, key)
+    
+    def delete_data(self, collection_type: str, key: str) -> bool:
+        """删除数据
+        
+        Args:
+            collection_type: 集合类型
+            key: 数据键
+            
+        Returns:
+            bool: 删除是否成功
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.delete(collection, key)
+    
+    def exists_data(self, collection_type: str, key: str) -> bool:
+        """检查数据是否存在
+        
+        Args:
+            collection_type: 集合类型
+            key: 数据键
+            
+        Returns:
+            bool: 数据是否存在
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.exists(collection, key)
+    
+    def list_keys(self, collection_type: str, pattern: str = "*") -> list[str]:
+        """列出集合中的所有键
+        
+        Args:
+            collection_type: 集合类型
+            pattern: 键的模式匹配
+            
+        Returns:
+            List[str]: 键列表
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.list_keys(collection, pattern)
+    
+    def save_batch_data(self, collection_type: str, data: Dict[str, Any]) -> bool:
+        """批量保存数据
+        
+        Args:
+            collection_type: 集合类型
+            data: 键值对数据
+            
+        Returns:
+            bool: 保存是否成功
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.save_batch(collection, data)
+    
+    def load_batch_data(self, collection_type: str, keys: list[str]) -> Dict[str, Any]:
+        """批量加载数据
+        
+        Args:
+            collection_type: 集合类型
+            keys: 键列表
+            
+        Returns:
+            Dict[str, Any]: 键值对数据
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.load_batch(collection, keys)
+    
+    def clear_collection(self, collection_type: str) -> bool:
+        """清空集合
+        
+        Args:
+            collection_type: 集合类型
+            
+        Returns:
+            bool: 清空是否成功
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.clear_collection(collection)
+    
+    def get_collection_stats(self, collection_type: str) -> Dict[str, Any]:
+        """获取集合统计信息
+        
+        Args:
+            collection_type: 集合类型
+            
+        Returns:
+            Dict[str, Any]: 统计信息
+        """
+        collection = self.get_collection_name(collection_type)
+        return self.storage.get_stats(collection)
+    
+    def get_all_collections(self) -> list[str]:
+        """获取服务的所有集合
+        
+        Returns:
+            List[str]: 集合名称列表
+        """
+        all_collections = self.storage.get_collections()
+        service_prefix = f"{self.get_service_name()}_"
+        return [c for c in all_collections if c.startswith(service_prefix)]
+    
+    def close(self):
+        """关闭服务"""
+        if hasattr(self.storage, 'close'):
+            self.storage.close()
+        logger.info(f"Service {self.get_service_name()} closed")
+
+class ProgressServiceBase(ServiceBase):
+    """支持进度回调的服务基类"""
+    
+    def __init__(self, storage: Optional[StorageInterface] = None):
+        super().__init__(storage)
+        self._progress_callback: Optional[Callable[[str], None]] = None
+    
+    def set_progress_callback(self, callback: Optional[Callable[[str], None]]):
+        """设置进度回调函数
+        
+        Args:
+            callback: 进度回调函数，接收进度消息字符串
+        """
+        self._progress_callback = callback
+    
+    def report_progress(self, message: str):
+        """报告进度
+        
+        Args:
+            message: 进度消息
+        """
+        if self._progress_callback:
+            self._progress_callback(message)
+        logger.debug(f"Progress: {message}")
+    
+    def execute_with_progress(self, operation_name: str, operation_func: Callable, *args, **kwargs) -> Any:
+        """执行带进度报告的操作
+        
+        Args:
+            operation_name: 操作名称
+            operation_func: 操作函数
+            *args: 操作函数参数
+            **kwargs: 操作函数关键字参数
+            
+        Returns:
+            Any: 操作结果
+        """
+        self.report_progress(f"开始 {operation_name}")
+        
+        try:
+            result = operation_func(*args, **kwargs)
+            self.report_progress(f"完成 {operation_name}")
+            return result
+        except Exception as e:
+            self.report_progress(f"失败 {operation_name}: {str(e)}")
+            raise