使用uvloop加速子域请求

oneforall/common/request.py

@@ -227,6 +227,15 @@ async def bulk_request(data, port):
     return to_req_data + no_req_data
 
 
+def set_loop_policy():
+    try:
+        import uvloop
+    except ImportError:
+        pass
+    else:
+        asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
+
+
 def run_request(domain, data, port):
     """
     调用子域请求入口函数
@@ -238,6 +247,7 @@ def run_request(domain, data, port):
     :rtype: list
     """
     logger.log('INFOR', f'开始执行子域请求模块')
+    set_loop_policy()
     loop = asyncio.get_event_loop()
     asyncio.set_event_loop(loop)
     data = utils.set_id_none(data)

oneforall/common/utils.py

@@ -519,6 +519,11 @@ def check_pre():
         if version < '3.8':
             logger.log('FATAL', 'OneForAll在Windows系统运行时需要Python 3.8以上版本')
             exit(1)
+    if system in {"Linux", "Darwin"}:
+        try:
+            import uvloop
+        except ImportError:
+            logger.log('ALERT', f'请手动安装uvloop的Python库加速子域请求')
 
 
 def check_env():