Added support for resultConsumedAfter

Author: pontusmelke

driver/src/main/java/org/neo4j/driver/internal/InternalStatementResult.java

@@ -144,6 +144,12 @@ public void resultAvailableAfter(long l)
             {
                 summaryBuilder.resultAvailableAfter( l );
             }
+
+            @Override
+            public void resultConsumedAfter(long l)
+            {
+                summaryBuilder.resultConsumedAfter( l );
+            }
         };
     }
 

driver/src/main/java/org/neo4j/driver/internal/net/SocketResponseHandler.java

@@ -91,6 +91,7 @@ public void handleSuccessMessage( Map<String,Value> meta )
         collectProfile( collector, meta.get( "profile" ) );
         collectNotifications( collector, meta.get( "notifications" ) );
         collectResultAvailableAfter( collector, meta.get("result_available_after"));
+        collectResultConsumedAfter( collector, meta.get("result_consumed_after"));
         collector.doneSuccess();
     }
 
@@ -102,6 +103,14 @@ private void collectResultAvailableAfter( StreamCollector collector, Value resul
         }
     }
 
+    private void collectResultConsumedAfter( StreamCollector collector, Value resultConsumedAfter )
+    {
+        if (resultConsumedAfter != null)
+        {
+            collector.resultConsumedAfter(resultConsumedAfter.asLong());
+        }
+    }
+
     private void collectNotifications( StreamCollector collector, Value notifications )
     {
         if ( notifications != null )

driver/src/main/java/org/neo4j/driver/internal/spi/StreamCollector.java

@@ -150,6 +150,12 @@ public void resultAvailableAfter( long l )
         {
 
         }
+
+        @Override
+        public void resultConsumedAfter( long l )
+        {
+
+        }
     }
 
     // TODO: This should be modified to simply have head/record/tail methods
@@ -177,5 +183,7 @@ public void resultAvailableAfter( long l )
     void doneIgnored();
 
     void resultAvailableAfter( long l );
+
+    void resultConsumedAfter( long l );
 }
 

driver/src/main/java/org/neo4j/driver/internal/summary/SummaryBuilder.java

@@ -44,6 +44,7 @@ public class SummaryBuilder implements StreamCollector
     private ProfiledPlan profile;
     private List<Notification> notifications = null;
     private long resultAvailableAfter;
+    private long resultConsumedAfter;
 
     public SummaryBuilder( Statement statement )
     {
@@ -156,6 +157,12 @@ public void resultAvailableAfter( long l )
       this.resultAvailableAfter = l;
     }
 
+    @Override
+    public void resultConsumedAfter( long l )
+    {
+        this.resultConsumedAfter = l;
+    }
+
     public ResultSummary build()
     {
         return new ResultSummary()
@@ -213,6 +220,12 @@ public long resultAvailableAfter( TimeUnit timeUnit )
             {
                 return timeUnit.convert( resultAvailableAfter, TimeUnit.MILLISECONDS );
             }
+
+            @Override
+            public long resultConsumedAfter( TimeUnit timeUnit )
+            {
+                return timeUnit.convert( resultConsumedAfter, TimeUnit.MILLISECONDS );
+            }
         };
     }
 }

driver/src/main/java/org/neo4j/driver/v1/summary/ResultSummary.java

@@ -99,4 +99,12 @@ public interface ResultSummary
      * @return The time it took for the server to have the result available in the provided time unit.
      */
     long resultAvailableAfter( TimeUnit unit );
+
+    /**
+     * The time it took the server to consume the result.
+     *
+     * @param unit The unit of the duration.
+     * @return The time it took for the server to consume the result in the provided time unit.
+     */
+    long resultConsumedAfter( TimeUnit unit );
 }

driver/src/test/java/org/neo4j/driver/v1/integration/SummaryIT.java

@@ -74,6 +74,7 @@ public void shouldContainBasicMetadata() throws Throwable
         assertFalse( summary.hasProfile() );
         assertThat( summary, equalTo( result.consume() ) );
         assertThat( summary.resultAvailableAfter( TimeUnit.MILLISECONDS ), greaterThan( 0L ) );
+        assertThat( summary.resultConsumedAfter( TimeUnit.MILLISECONDS ), greaterThan( 0L ) );
     }
 
     @Test