crawlo 1.3.6__py3-none-any.whl → 1.3.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of crawlo might be problematic. Click here for more details.

Files changed (291) hide show
  1. crawlo/__init__.py +87 -87
  2. crawlo/__version__.py +1 -1
  3. crawlo/cli.py +75 -75
  4. crawlo/commands/__init__.py +14 -14
  5. crawlo/commands/check.py +594 -594
  6. crawlo/commands/genspider.py +151 -151
  7. crawlo/commands/help.py +138 -138
  8. crawlo/commands/list.py +155 -155
  9. crawlo/commands/run.py +341 -341
  10. crawlo/commands/startproject.py +436 -436
  11. crawlo/commands/stats.py +187 -187
  12. crawlo/commands/utils.py +196 -196
  13. crawlo/config.py +312 -312
  14. crawlo/config_validator.py +277 -277
  15. crawlo/core/__init__.py +45 -45
  16. crawlo/core/engine.py +439 -439
  17. crawlo/core/processor.py +40 -40
  18. crawlo/core/scheduler.py +257 -257
  19. crawlo/crawler.py +638 -638
  20. crawlo/data/__init__.py +5 -5
  21. crawlo/data/user_agents.py +194 -194
  22. crawlo/downloader/__init__.py +273 -273
  23. crawlo/downloader/aiohttp_downloader.py +228 -228
  24. crawlo/downloader/cffi_downloader.py +245 -245
  25. crawlo/downloader/httpx_downloader.py +259 -259
  26. crawlo/downloader/hybrid_downloader.py +212 -212
  27. crawlo/downloader/playwright_downloader.py +402 -402
  28. crawlo/downloader/selenium_downloader.py +472 -472
  29. crawlo/event.py +11 -11
  30. crawlo/exceptions.py +81 -81
  31. crawlo/extension/__init__.py +39 -39
  32. crawlo/extension/health_check.py +141 -141
  33. crawlo/extension/log_interval.py +57 -57
  34. crawlo/extension/log_stats.py +81 -81
  35. crawlo/extension/logging_extension.py +61 -61
  36. crawlo/extension/memory_monitor.py +104 -104
  37. crawlo/extension/performance_profiler.py +133 -133
  38. crawlo/extension/request_recorder.py +107 -107
  39. crawlo/factories/__init__.py +27 -27
  40. crawlo/factories/base.py +68 -68
  41. crawlo/factories/crawler.py +103 -103
  42. crawlo/factories/registry.py +84 -84
  43. crawlo/filters/__init__.py +154 -154
  44. crawlo/filters/aioredis_filter.py +257 -257
  45. crawlo/filters/memory_filter.py +269 -269
  46. crawlo/framework.py +292 -292
  47. crawlo/initialization/__init__.py +39 -39
  48. crawlo/initialization/built_in.py +425 -425
  49. crawlo/initialization/context.py +141 -141
  50. crawlo/initialization/core.py +193 -193
  51. crawlo/initialization/phases.py +148 -148
  52. crawlo/initialization/registry.py +145 -145
  53. crawlo/items/__init__.py +23 -23
  54. crawlo/items/base.py +23 -23
  55. crawlo/items/fields.py +52 -52
  56. crawlo/items/items.py +104 -104
  57. crawlo/logging/__init__.py +37 -37
  58. crawlo/logging/config.py +96 -96
  59. crawlo/logging/factory.py +128 -128
  60. crawlo/logging/manager.py +111 -111
  61. crawlo/middleware/__init__.py +21 -21
  62. crawlo/middleware/default_header.py +132 -132
  63. crawlo/middleware/download_delay.py +104 -104
  64. crawlo/middleware/middleware_manager.py +135 -135
  65. crawlo/middleware/offsite.py +123 -123
  66. crawlo/middleware/proxy.py +386 -386
  67. crawlo/middleware/request_ignore.py +86 -86
  68. crawlo/middleware/response_code.py +163 -163
  69. crawlo/middleware/response_filter.py +136 -136
  70. crawlo/middleware/retry.py +124 -124
  71. crawlo/middleware/simple_proxy.py +65 -65
  72. crawlo/mode_manager.py +212 -212
  73. crawlo/network/__init__.py +21 -21
  74. crawlo/network/request.py +379 -379
  75. crawlo/network/response.py +359 -359
  76. crawlo/pipelines/__init__.py +21 -21
  77. crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
  78. crawlo/pipelines/console_pipeline.py +39 -39
  79. crawlo/pipelines/csv_pipeline.py +316 -316
  80. crawlo/pipelines/database_dedup_pipeline.py +222 -222
  81. crawlo/pipelines/json_pipeline.py +218 -218
  82. crawlo/pipelines/memory_dedup_pipeline.py +115 -115
  83. crawlo/pipelines/mongo_pipeline.py +131 -131
  84. crawlo/pipelines/mysql_pipeline.py +325 -325
  85. crawlo/pipelines/pipeline_manager.py +76 -76
  86. crawlo/pipelines/redis_dedup_pipeline.py +166 -166
  87. crawlo/project.py +327 -327
  88. crawlo/queue/pqueue.py +42 -42
  89. crawlo/queue/queue_manager.py +522 -503
  90. crawlo/queue/redis_priority_queue.py +367 -326
  91. crawlo/settings/__init__.py +7 -7
  92. crawlo/settings/default_settings.py +321 -321
  93. crawlo/settings/setting_manager.py +214 -214
  94. crawlo/spider/__init__.py +657 -657
  95. crawlo/stats_collector.py +73 -73
  96. crawlo/subscriber.py +129 -129
  97. crawlo/task_manager.py +138 -138
  98. crawlo/templates/crawlo.cfg.tmpl +10 -10
  99. crawlo/templates/project/__init__.py.tmpl +3 -3
  100. crawlo/templates/project/items.py.tmpl +17 -17
  101. crawlo/templates/project/middlewares.py.tmpl +118 -118
  102. crawlo/templates/project/pipelines.py.tmpl +96 -96
  103. crawlo/templates/project/settings.py.tmpl +170 -167
  104. crawlo/templates/project/settings_distributed.py.tmpl +169 -166
  105. crawlo/templates/project/settings_gentle.py.tmpl +166 -166
  106. crawlo/templates/project/settings_high_performance.py.tmpl +167 -167
  107. crawlo/templates/project/settings_minimal.py.tmpl +65 -65
  108. crawlo/templates/project/settings_simple.py.tmpl +164 -164
  109. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  110. crawlo/templates/run.py.tmpl +34 -34
  111. crawlo/templates/spider/spider.py.tmpl +143 -143
  112. crawlo/templates/spiders_init.py.tmpl +9 -9
  113. crawlo/tools/__init__.py +200 -200
  114. crawlo/tools/anti_crawler.py +268 -268
  115. crawlo/tools/authenticated_proxy.py +240 -240
  116. crawlo/tools/data_formatter.py +225 -225
  117. crawlo/tools/data_validator.py +180 -180
  118. crawlo/tools/date_tools.py +289 -289
  119. crawlo/tools/distributed_coordinator.py +388 -388
  120. crawlo/tools/encoding_converter.py +127 -127
  121. crawlo/tools/network_diagnostic.py +364 -364
  122. crawlo/tools/request_tools.py +82 -82
  123. crawlo/tools/retry_mechanism.py +224 -224
  124. crawlo/tools/scenario_adapter.py +262 -262
  125. crawlo/tools/text_cleaner.py +232 -232
  126. crawlo/utils/__init__.py +34 -34
  127. crawlo/utils/batch_processor.py +259 -259
  128. crawlo/utils/class_loader.py +25 -25
  129. crawlo/utils/controlled_spider_mixin.py +439 -439
  130. crawlo/utils/db_helper.py +343 -343
  131. crawlo/utils/enhanced_error_handler.py +356 -356
  132. crawlo/utils/env_config.py +142 -142
  133. crawlo/utils/error_handler.py +165 -165
  134. crawlo/utils/func_tools.py +82 -82
  135. crawlo/utils/large_scale_config.py +286 -286
  136. crawlo/utils/large_scale_helper.py +344 -344
  137. crawlo/utils/log.py +79 -79
  138. crawlo/utils/performance_monitor.py +285 -285
  139. crawlo/utils/queue_helper.py +175 -175
  140. crawlo/utils/redis_connection_pool.py +388 -388
  141. crawlo/utils/redis_key_validator.py +198 -198
  142. crawlo/utils/request.py +267 -267
  143. crawlo/utils/request_serializer.py +225 -225
  144. crawlo/utils/spider_loader.py +61 -61
  145. crawlo/utils/system.py +11 -11
  146. crawlo/utils/tools.py +4 -4
  147. crawlo/utils/url.py +39 -39
  148. {crawlo-1.3.6.dist-info → crawlo-1.3.7.dist-info}/METADATA +1199 -1126
  149. crawlo-1.3.7.dist-info/RECORD +292 -0
  150. examples/__init__.py +7 -7
  151. tests/__init__.py +7 -7
  152. tests/advanced_tools_example.py +275 -275
  153. tests/authenticated_proxy_example.py +106 -106
  154. tests/baidu_performance_test.py +108 -108
  155. tests/baidu_test.py +59 -59
  156. tests/cleaners_example.py +160 -160
  157. tests/comprehensive_framework_test.py +212 -212
  158. tests/comprehensive_test.py +81 -81
  159. tests/comprehensive_testing_summary.md +186 -186
  160. tests/config_validation_demo.py +142 -142
  161. tests/controlled_spider_example.py +205 -205
  162. tests/date_tools_example.py +180 -180
  163. tests/debug_configure.py +69 -69
  164. tests/debug_framework_logger.py +84 -84
  165. tests/debug_log_config.py +126 -126
  166. tests/debug_log_levels.py +63 -63
  167. tests/debug_pipelines.py +66 -66
  168. tests/detailed_log_test.py +233 -233
  169. tests/distributed_test.py +66 -66
  170. tests/distributed_test_debug.py +76 -76
  171. tests/dynamic_loading_example.py +523 -523
  172. tests/dynamic_loading_test.py +104 -104
  173. tests/env_config_example.py +133 -133
  174. tests/error_handling_example.py +171 -171
  175. tests/final_comprehensive_test.py +151 -151
  176. tests/final_log_test.py +260 -260
  177. tests/final_validation_test.py +182 -182
  178. tests/fix_log_test.py +142 -142
  179. tests/framework_performance_test.py +202 -202
  180. tests/log_buffering_test.py +111 -111
  181. tests/log_generation_timing_test.py +153 -153
  182. tests/optimized_performance_test.py +211 -211
  183. tests/performance_comparison.py +245 -245
  184. tests/queue_blocking_test.py +113 -113
  185. tests/queue_test.py +89 -89
  186. tests/redis_key_validation_demo.py +130 -130
  187. tests/request_params_example.py +150 -150
  188. tests/response_improvements_example.py +144 -144
  189. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  190. tests/scrapy_comparison/scrapy_test.py +133 -133
  191. tests/simple_command_test.py +119 -119
  192. tests/simple_crawlo_test.py +127 -127
  193. tests/simple_log_test.py +57 -57
  194. tests/simple_log_test2.py +137 -137
  195. tests/simple_optimization_test.py +128 -128
  196. tests/simple_queue_type_test.py +41 -41
  197. tests/simple_spider_test.py +49 -49
  198. tests/simple_test.py +47 -47
  199. tests/spider_log_timing_test.py +177 -177
  200. tests/test_advanced_tools.py +148 -148
  201. tests/test_all_commands.py +230 -230
  202. tests/test_all_redis_key_configs.py +145 -145
  203. tests/test_authenticated_proxy.py +141 -141
  204. tests/test_batch_processor.py +178 -178
  205. tests/test_cleaners.py +54 -54
  206. tests/test_component_factory.py +174 -174
  207. tests/test_comprehensive.py +146 -146
  208. tests/test_config_consistency.py +80 -80
  209. tests/test_config_merge.py +152 -152
  210. tests/test_config_validator.py +182 -182
  211. tests/test_controlled_spider_mixin.py +79 -79
  212. tests/test_crawlo_proxy_integration.py +108 -108
  213. tests/test_date_tools.py +123 -123
  214. tests/test_default_header_middleware.py +158 -158
  215. tests/test_distributed.py +65 -65
  216. tests/test_double_crawlo_fix.py +204 -207
  217. tests/test_double_crawlo_fix_simple.py +124 -124
  218. tests/test_download_delay_middleware.py +221 -221
  219. tests/test_downloader_proxy_compatibility.py +268 -268
  220. tests/test_dynamic_downloaders_proxy.py +124 -124
  221. tests/test_dynamic_proxy.py +92 -92
  222. tests/test_dynamic_proxy_config.py +146 -146
  223. tests/test_dynamic_proxy_real.py +109 -109
  224. tests/test_edge_cases.py +303 -303
  225. tests/test_enhanced_error_handler.py +270 -270
  226. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  227. tests/test_env_config.py +121 -121
  228. tests/test_error_handler_compatibility.py +112 -112
  229. tests/test_factories.py +252 -252
  230. tests/test_final_validation.py +153 -153
  231. tests/test_framework_env_usage.py +103 -103
  232. tests/test_framework_logger.py +66 -66
  233. tests/test_framework_startup.py +64 -64
  234. tests/test_get_component_logger.py +83 -83
  235. tests/test_integration.py +169 -169
  236. tests/test_item_dedup_redis_key.py +122 -122
  237. tests/test_large_scale_config.py +112 -112
  238. tests/test_large_scale_helper.py +235 -235
  239. tests/test_logging_system.py +282 -282
  240. tests/test_mode_change.py +72 -72
  241. tests/test_mode_consistency.py +51 -51
  242. tests/test_offsite_middleware.py +221 -221
  243. tests/test_parsel.py +29 -29
  244. tests/test_performance.py +327 -327
  245. tests/test_performance_monitor.py +115 -115
  246. tests/test_proxy_api.py +264 -264
  247. tests/test_proxy_health_check.py +32 -32
  248. tests/test_proxy_middleware.py +121 -121
  249. tests/test_proxy_middleware_enhanced.py +216 -216
  250. tests/test_proxy_middleware_integration.py +136 -136
  251. tests/test_proxy_middleware_refactored.py +184 -184
  252. tests/test_proxy_providers.py +56 -56
  253. tests/test_proxy_stats.py +19 -19
  254. tests/test_proxy_strategies.py +59 -59
  255. tests/test_queue_empty_check.py +41 -41
  256. tests/test_queue_manager_double_crawlo.py +173 -173
  257. tests/test_queue_manager_redis_key.py +179 -176
  258. tests/test_queue_naming.py +155 -0
  259. tests/test_queue_type.py +106 -106
  260. tests/test_random_user_agent.py +72 -72
  261. tests/test_real_scenario_proxy.py +195 -195
  262. tests/test_redis_config.py +28 -28
  263. tests/test_redis_connection_pool.py +294 -294
  264. tests/test_redis_key_naming.py +181 -181
  265. tests/test_redis_key_validator.py +123 -123
  266. tests/test_redis_queue.py +224 -224
  267. tests/test_redis_queue_name_fix.py +176 -0
  268. tests/test_request_ignore_middleware.py +182 -182
  269. tests/test_request_params.py +111 -111
  270. tests/test_request_serialization.py +70 -70
  271. tests/test_response_code_middleware.py +349 -349
  272. tests/test_response_filter_middleware.py +427 -427
  273. tests/test_response_improvements.py +152 -152
  274. tests/test_retry_middleware.py +241 -241
  275. tests/test_scheduler.py +252 -252
  276. tests/test_scheduler_config_update.py +133 -133
  277. tests/test_simple_response.py +61 -61
  278. tests/test_telecom_spider_redis_key.py +205 -205
  279. tests/test_template_content.py +87 -87
  280. tests/test_template_redis_key.py +134 -134
  281. tests/test_tools.py +159 -159
  282. tests/test_user_agents.py +96 -96
  283. tests/tools_example.py +260 -260
  284. tests/untested_features_report.md +138 -138
  285. tests/verify_debug.py +51 -51
  286. tests/verify_distributed.py +117 -117
  287. tests/verify_log_fix.py +111 -111
  288. crawlo-1.3.6.dist-info/RECORD +0 -290
  289. {crawlo-1.3.6.dist-info → crawlo-1.3.7.dist-info}/WHEEL +0 -0
  290. {crawlo-1.3.6.dist-info → crawlo-1.3.7.dist-info}/entry_points.txt +0 -0
  291. {crawlo-1.3.6.dist-info → crawlo-1.3.7.dist-info}/top_level.txt +0 -0
