From 25e8541964aedb21812294a9c933c20666f9de24 Mon Sep 17 00:00:00 2001
From: shmilylty <admin@hackfun.org>
Date: Sun, 18 Aug 2019 19:31:22 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=BF=9B=E5=BA=A6=E6=9D=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 oneforall/common/resolve.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/oneforall/common/resolve.py b/oneforall/common/resolve.py
index 6fb413e..4589645 100644
--- a/oneforall/common/resolve.py
+++ b/oneforall/common/resolve.py
@@ -4,6 +4,8 @@ import functools
 
 import dns.resolver
 import aiodns
+import tqdm
+
 import config
 from common import utils
 from config import logger
@@ -99,6 +101,12 @@ async def bulk_query_a(datas):
                                                      datas=datas))  # 回调
             tasks.append(task)
     if tasks:  # 任务列表里有任务不空时才进行解析
-        await asyncio.wait(tasks)  # 等待所有task完成
+        futures = asyncio.as_completed(tasks)
+        for future in tqdm.tqdm(futures, total=len(tasks)):
+            try:
+                await future
+            except:
+                pass
+        # await asyncio.wait(tasks)  # 等待所有task完成
     logger.log('INFOR', '完成异步查询子域的A记录')
     return datas