From 597f45a0de03c9c7ed996d0af36932d86d7923ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Guillermo=20D=C3=ADaz?= <guillermo@otherplane.com>
Date: Wed, 13 Dec 2023 18:02:15 +0100
Subject: [PATCH] feat(rad): include retrieval response time within
 WitnetResultReports

---
 rad/src/lib.rs | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/rad/src/lib.rs b/rad/src/lib.rs
index 64c826780..319f4a483 100644
--- a/rad/src/lib.rs
+++ b/rad/src/lib.rs
@@ -282,6 +282,7 @@ async fn http_response(
         })
     })?;
 
+    let start_ts = std::time::SystemTime::now();
     let response = client
         .send(request)
         .await
@@ -306,7 +307,21 @@ async fn http_response(
             message: x.to_string(),
         })?;
 
-    let result = run_retrieval_with_data_report(retrieve, &response_string, context, settings);
+    let result = run_retrieval_with_data_report(retrieve, &response_string, context, settings).map(
+        |report| {
+            let completion_ts = std::time::SystemTime::now();
+            
+            RadonReport {
+                context: ReportContext {
+                    start_time: Some(start_ts),
+                    completion_time: Some(completion_ts),
+                    ..report.context
+                },
+                running_time: completion_ts.duration_since(start_ts).unwrap_or_default(),
+                ..report
+            }
+        },
+    );
 
     match &result {
         Ok(report) => {