From b39ba8c9dd3064e115d47640da76e2f5aaf92a9a Mon Sep 17 00:00:00 2001 From: Grzegorz Banasiak Date: Wed, 26 Jun 2024 17:29:06 +0200 Subject: [PATCH] Restore time measurement on request end --- esrally/client/factory.py | 1 + 1 file changed, 1 insertion(+) diff --git a/esrally/client/factory.py b/esrally/client/factory.py index 2b5b99e3e..a30abfd11 100644 --- a/esrally/client/factory.py +++ b/esrally/client/factory.py @@ -236,6 +236,7 @@ async def on_request_end(session, trace_config_ctx, params): # will ultimately record the time when it received the *last* chunk. This is what we want because any code # that is using the Elasticsearch client library can only act on the response once it is fully received. trace_config.on_response_chunk_received.append(on_request_end) + trace_config.on_request_end.append(on_request_end) # ensure that we also stop the timer when a request "ends" with an exception (e.g. a timeout) trace_config.on_request_exception.append(on_request_end)