Added support to expose query response stat: timeUsedMs (#87)

python-pinot-dbapi · Jan 12, 2024 · b43b5d1 · b43b5d1
1 parent d4f1665
commit b43b5d1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pinotdb/db.py b/pinotdb/db.py
@@ -311,6 +311,7 @@ def __init__(
         self.schema = None
         self.rowcount = -1
         self._results = None
+        self.timeUsedMs = -1
         self._debug = debug
         self._preserve_types = preserve_types
         self._use_multistage_engine = use_multistage_engine
@@ -405,6 +406,7 @@ def normalize_query_response(self, input_query, query_response):
 
         num_servers_responded = payload.get("numServersResponded", -1)
         num_servers_queried = payload.get("numServersQueried", -1)
+        self.timeUsedMs = payload.get("timeUsedMs", -1)
 
         self.check_sufficient_responded(
             input_query, num_servers_queried, num_servers_responded