fix: retry logic

Signed-off-by: Dup4 <lyuzhi.pan@gmail.com>
xcpcio · Apr 10, 2023 · f7bf0c4 · f7bf0c4
1 parent fda4e1b
commit f7bf0c4
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/submit-stress-test/main.py b/submit-stress-test/main.py
@@ -119,19 +119,19 @@ def submit(pid, filepath):
             try:
                 res = requests.post(
                     url=url, headers=headers, data=m, timeout=10)
+                if res.status_code < 200:
+                    break
             except Exception as e:
                 logger.error(e)
 
-            if res.status_code >= 500:
-                continue
+        try:
+            if res.status_code != 200:
+                logger.error("submit faield. [filepath={}] [status_code={}]".format(
+                    filepath, res.status_code))
             else:
-                break
-
-        if res.status_code != 200:
-            logger.error("submit faield. [filepath={}] [status_code={}]".format(
-                filepath, res.status_code))
-        else:
-            logger.info("submit success. [filepath={}]".format(filepath))
+                logger.info("submit success. [filepath={}]".format(filepath))
+        except Exception as e:
+            logger.error(e)
 
 
 def stress():