From 51c6a932dd232d53ca8bd795faf3ca6737723a5a Mon Sep 17 00:00:00 2001
From: Pablo Martin <pablo.martin@lolamarket.com>
Date: Tue, 26 Jul 2022 14:13:39 +0200
Subject: [PATCH] Number of rows obtained by each query is now printed.

---
 CHANGELOG.MD               | 8 ++++++++
 query_performance_gauge.py | 3 ++-
 2 files changed, 10 insertions(+), 1 deletion(-)
 create mode 100644 CHANGELOG.MD

diff --git a/CHANGELOG.MD b/CHANGELOG.MD
new file mode 100644
index 0000000..e834bff
--- /dev/null
+++ b/CHANGELOG.MD
@@ -0,0 +1,8 @@
+# Changelog 
+
+## Unreleased
+
+### Added 
+
+- Successful print message for each query will now also show the number of rows returned by the query.
+
diff --git a/query_performance_gauge.py b/query_performance_gauge.py
index 702e8d0..cfa2d2f 100644
--- a/query_performance_gauge.py
+++ b/query_performance_gauge.py
@@ -59,6 +59,7 @@ def measure_query_runtime(
     cur = connection.cursor()
     cur.execute(query_to_measure.query_string)
     rows = cur.fetchall()
+    obtained_rows = len(rows)
     print(
-        f"Query '{query_to_measure.name}' took {int(time.time() - start_time)} seconds to run."
+        f"Query '{query_to_measure.name}' took {int(time.time() - start_time)} seconds to run and returned {obtained_rows} rows."
     )