birbnetes/benchmarks
birbnetes/benchmarks
4
0
Fork 0
Code Pull Requests Releases Activity

Added interrupt handler to inflight wait
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell 2021-07-26 16:06:31 +02:00
parent c430232df0
commit 423cb1b442
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
benchmark2.py
View File

@ -337,10 +337,13 @@ def run_benchmark(
total_successful_uploads = len([req for req in results.values() if req.upload_status_code == 200]) total_successful_uploads = len([req for req in results.values() if req.upload_status_code == 200])
# Wait for inflight messages for a little # Wait for inflight messages for a little
try:
for _ in range(int(inflight_timeout)): for _ in range(int(inflight_timeout)):
time.sleep(1) time.sleep(1)
if returned_queue.qsize() >= total_successful_uploads: if returned_queue.qsize() >= total_successful_uploads:
break break
except KeyboardInterrupt:
print("jovan akkor nem várom meg...")
mqtt_client.disconnect() mqtt_client.disconnect()
mqtt_client.loop_stop() # This stops further recieving MQTT messages mqtt_client.loop_stop() # This stops further recieving MQTT messages