@@ -1,326 +1,367 @@
1
- import asyncio
2
- import asyncio
3
- import pickle
4
- import time
5
- import traceback
6
- from typing import Optional, TYPE_CHECKING
7
-
8
- import redis.asyncio as aioredis
9
-
10
- # 使用 TYPE_CHECKING 避免运行时循环导入
11
- if TYPE_CHECKING:
12
- from crawlo import Request
13
-
14
- from crawlo.utils.error_handler import ErrorHandler
15
- from crawlo.utils.log import get_logger
16
- from crawlo.utils.redis_connection_pool import get_redis_pool, OptimizedRedisConnectionPool
17
- from crawlo.utils.request_serializer import RequestSerializer
18
-
19
- # 延迟初始化避免循环依赖
20
- _logger = None
21
- _error_handler = None
22
-
23
-
24
- def get_module_logger():
25
- global _logger
26
- if _logger is None:
27
- _logger = get_logger(__name__)
28
- return _logger
29
-
30
-
31
- def get_module_error_handler():
32
- global _error_handler
33
- if _error_handler is None:
34
- _error_handler = ErrorHandler(__name__)
35
- return _error_handler
36
-
37
-
38
- class RedisPriorityQueue:
39
- """
40
- 基于 Redis 的分布式异步优先级队列
41
- """
42
-
43
- def __init__(
44
- self,
45
- redis_url: str = None,
46
- queue_name: str = None, # 修改默认值为 None
47
- processing_queue: str = None, # 修改默认值为 None
48
- failed_queue: str = None, # 修改默认值为 None
49
- max_retries: int = 3,
50
- timeout: int = 300, # 任务处理超时时间(秒)
51
- max_connections: int = 10, # 连接池大小
52
- module_name: str = "default" # 添加 module_name 参数
53
- ):
54
- # 移除直接使用 os.getenv(),要求通过参数传递 redis_url
55
- if redis_url is None:
56
- # 如果没有提供 redis_url,则抛出异常,要求在 settings 中配置
57
- raise ValueError("redis_url must be provided. Configure it in settings instead of using os.getenv()")
58
-
59
- self.redis_url = redis_url
60
- self.module_name = module_name # 保存 module_name
61
-
62
- # 如果未提供 queue_name,则根据 module_name 自动生成
63
- if queue_name is None:
64
- self.queue_name = f"crawlo:{module_name}:queue:requests"
65
- else:
66
- # 保持用户提供的队列名称不变,不做修改
67
- self.queue_name = queue_name
68
-
69
- # 如果未提供 processing_queue,则根据 queue_name 自动生成
70
- if processing_queue is None:
71
- if ":queue:requests" in self.queue_name:
72
- self.processing_queue = self.queue_name.replace(":queue:requests", ":queue:processing")
73
- else:
74
- self.processing_queue = f"{self.queue_name}:processing"
75
- else:
76
- self.processing_queue = processing_queue
77
-
78
- # 如果未提供 failed_queue,则根据 queue_name 自动生成
79
- if failed_queue is None:
80
- if ":queue:requests" in self.queue_name:
81
- self.failed_queue = self.queue_name.replace(":queue:requests", ":queue:failed")
82
- else:
83
- self.failed_queue = f"{self.queue_name}:failed"
84
- else:
85
- self.failed_queue = failed_queue
86
-
87
- self.max_retries = max_retries
88
- self.timeout = timeout
89
- self.max_connections = max_connections
90
- self._redis_pool: Optional[OptimizedRedisConnectionPool] = None
91
- self._redis: Optional[aioredis.Redis] = None
92
- self._lock = asyncio.Lock() # 用于连接初始化的锁
93
- self.request_serializer = RequestSerializer() # 处理序列化
94
-
95
- async def connect(self, max_retries=3, delay=1):
96
- """异步连接 Redis,支持重试"""
97
- async with self._lock:
98
- if self._redis is not None:
99
- # 如果已经连接,测试连接是否仍然有效
100
- try:
101
- await self._redis.ping()
102
- return self._redis
103
- except Exception:
104
- # 连接失效,重新连接
105
- self._redis = None
106
-
107
- for attempt in range(max_retries):
108
- try:
109
- # 使用优化的连接池,确保 decode_responses=False 以避免编码问题
110
- self._redis_pool = get_redis_pool(
111
- self.redis_url,
112
- max_connections=self.max_connections,
113
- socket_connect_timeout=5,
114
- socket_timeout=30,
115
- health_check_interval=30,
116
- retry_on_timeout=True,
117
- decode_responses=False, # 确保不自动解码响应
118
- encoding='utf-8'
119
- )
120
-
121
- self._redis = await self._redis_pool.get_connection()
122
-
123
- # 测试连接
124
- await self._redis.ping()
125
- # 只在调试模式下输出详细连接信息
126
- # get_module_logger().debug(f"Redis 连接成功 (Module: {self.module_name})") # 注释掉重复的日志
127
- return self._redis
128
- except Exception as e:
129
- error_msg = f"Redis 连接失败 (尝试 {attempt + 1}/{max_retries}, Module: {self.module_name}): {e}"
130
- get_module_logger().warning(error_msg)
131
- get_module_logger().debug(f"详细错误信息:\n{traceback.format_exc()}")
132
- if attempt < max_retries - 1:
133
- await asyncio.sleep(delay)
134
- else:
135
- raise ConnectionError(f"无法连接 Redis (Module: {self.module_name}): {e}")
136
-
137
- async def _ensure_connection(self):
138
- """确保连接有效"""
139
- if self._redis is None:
140
- await self.connect()
141
- try:
142
- await self._redis.ping()
143
- except Exception as e:
144
- get_module_logger().warning(f"Redis 连接失效 (Module: {self.module_name}),尝试重连...: {e}")
145
- self._redis = None
146
- await self.connect()
147
-
148
- async def put(self, request, priority: int = 0) -> bool:
149
- """放入请求到队列"""
150
- try:
151
- await self._ensure_connection()
152
- score = -priority
153
- key = self._get_request_key(request)
154
-
155
- # 🔥 使用专用的序列化工具清理 Request
156
- clean_request = self.request_serializer.prepare_for_serialization(request)
157
-
158
- # 确保序列化后的数据可以被正确反序列化
159
- try:
160
- serialized = pickle.dumps(clean_request)
161
- # 验证序列化数据可以被反序列化
162
- pickle.loads(serialized)
163
- except Exception as serialize_error:
164
- get_module_logger().error(f"请求序列化验证失败 (Module: {self.module_name}): {serialize_error}")
165
- return False
166
-
167
- pipe = self._redis.pipeline()
168
- pipe.zadd(self.queue_name, {key: score})
169
- pipe.hset(f"{self.queue_name}:data", key, serialized)
170
- result = await pipe.execute()
171
-
172
- if result[0] > 0:
173
- get_module_logger().debug(f"成功入队 (Module: {self.module_name}): {request.url}") # 注释掉重复的日志
174
- return result[0] > 0
175
- except Exception as e:
176
- get_module_error_handler().handle_error(
177
- e,
178
- context=f"放入队列失败 (Module: {self.module_name})",
179
- raise_error=False
180
- )
181
- return False
182
-
183
- async def get(self, timeout: float = 5.0):
184
- """
185
- 获取请求(带超时)
186
- :param timeout: 最大等待时间(秒),避免无限轮询
187
- """
188
- try:
189
- await self._ensure_connection()
190
- start_time = asyncio.get_event_loop().time()
191
-
192
- while True:
193
- # 尝试获取任务
194
- result = await self._redis.zpopmin(self.queue_name, count=1)
195
- if result:
196
- key, score = result[0]
197
- serialized = await self._redis.hget(f"{self.queue_name}:data", key)
198
- if not serialized:
199
- continue
200
-
201
- # 移动到 processing
202
- processing_key = f"{key}:{int(time.time())}"
203
- pipe = self._redis.pipeline()
204
- pipe.zadd(self.processing_queue, {processing_key: time.time() + self.timeout})
205
- pipe.hset(f"{self.processing_queue}:data", processing_key, serialized)
206
- pipe.hdel(f"{self.queue_name}:data", key)
207
- await pipe.execute()
208
-
209
- # 更安全的反序列化方式
210
- try:
211
- # 首先尝试标准的 pickle 反序列化
212
- request = pickle.loads(serialized)
213
- return request
214
- except UnicodeDecodeError:
215
- # 如果出现编码错误,尝试使用 latin1 解码
216
- request = pickle.loads(serialized, encoding='latin1')
217
- return request
218
- except Exception as pickle_error:
219
- # 如果pickle反序列化失败,记录错误并跳过这个任务
220
- get_module_logger().error(f"无法反序列化请求数据 (Module: {self.module_name}): {pickle_error}")
221
- # 从processing队列中移除这个无效的任务
222
- await self._redis.zrem(self.processing_queue, processing_key)
223
- await self._redis.hdel(f"{self.processing_queue}:data", processing_key)
224
- # 继续尝试下一个任务
225
- continue
226
-
227
- # 检查是否超时
228
- if asyncio.get_event_loop().time() - start_time > timeout:
229
- return None
230
-
231
- # 短暂等待,避免空轮询,但减少等待时间以提高响应速度
232
- await asyncio.sleep(0.001) # 从0.01减少到0.001
233
-
234
- except Exception as e:
235
- get_module_error_handler().handle_error(
236
- e,
237
- context=f"获取队列任务失败 (Module: {self.module_name})",
238
- raise_error=False
239
- )
240
- return None
241
-
242
- async def ack(self, request: "Request"):
243
- """确认任务完成"""
244
- try:
245
- await self._ensure_connection()
246
- key = self._get_request_key(request)
247
- cursor = 0
248
- while True:
249
- cursor, keys = await self._redis.zscan(self.processing_queue, cursor, match=f"{key}:*")
250
- if keys:
251
- pipe = self._redis.pipeline()
252
- for k in keys:
253
- pipe.zrem(self.processing_queue, k)
254
- pipe.hdel(f"{self.processing_queue}:data", k)
255
- await pipe.execute()
256
- if cursor == 0:
257
- break
258
- except Exception as e:
259
- get_module_error_handler().handle_error(
260
- e,
261
- context=f"确认任务完成失败 (Module: {self.module_name})",
262
- raise_error=False
263
- )
264
-
265
- async def fail(self, request: "Request", reason: str = ""):
266
- """标记任务失败"""
267
- try:
268
- await self._ensure_connection()
269
- key = self._get_request_key(request)
270
- await self.ack(request)
271
-
272
- retry_key = f"{self.failed_queue}:retries:{key}"
273
- retries = await self._redis.incr(retry_key)
274
- await self._redis.expire(retry_key, 86400)
275
-
276
- if retries <= self.max_retries:
277
- await self.put(request, priority=request.priority + 1)
278
- get_module_logger().info(
279
- f"任务重试 [{retries}/{self.max_retries}] (Module: {self.module_name}): {request.url}")
280
- else:
281
- failed_data = {
282
- "url": request.url,
283
- "reason": reason,
284
- "retries": retries,
285
- "failed_at": time.time(),
286
- "request_pickle": pickle.dumps(request).hex(), # 可选:保存完整请求
287
- }
288
- await self._redis.lpush(self.failed_queue, pickle.dumps(failed_data))
289
- get_module_logger().error(f"任务彻底失败 [{retries}次] (Module: {self.module_name}): {request.url}")
290
- except Exception as e:
291
- get_module_error_handler().handle_error(
292
- e,
293
- context=f"标记任务失败失败 (Module: {self.module_name})",
294
- raise_error=False
295
- )
296
-
297
- def _get_request_key(self, request) -> str:
298
- """生成请求唯一键"""
299
- return f"{self.module_name}:url:{hash(request.url) & 0x7FFFFFFF}" # 确保正数
300
-
301
- async def qsize(self) -> int:
302
- """Get queue size"""
303
- try:
304
- await self._ensure_connection()
305
- return await self._redis.zcard(self.queue_name)
306
- except Exception as e:
307
- get_module_error_handler().handle_error(
308
- e,
309
- context=f"Failed to get queue size (Module: {self.module_name})",
310
- raise_error=False
311
- )
312
- return 0
313
-
314
- async def close(self):
315
- """关闭连接"""
316
- try:
317
- # 连接池会自动管理连接,这里不需要显式关闭单个连接
318
- self._redis = None
319
- self._redis_pool = None
320
- get_module_logger().debug(f"Redis 连接已释放 (Module: {self.module_name})")
321
- except Exception as e:
322
- get_module_error_handler().handle_error(
323
- e,
324
- context=f"释放 Redis 连接失败 (Module: {self.module_name})",
325
- raise_error=False
326
- )
1
+ import asyncio
2
+ import asyncio
3
+ import pickle
4
+ import time
5
+ import traceback
6
+ from typing import Optional, TYPE_CHECKING
7
+
8
+ import redis.asyncio as aioredis
9
+
10
+ # 使用 TYPE_CHECKING 避免运行时循环导入
11
+ if TYPE_CHECKING:
12
+ from crawlo import Request
13
+
14
+ from crawlo.utils.error_handler import ErrorHandler
15
+ from crawlo.utils.log import get_logger
16
+ from crawlo.utils.redis_connection_pool import get_redis_pool, OptimizedRedisConnectionPool
17
+ from crawlo.utils.request_serializer import RequestSerializer
18
+
19
+ # 延迟初始化避免循环依赖
20
+ _logger = None
21
+ _error_handler = None
22
+
23
+
24
+ def get_module_logger():
25
+ global _logger
26
+ if _logger is None:
27
+ _logger = get_logger(__name__)
28
+ return _logger
29
+
30
+
31
+ def get_module_error_handler():
32
+ global _error_handler
33
+ if _error_handler is None:
34
+ _error_handler = ErrorHandler(__name__)
35
+ return _error_handler
36
+
37
+
38
+ class RedisPriorityQueue:
39
+ """
40
+ 基于 Redis 的分布式异步优先级队列
41
+ """
42
+
43
+ def __init__(
44
+ self,
45
+ redis_url: str = None,
46
+ queue_name: str = None, # 修改默认值为 None
47
+ processing_queue: str = None, # 修改默认值为 None
48
+ failed_queue: str = None, # 修改默认值为 None
49
+ max_retries: int = 3,
50
+ timeout: int = 300, # 任务处理超时时间(秒)
51
+ max_connections: int = 10, # 连接池大小
52
+ module_name: str = "default" # 添加 module_name 参数
53
+ ):
54
+ # 移除直接使用 os.getenv(),要求通过参数传递 redis_url
55
+ if redis_url is None:
56
+ # 如果没有提供 redis_url,则抛出异常,要求在 settings 中配置
57
+ raise ValueError("redis_url must be provided. Configure it in settings instead of using os.getenv()")
58
+
59
+ self.redis_url = redis_url
60
+ self.module_name = module_name # 保存 module_name
61
+
62
+ # 如果未提供 queue_name,则根据 module_name 自动生成
63
+ if queue_name is None:
64
+ self.queue_name = f"crawlo:{module_name}:queue:requests"
65
+ else:
66
+ # 处理多重 crawlo 前缀,规范化队列名称
67
+ self.queue_name = self._normalize_queue_name(queue_name)
68
+
69
+ # 如果未提供 processing_queue,则根据 queue_name 自动生成
70
+ if processing_queue is None:
71
+ if ":queue:requests" in self.queue_name:
72
+ self.processing_queue = self.queue_name.replace(":queue:requests", ":queue:processing")
73
+ else:
74
+ self.processing_queue = f"{self.queue_name}:processing"
75
+ else:
76
+ self.processing_queue = processing_queue
77
+
78
+ # 如果未提供 failed_queue,则根据 queue_name 自动生成
79
+ if failed_queue is None:
80
+ if ":queue:requests" in self.queue_name:
81
+ self.failed_queue = self.queue_name.replace(":queue:requests", ":queue:failed")
82
+ else:
83
+ self.failed_queue = f"{self.queue_name}:failed"
84
+ else:
85
+ self.failed_queue = failed_queue
86
+
87
+ self.max_retries = max_retries
88
+ self.timeout = timeout
89
+ self.max_connections = max_connections
90
+ self._redis_pool: Optional[OptimizedRedisConnectionPool] = None
91
+ self._redis: Optional[aioredis.Redis] = None
92
+ self._lock = asyncio.Lock() # 用于连接初始化的锁
93
+ self.request_serializer = RequestSerializer() # 处理序列化
94
+
95
+ def _normalize_queue_name(self, queue_name: str) -> str:
96
+ """
97
+ 规范化队列名称,处理多重 crawlo 前缀
98
+
99
+ :param queue_name: 原始队列名称
100
+ :return: 规范化后的队列名称
101
+ """
102
+ # 如果队列名称已经符合规范(以 crawlo: 开头且不是 crawlo:crawlo:),则保持不变
103
+ if queue_name.startswith("crawlo:") and not queue_name.startswith("crawlo:crawlo:"):
104
+ return queue_name
105
+
106
+ # 处理三重 crawlo 前缀,简化为标准格式
107
+ if queue_name.startswith("crawlo:crawlo:crawlo:"):
108
+ # 三重 crawlo 前缀,简化为标准 crawlo: 格式
109
+ remaining = queue_name[21:] # 去掉 "crawlo:crawlo:crawlo:" 前缀
110
+ if remaining:
111
+ return f"crawlo:{remaining}"
112
+ else:
113
+ return "crawlo:requests" # 默认名称
114
+
115
+ # 处理双重 crawlo 前缀
116
+ elif queue_name.startswith("crawlo:crawlo:"):
117
+ # 双重 crawlo 前缀,简化为标准 crawlo: 格式
118
+ remaining = queue_name[14:] # 去掉 "crawlo:crawlo:" 前缀
119
+ if remaining:
120
+ return f"crawlo:{remaining}"
121
+ else:
122
+ return "crawlo:requests" # 默认名称
123
+
124
+ # 处理无 crawlo 前缀的情况
125
+ elif not queue_name.startswith("crawlo:"):
126
+ # crawlo 前缀,添加 crawlo: 前缀
127
+ if queue_name:
128
+ return f"crawlo:{queue_name}"
129
+ else:
130
+ return "crawlo:requests" # 默认名称
131
+
132
+ # 其他情况,保持不变
133
+ else:
134
+ return queue_name
135
+
136
+ async def connect(self, max_retries=3, delay=1):
137
+ """异步连接 Redis,支持重试"""
138
+ async with self._lock:
139
+ if self._redis is not None:
140
+ # 如果已经连接,测试连接是否仍然有效
141
+ try:
142
+ await self._redis.ping()
143
+ return self._redis
144
+ except Exception:
145
+ # 连接失效,重新连接
146
+ self._redis = None
147
+
148
+ for attempt in range(max_retries):
149
+ try:
150
+ # 使用优化的连接池,确保 decode_responses=False 以避免编码问题
151
+ self._redis_pool = get_redis_pool(
152
+ self.redis_url,
153
+ max_connections=self.max_connections,
154
+ socket_connect_timeout=5,
155
+ socket_timeout=30,
156
+ health_check_interval=30,
157
+ retry_on_timeout=True,
158
+ decode_responses=False, # 确保不自动解码响应
159
+ encoding='utf-8'
160
+ )
161
+
162
+ self._redis = await self._redis_pool.get_connection()
163
+
164
+ # 测试连接
165
+ await self._redis.ping()
166
+ # 只在调试模式下输出详细连接信息
167
+ # get_module_logger().debug(f"Redis 连接成功 (Module: {self.module_name})") # 注释掉重复的日志
168
+ return self._redis
169
+ except Exception as e:
170
+ error_msg = f"Redis 连接失败 (尝试 {attempt + 1}/{max_retries}, Module: {self.module_name}): {e}"
171
+ get_module_logger().warning(error_msg)
172
+ get_module_logger().debug(f"详细错误信息:\n{traceback.format_exc()}")
173
+ if attempt < max_retries - 1:
174
+ await asyncio.sleep(delay)
175
+ else:
176
+ raise ConnectionError(f"无法连接 Redis (Module: {self.module_name}): {e}")
177
+
178
+ async def _ensure_connection(self):
179
+ """确保连接有效"""
180
+ if self._redis is None:
181
+ await self.connect()
182
+ try:
183
+ await self._redis.ping()
184
+ except Exception as e:
185
+ get_module_logger().warning(f"Redis 连接失效 (Module: {self.module_name}),尝试重连...: {e}")
186
+ self._redis = None
187
+ await self.connect()
188
+
189
+ async def put(self, request, priority: int = 0) -> bool:
190
+ """放入请求到队列"""
191
+ try:
192
+ await self._ensure_connection()
193
+ score = -priority
194
+ key = self._get_request_key(request)
195
+
196
+ # 🔥 使用专用的序列化工具清理 Request
197
+ clean_request = self.request_serializer.prepare_for_serialization(request)
198
+
199
+ # 确保序列化后的数据可以被正确反序列化
200
+ try:
201
+ serialized = pickle.dumps(clean_request)
202
+ # 验证序列化数据可以被反序列化
203
+ pickle.loads(serialized)
204
+ except Exception as serialize_error:
205
+ get_module_logger().error(f"请求序列化验证失败 (Module: {self.module_name}): {serialize_error}")
206
+ return False
207
+
208
+ pipe = self._redis.pipeline()
209
+ pipe.zadd(self.queue_name, {key: score})
210
+ pipe.hset(f"{self.queue_name}:data", key, serialized)
211
+ result = await pipe.execute()
212
+
213
+ if result[0] > 0:
214
+ get_module_logger().debug(f"成功入队 (Module: {self.module_name}): {request.url}") # 注释掉重复的日志
215
+ return result[0] > 0
216
+ except Exception as e:
217
+ get_module_error_handler().handle_error(
218
+ e,
219
+ context=f"放入队列失败 (Module: {self.module_name})",
220
+ raise_error=False
221
+ )
222
+ return False
223
+
224
+ async def get(self, timeout: float = 5.0):
225
+ """
226
+ 获取请求(带超时)
227
+ :param timeout: 最大等待时间(秒),避免无限轮询
228
+ """
229
+ try:
230
+ await self._ensure_connection()
231
+ start_time = asyncio.get_event_loop().time()
232
+
233
+ while True:
234
+ # 尝试获取任务
235
+ result = await self._redis.zpopmin(self.queue_name, count=1)
236
+ if result:
237
+ key, score = result[0]
238
+ serialized = await self._redis.hget(f"{self.queue_name}:data", key)
239
+ if not serialized:
240
+ continue
241
+
242
+ # 移动到 processing
243
+ processing_key = f"{key}:{int(time.time())}"
244
+ pipe = self._redis.pipeline()
245
+ pipe.zadd(self.processing_queue, {processing_key: time.time() + self.timeout})
246
+ pipe.hset(f"{self.processing_queue}:data", processing_key, serialized)
247
+ pipe.hdel(f"{self.queue_name}:data", key)
248
+ await pipe.execute()
249
+
250
+ # 更安全的反序列化方式
251
+ try:
252
+ # 首先尝试标准的 pickle 反序列化
253
+ request = pickle.loads(serialized)
254
+ return request
255
+ except UnicodeDecodeError:
256
+ # 如果出现编码错误,尝试使用 latin1 解码
257
+ request = pickle.loads(serialized, encoding='latin1')
258
+ return request
259
+ except Exception as pickle_error:
260
+ # 如果pickle反序列化失败,记录错误并跳过这个任务
261
+ get_module_logger().error(f"无法反序列化请求数据 (Module: {self.module_name}): {pickle_error}")
262
+ # 从processing队列中移除这个无效的任务
263
+ await self._redis.zrem(self.processing_queue, processing_key)
264
+ await self._redis.hdel(f"{self.processing_queue}:data", processing_key)
265
+ # 继续尝试下一个任务
266
+ continue
267
+
268
+ # 检查是否超时
269
+ if asyncio.get_event_loop().time() - start_time > timeout:
270
+ return None
271
+
272
+ # 短暂等待,避免空轮询,但减少等待时间以提高响应速度
273
+ await asyncio.sleep(0.001) # 从0.01减少到0.001
274
+
275
+ except Exception as e:
276
+ get_module_error_handler().handle_error(
277
+ e,
278
+ context=f"获取队列任务失败 (Module: {self.module_name})",
279
+ raise_error=False
280
+ )
281
+ return None
282
+
283
+ async def ack(self, request: "Request"):
284
+ """确认任务完成"""
285
+ try:
286
+ await self._ensure_connection()
287
+ key = self._get_request_key(request)
288
+ cursor = 0
289
+ while True:
290
+ cursor, keys = await self._redis.zscan(self.processing_queue, cursor, match=f"{key}:*")
291
+ if keys:
292
+ pipe = self._redis.pipeline()
293
+ for k in keys:
294
+ pipe.zrem(self.processing_queue, k)
295
+ pipe.hdel(f"{self.processing_queue}:data", k)
296
+ await pipe.execute()
297
+ if cursor == 0:
298
+ break
299
+ except Exception as e:
300
+ get_module_error_handler().handle_error(
301
+ e,
302
+ context=f"确认任务完成失败 (Module: {self.module_name})",
303
+ raise_error=False
304
+ )
305
+
306
+ async def fail(self, request: "Request", reason: str = ""):
307
+ """标记任务失败"""
308
+ try:
309
+ await self._ensure_connection()
310
+ key = self._get_request_key(request)
311
+ await self.ack(request)
312
+
313
+ retry_key = f"{self.failed_queue}:retries:{key}"
314
+ retries = await self._redis.incr(retry_key)
315
+ await self._redis.expire(retry_key, 86400)
316
+
317
+ if retries <= self.max_retries:
318
+ await self.put(request, priority=request.priority + 1)
319
+ get_module_logger().info(
320
+ f"任务重试 [{retries}/{self.max_retries}] (Module: {self.module_name}): {request.url}")
321
+ else:
322
+ failed_data = {
323
+ "url": request.url,
324
+ "reason": reason,
325
+ "retries": retries,
326
+ "failed_at": time.time(),
327
+ "request_pickle": pickle.dumps(request).hex(), # 可选:保存完整请求
328
+ }
329
+ await self._redis.lpush(self.failed_queue, pickle.dumps(failed_data))
330
+ get_module_logger().error(f"任务彻底失败 [{retries}次] (Module: {self.module_name}): {request.url}")
331
+ except Exception as e:
332
+ get_module_error_handler().handle_error(
333
+ e,
334
+ context=f"标记任务失败失败 (Module: {self.module_name})",
335
+ raise_error=False
336
+ )
337
+
338
+ def _get_request_key(self, request) -> str:
339
+ """生成请求唯一键"""
340
+ return f"{self.module_name}:url:{hash(request.url) & 0x7FFFFFFF}" # 确保正数
341
+
342
+ async def qsize(self) -> int:
343
+ """Get queue size"""
344
+ try:
345
+ await self._ensure_connection()
346
+ return await self._redis.zcard(self.queue_name)
347
+ except Exception as e:
348
+ get_module_error_handler().handle_error(
349
+ e,
350
+ context=f"Failed to get queue size (Module: {self.module_name})",
351
+ raise_error=False
352
+ )
353
+ return 0
354
+
355
+ async def close(self):
356
+ """关闭连接"""
357
+ try:
358
+ # 连接池会自动管理连接,这里不需要显式关闭单个连接
359
+ self._redis = None
360
+ self._redis_pool = None
361
+ get_module_logger().debug(f"Redis 连接已释放 (Module: {self.module_name})")
362
+ except Exception as e:
363
+ get_module_error_handler().handle_error(
364
+ e,
365
+ context=f"释放 Redis 连接失败 (Module: {self.module_name})",
366
+ raise_error=False
367
+ )