Pārlūkot izejas kodu

feat:request yarn restful api set timeout

zhaohaipeng 10 mēneši atpakaļ
vecāks
revīzija
6053683098
2 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 1 1
      client/YarnClient.py
  2. 1 1
      util/convert_util.py

+ 1 - 1
client/YarnClient.py

@@ -30,7 +30,7 @@ class YarnClient(object):
 
         url = f"http://{self.cluster_ip}:8088/ws/v1/cluster/apps?{query_str}"
         print(f"请求的Yarn Restful API为: {url}")
-        response = requests.get(url)
+        response = requests.get(url, timeout=60)
         res = json.loads(response.text)
         result = []
         for app in res['apps']['app']:

+ 1 - 1
util/convert_util.py

@@ -10,4 +10,4 @@ def byte_convert(bytes: int):
         unit_index += 1
 
     # 返回结果
-    return f"{bytes:.2f} {units[unit_index]}"
+    return f"{bytes:.2f}{units[unit_index]}"