crawlo 1.3.3__py3-none-any.whl → 1.3.4__py3-none-any.whl

crawlo/filters/aioredis_filter.py

@@ -1,234 +1,257 @@
-from typing import Optional
-import redis.asyncio as aioredis
-
-from crawlo.filters import BaseFilter
-from crawlo.utils.log import get_logger
-from crawlo.utils.request import request_fingerprint
-from crawlo.utils.redis_connection_pool import get_redis_pool
-
-
-class AioRedisFilter(BaseFilter):
-    """
-    基于Redis集合实现的异步请求去重过滤器
-    
-    支持特性:
-    - 分布式爬虫多节点共享去重数据
-    - TTL 自动过期清理机制
-    - Pipeline 批量操作优化性能
-    - 容错设计和连接池管理
-    
-    适用场景:
-    - 分布式爬虫系统
-    - 大规模数据处理
-    - 需要持久化去重的场景
-    """
-
-    def __init__(
-            self,
-            redis_key: str,
-            client: aioredis.Redis,
-            stats: dict,
-            debug: bool = False,
-            log_level: str = 'INFO',
-            cleanup_fp: bool = False,
-            ttl: Optional[int] = None
-    ):
-        """
-        初始化Redis过滤器
-        
-        :param redis_key: Redis中存储指纹的键名
-        :param client: Redis客户端实例（可以为None，稍后初始化）
-        :param stats: 统计信息存储
-        :param debug: 是否启用调试模式
-        :param log_level: 日志级别
-        :param cleanup_fp: 关闭时是否清理指纹
-        :param ttl: 指纹过期时间（秒）
-        """
-        self.logger = get_logger(self.__class__.__name__, log_level)
-        super().__init__(self.logger, stats, debug)
-
-        self.redis_key = redis_key
-        self.redis = client
-        self.cleanup_fp = cleanup_fp
-        self.ttl = ttl
-        
-        # 保存连接池引用（用于延迟初始化）
-        self._redis_pool = None
-        
-        # 性能计数器
-        self._redis_operations = 0
-        self._pipeline_operations = 0
-        
-        # 连接状态标记，避免重复尝试连接失败的Redis
-        self._connection_failed = False
-
-    @classmethod
-    def create_instance(cls, crawler) -> 'BaseFilter':
-        """从爬虫配置创建过滤器实例"""
-        redis_url = crawler.settings.get('REDIS_URL', 'redis://localhost:6379')
-        # 确保 decode_responses=False 以避免编码问题
-        decode_responses = False  # crawler.settings.get_bool('DECODE_RESPONSES', False)
-        ttl_setting = crawler.settings.get_int('REDIS_TTL')
-
-        # 处理TTL设置
-        ttl = None
-        if ttl_setting is not None:
-            ttl = max(0, int(ttl_setting)) if ttl_setting > 0 else None
-
-        try:
-            # 使用优化的连接池，确保 decode_responses=False
-            redis_pool = get_redis_pool(
-                redis_url,
-                max_connections=20,
-                socket_connect_timeout=5,
-                socket_timeout=30,
-                health_check_interval=30,
-                retry_on_timeout=True,
-                decode_responses=decode_responses,  # 确保不自动解码响应
-                encoding='utf-8'
-            )
-            
-            # 注意：这里不应该使用 await，因为 create_instance 不是异步方法
-            # 我们将在实际使用时获取连接
-            redis_client = None  # 延迟初始化
-        except Exception as e:
-            raise RuntimeError(f"Redis连接池初始化失败: {redis_url} - {str(e)}")
-
-        # 使用统一的Redis key命名规范: crawlo:{project_name}:filter:fingerprint
-        project_name = crawler.settings.get('PROJECT_NAME', 'default')
-        redis_key = f"crawlo:{project_name}:filter:fingerprint"
-
-        instance = cls(
-            redis_key=redis_key,
-            client=redis_client,
-            stats=crawler.stats,
-            cleanup_fp=crawler.settings.get_bool('CLEANUP_FP', False),
-            ttl=ttl,
-            debug=crawler.settings.get_bool('FILTER_DEBUG', False),
-            log_level=crawler.settings.get('LOG_LEVEL', 'INFO')
-        )
-        
-        # 保存连接池引用，以便在需要时获取连接
-        instance._redis_pool = redis_pool
-        return instance
-
-    async def _get_redis_client(self):
-        """获取Redis客户端实例（延迟初始化）"""
-        # 如果之前连接失败，直接返回None
-        if self._connection_failed:
-            return None
-            
-        if self.redis is None and self._redis_pool is not None:
-            try:
-                self.redis = await self._redis_pool.get_connection()
-            except Exception as e:
-                self._connection_failed = True
-                self.logger.error(f"Redis连接失败，将使用本地去重: {e}")
-                return None
-        return self.redis
-
-    async def requested(self, request) -> bool:
-        """
-        检查请求是否已存在（优化版本）
-        
-        :param request: 请求对象
-        :return: True 表示重复，False 表示新请求
-        """
-        try:
-            # 确保Redis客户端已初始化
-            redis_client = await self._get_redis_client()
-            
-            # 如果Redis不可用，返回False表示不重复（避免丢失请求）
-            if redis_client is None:
-                return False
-            
-            fp = str(request_fingerprint(request))
-            self._redis_operations += 1
-
-            # 使用 pipeline 优化性能
-            pipe = redis_client.pipeline()
-            pipe.sismember(self.redis_key, fp)
-            
-            results = await pipe.execute()
-            exists = results[0]
-            
-            self._pipeline_operations += 1
-
-            if exists:
-                if self.debug:
-                    self.logger.debug(f"发现重复请求: {fp[:20]}...")
-                return True
-
-            # 如果不存在，添加指纹并设置TTL
-            await self.add_fingerprint(fp)
-            return False
-
-        except Exception as e:
-            self.logger.error(f"请求检查失败: {getattr(request, 'url', '未知URL')} - {e}")
-            # 在网络异常时返回False，避免丢失请求
-            return False
-
-    async def add_fingerprint(self, fp: str) -> bool:
-        """
-        添加新指纹到Redis集合（优化版本）
-        
-        :param fp: 请求指纹字符串
-        :return: 是否成功添加（True 表示新添加，False 表示已存在）
-        """
-        try:
-            # 确保Redis客户端已初始化
-            redis_client = await self._get_redis_client()
-            
-            # 如果Redis不可用，返回False表示添加失败
-            if redis_client is None:
-                return False
-            
-            fp = str(fp)
-            
-            # 使用 pipeline 优化性能
-            pipe = redis_client.pipeline()
-            pipe.sadd(self.redis_key, fp)
-            
-            if self.ttl and self.ttl > 0:
-                pipe.expire(self.redis_key, self.ttl)
-            
-            results = await pipe.execute()
-            added = results[0] == 1  # sadd 返回 1 表示新添加
-            
-            self._pipeline_operations += 1
-            
-            if self.debug and added:
-                self.logger.debug(f"添加新指纹: {fp[:20]}...")
-            
-            return added
-            
-        except Exception as e:
-            self.logger.error(f"添加指纹失败: {fp[:20]}... - {e}")
-            return False
-
-    async def __contains__(self, fp: str) -> bool:
-        """
-        检查指纹是否存在于Redis集合中
-        
-        :param fp: 请求指纹字符串
-        :return: 是否存在
-        """
-        try:
-            # 确保Redis客户端已初始化
-            redis_client = await self._get_redis_client()
-            
-            # 如果Redis不可用，返回False表示不存在
-            if redis_client is None:
-                return False
-            
-            # 检查指纹是否存在
-            exists = await redis_client.sismember(self.redis_key, str(fp))
-            return exists
-        except Exception as e:
-            self.logger.error(f"检查指纹存在性失败: {fp[:20]}... - {e}")
-            # 在网络异常时返回False，避免丢失请求
-            return False
-
-
-# 为了兼容性，确保导出类
-__all__ = ['AioRedisFilter']
+from typing import Optional
+import redis.asyncio as aioredis
+
+from crawlo.filters import BaseFilter
+from crawlo.utils.log import get_logger
+from crawlo.utils.request import request_fingerprint
+from crawlo.utils.redis_connection_pool import get_redis_pool
+
+
+class AioRedisFilter(BaseFilter):
+    """
+    基于Redis集合实现的异步请求去重过滤器
+    
+    支持特性:
+    - 分布式爬虫多节点共享去重数据
+    - TTL 自动过期清理机制
+    - Pipeline 批量操作优化性能
+    - 容错设计和连接池管理
+    
+    适用场景:
+    - 分布式爬虫系统
+    - 大规模数据处理
+    - 需要持久化去重的场景
+    """
+
+    def __init__(
+            self,
+            redis_key: str,
+            client: aioredis.Redis,
+            stats: dict,
+            debug: bool = False,
+            log_level: str = 'INFO',
+            cleanup_fp: bool = False,
+            ttl: Optional[int] = None
+    ):
+        """
+        初始化Redis过滤器
+        
+        :param redis_key: Redis中存储指纹的键名
+        :param client: Redis客户端实例（可以为None，稍后初始化）
+        :param stats: 统计信息存储
+        :param debug: 是否启用调试模式
+        :param log_level: 日志级别
+        :param cleanup_fp: 关闭时是否清理指纹
+        :param ttl: 指纹过期时间（秒）
+        """
+        self.logger = get_logger(self.__class__.__name__, log_level)
+        super().__init__(self.logger, stats, debug)
+
+        self.redis_key = redis_key
+        self.redis = client
+        self.cleanup_fp = cleanup_fp
+        self.ttl = ttl
+        
+        # 保存连接池引用（用于延迟初始化）
+        self._redis_pool = None
+        
+        # 性能计数器
+        self._redis_operations = 0
+        self._pipeline_operations = 0
+        
+        # 连接状态标记，避免重复尝试连接失败的Redis
+        self._connection_failed = False
+
+    @classmethod
+    def create_instance(cls, crawler) -> 'BaseFilter':
+        """从爬虫配置创建过滤器实例"""
+        redis_url = crawler.settings.get('REDIS_URL', 'redis://localhost:6379')
+        # 确保 decode_responses=False 以避免编码问题
+        decode_responses = False  # crawler.settings.get_bool('DECODE_RESPONSES', False)
+        ttl_setting = crawler.settings.get_int('REDIS_TTL')
+
+        # 处理TTL设置
+        ttl = None
+        if ttl_setting is not None:
+            ttl = max(0, int(ttl_setting)) if ttl_setting > 0 else None
+
+        try:
+            # 使用优化的连接池，确保 decode_responses=False
+            redis_pool = get_redis_pool(
+                redis_url,
+                max_connections=20,
+                socket_connect_timeout=5,
+                socket_timeout=30,
+                health_check_interval=30,
+                retry_on_timeout=True,
+                decode_responses=decode_responses,  # 确保不自动解码响应
+                encoding='utf-8'
+            )
+            
+            # 注意：这里不应该使用 await，因为 create_instance 不是异步方法
+            # 我们将在实际使用时获取连接
+            redis_client = None  # 延迟初始化
+        except Exception as e:
+            raise RuntimeError(f"Redis连接池初始化失败: {redis_url} - {str(e)}")
+
+        # 使用统一的Redis key命名规范: crawlo:{project_name}:filter:fingerprint
+        project_name = crawler.settings.get('PROJECT_NAME', 'default')
+        redis_key = f"crawlo:{project_name}:filter:fingerprint"
+
+        instance = cls(
+            redis_key=redis_key,
+            client=redis_client,
+            stats=crawler.stats,
+            cleanup_fp=crawler.settings.get_bool('CLEANUP_FP', False),
+            ttl=ttl,
+            debug=crawler.settings.get_bool('FILTER_DEBUG', False),
+            log_level=crawler.settings.get('LOG_LEVEL', 'INFO')
+        )
+        
+        # 保存连接池引用，以便在需要时获取连接
+        instance._redis_pool = redis_pool
+        return instance
+
+    async def _get_redis_client(self):
+        """获取Redis客户端实例（延迟初始化）"""
+        # 如果之前连接失败，直接返回None
+        if self._connection_failed:
+            return None
+            
+        if self.redis is None and self._redis_pool is not None:
+            try:
+                self.redis = await self._redis_pool.get_connection()
+            except Exception as e:
+                self._connection_failed = True
+                self.logger.error(f"Redis连接失败，将使用本地去重: {e}")
+                return None
+        return self.redis
+
+    async def requested(self, request) -> bool:
+        """
+        检查请求是否已存在（优化版本）
+        
+        :param request: 请求对象
+        :return: True 表示重复，False 表示新请求
+        """
+        try:
+            # 确保Redis客户端已初始化
+            redis_client = await self._get_redis_client()
+            
+            # 如果Redis不可用，返回False表示不重复（避免丢失请求）
+            if redis_client is None:
+                return False
+            
+            fp = str(request_fingerprint(request))
+            self._redis_operations += 1
+
+            # 使用 pipeline 优化性能
+            pipe = redis_client.pipeline()
+            pipe.sismember(self.redis_key, fp)
+            
+            results = await pipe.execute()
+            exists = results[0]
+            
+            self._pipeline_operations += 1
+
+            if exists:
+                if self.debug:
+                    self.logger.debug(f"发现重复请求: {fp[:20]}...")
+                return True
+
+            # 如果不存在，添加指纹并设置TTL
+            await self.add_fingerprint(fp)
+            return False
+
+        except Exception as e:
+            self.logger.error(f"请求检查失败: {getattr(request, 'url', '未知URL')} - {e}")
+            # 在网络异常时返回False，避免丢失请求
+            return False
+
+    async def add_fingerprint(self, fp: str) -> bool:
+        """
+        添加新指纹到Redis集合（优化版本）
+        
+        :param fp: 请求指纹字符串
+        :return: 是否成功添加（True 表示新添加，False 表示已存在）
+        """
+        try:
+            # 确保Redis客户端已初始化
+            redis_client = await self._get_redis_client()
+            
+            # 如果Redis不可用，返回False表示添加失败
+            if redis_client is None:
+                return False
+            
+            fp = str(fp)
+            
+            # 使用 pipeline 优化性能
+            pipe = redis_client.pipeline()
+            pipe.sadd(self.redis_key, fp)
+            
+            if self.ttl and self.ttl > 0:
+                pipe.expire(self.redis_key, self.ttl)
+            
+            results = await pipe.execute()
+            added = results[0] == 1  # sadd 返回 1 表示新添加
+            
+            self._pipeline_operations += 1
+            
+            if self.debug and added:
+                self.logger.debug(f"添加新指纹: {fp[:20]}...")
+            
+            return added
+            
+        except Exception as e:
+            self.logger.error(f"添加指纹失败: {fp[:20]}... - {e}")
+            return False
+
+    def __contains__(self, fp: str) -> bool:
+        """
+        检查指纹是否存在于Redis集合中（同步方法）
+        
+        注意：Python的魔术方法__contains__不能是异步的，
+        所以这个方法提供同步接口，仅用于基本的存在性检查。
+        对于需要异步检查的场景，请使用 contains_async() 方法。
+        
+        :param fp: 请求指纹字符串
+        :return: 是否存在
+        """
+        # 由于__contains__不能是异步的，我们只能提供一个基本的同步检查
+        # 如果Redis客户端未初始化，返回False
+        if self.redis is None:
+            return False
+            
+        # 对于同步场景，我们无法进行真正的Redis查询
+        # 所以返回False，避免阻塞调用
+        # 真正的异步检查应该使用 contains_async() 方法
+        return False
+    
+    async def contains_async(self, fp: str) -> bool:
+        """
+        异步检查指纹是否存在于Redis集合中
+        
+        这是真正的异步检查方法，应该优先使用这个方法而不是__contains__
+        
+        :param fp: 请求指纹字符串
+        :return: 是否存在
+        """
+        try:
+            # 确保Redis客户端已初始化
+            redis_client = await self._get_redis_client()
+            
+            # 如果Redis不可用，返回False表示不存在
+            if redis_client is None:
+                return False
+            
+            # 检查指纹是否存在
+            exists = await redis_client.sismember(self.redis_key, str(fp))
+            return exists
+        except Exception as e:
+            self.logger.error(f"检查指纹存在性失败: {fp[:20]}... - {e}")
+            # 在网络异常时返回False，避免丢失请求
+            return False
+
+
+# 为了兼容性，确保导出类
+__all__ = ['AioRedisFilter']