From 1b03614823f1de54b241f3afb99c45571bc6a49e Mon Sep 17 00:00:00 2001
From: Dorian Koch <doriank@fsmpi.rwth-aachen.de>
Date: Mon, 18 Nov 2024 12:34:56 +0100
Subject: [PATCH] Better error handling
---
main.py | 14 +++++++++++---
1 file changed, 11 insertions(+), 3 deletions(-)
diff --git a/main.py b/main.py
index b47b11a..b4ff34e 100644
--- a/main.py
+++ b/main.py
@@ -55,7 +55,12 @@ while True:
# Send HTTP request and measure response time
t = random.choice(list(target.items()))
start_time = time.time()
- response = requests.get(t[1], timeout=10)
+ try:
+ response = requests.get(t[1], timeout=20)
+ error = False
+ except requests.exceptions.RequestException as e:
+ print(f"Request failed: {e}")
+ error = True
end_time = time.time()
duration = end_time - start_time
@@ -67,9 +72,12 @@ while True:
bname = j["backend"]
except:
pass'''
- cur.execute("INSERT INTO response_times (duration, backend_name, url, status_code) VALUES (%s, %s, %s, %s)", (duration, bname, t[1], response.status_code))
+ if error:
+ cur.execute("INSERT INTO response_times (duration, backend_name, url, status_code) VALUES (%s, %s, %s, %s)", (duration, bname, t[1], 0))
+ else:
+ cur.execute("INSERT INTO response_times (duration, backend_name, url, status_code) VALUES (%s, %s, %s, %s)", (duration, bname, t[1], response.status_code))
conn.commit()
- print(f"Response time: {duration:.2f} seconds", flush=True)
+ print(f"Response time: {duration:.2f} seconds {bname}", flush=True)
time.sleep(3)
\ No newline at end of file
--
GitLab