浏览代码

add batch

Germey 8 年之前
父节点
当前提交
603e65eff8
共有 2 个文件被更改,包括 5 次插入3 次删除
  1. 3 0
      proxypool/setting.py
  2. 2 3
      proxypool/tester.py

+ 3 - 0
proxypool/setting.py

@@ -35,3 +35,6 @@ API_PORT = 5555
 TESTER_ENABLED = True
 GETTER_ENABLED = True
 API_ENABLED = True
+
+# 最大批测试量
+BATCH_TEST_SIZE = 100

+ 2 - 3
proxypool/tester.py

@@ -47,9 +47,8 @@ class Tester(object):
         try:
             proxies = self.redis.all()
             loop = asyncio.get_event_loop()
-            batch_size = 100
-            for i in range(0, len(proxies), batch_size):
-                test_proxies = proxies[i:i + batch_size]
+            for i in range(0, len(proxies), BATCH_TEST_SIZE):
+                test_proxies = proxies[i:i + BATCH_TEST_SIZE]
                 tasks = [self.test_single_proxy(proxy) for proxy in test_proxies]
                 loop.run_until_complete(asyncio.wait(tasks))
                 time.sleep(5)