Merge pull request #118 from srbaker/1.1-expose-execution-times

Expose the execution time metadata on ResultSummary.

Author: technige

neo4j/v1/bolt.py

@@ -202,8 +202,11 @@ class BoltStatementResult(StatementResult):
     def __init__(self, session, run_response, pull_all_response):
         super(BoltStatementResult, self).__init__(session)
 
+        all_metadata = {}
+
         def on_header(metadata):
             # Called on receipt of the result header.
+            all_metadata.update(metadata)
             self._keys = tuple(metadata["fields"])
 
         def on_record(values):
@@ -212,7 +215,8 @@ def on_record(values):
 
         def on_footer(metadata):
             # Called on receipt of the result footer.
-            self._summary = ResultSummary(self.statement, self.parameters, **metadata)
+            all_metadata.update(metadata)
+            self._summary = ResultSummary(self.statement, self.parameters, **all_metadata)
             self._session = None
 
         def on_failure(metadata):

neo4j/v1/summary.py

@@ -50,6 +50,12 @@ class ResultSummary(object):
     #: A :class:`.ProfiledPlan` instance
     profile = None
 
+    #: The time it took for the server to have the result available.
+    result_available_after = None
+
+    #: The time it took for the server to consume the result.
+    result_consumed_after = None
+
     #: Notifications provide extra information for a user executing a statement.
     #: They can be warnings about problematic queries or other valuable information that can be
     #: presented in a client.
@@ -62,6 +68,8 @@ def __init__(self, statement, parameters, **metadata):
         self.metadata = metadata
         self.statement_type = metadata.get("type")
         self.counters = SummaryCounters(metadata.get("stats", {}))
+        self.result_available_after = metadata.get("result_available_after")
+        self.result_consumed_after = metadata.get("result_consumed_after")
         if "plan" in metadata:
             self.plan = make_plan(metadata["plan"])
         if "profile" in metadata:

test/test_session.py

@@ -255,6 +255,12 @@ def test_can_obtain_notification_info(self):
             assert position.line == 1
             assert position.column == 1
 
+    @skipUnless(SERVER_VERSION >= (3, 1), "Execution times are not supported in 3.0.")
+    def test_contains_time_information(self):
+        with self.driver.session() as session:
+            summary = session.run("UNWIND range(1,1000) AS n RETURN n AS number").consume()
+            assert summary.result_consumed_after > 0
+            assert summary.result_available_after > 0
 
 class ResetTestCase(ServerTestCase):