mirror of
https://github.com/shmilylty/OneForAll.git
synced 2026-08-26 12:57:50 +08:00
resolve_callback函数添加说明并做容错处理
This commit is contained in:
@@ -59,16 +59,26 @@ async def aiodns_query_a(hostname, semaphore=None):
|
|||||||
|
|
||||||
|
|
||||||
def resolve_callback(future, index, datas):
|
def resolve_callback(future, index, datas):
|
||||||
|
"""
|
||||||
|
解析结果回调处理
|
||||||
|
:param future: future对象
|
||||||
|
:param index: 下标
|
||||||
|
:param datas: 结果集
|
||||||
|
"""
|
||||||
try:
|
try:
|
||||||
result = future.result()
|
result = future.result()
|
||||||
except aiodns.error.DNSError as e:
|
except Exception as e:
|
||||||
datas[index]['ips'] = str(e.args)
|
datas[index]['ips'] = str(e.args)
|
||||||
datas[index]['valid'] = 0
|
datas[index]['valid'] = 0
|
||||||
else:
|
else:
|
||||||
if isinstance(result, tuple):
|
if isinstance(result, tuple):
|
||||||
_, answers = result
|
_, answers = result
|
||||||
|
if answers:
|
||||||
ips = {record.host for record in answers}
|
ips = {record.host for record in answers}
|
||||||
datas[index]['ips'] = str(ips)
|
datas[index]['ips'] = str(ips)
|
||||||
|
else:
|
||||||
|
datas[index]['ips'] = 'No answers'
|
||||||
|
datas[index]['valid'] = 0
|
||||||
|
|
||||||
|
|
||||||
async def bulk_query_a(datas):
|
async def bulk_query_a(datas):
|
||||||
|
|||||||
Reference in New Issue
Block a user