Skip to content
Snippets Groups Projects
Verified Commit 1b036148 authored by Dorian Koch's avatar Dorian Koch
Browse files

Better error handling

parent 3a4678dc
No related branches found
No related tags found
No related merge requests found
...@@ -55,7 +55,12 @@ while True: ...@@ -55,7 +55,12 @@ while True:
# Send HTTP request and measure response time # Send HTTP request and measure response time
t = random.choice(list(target.items())) t = random.choice(list(target.items()))
start_time = time.time() 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() end_time = time.time()
duration = end_time - start_time duration = end_time - start_time
...@@ -67,9 +72,12 @@ while True: ...@@ -67,9 +72,12 @@ while True:
bname = j["backend"] bname = j["backend"]
except: except:
pass''' pass'''
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)) 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() conn.commit()
print(f"Response time: {duration:.2f} seconds", flush=True) print(f"Response time: {duration:.2f} seconds {bname}", flush=True)
time.sleep(3) time.sleep(3)
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment