Added support for resultAvailableAfter

Author: pontusmelke

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

@@ -82,12 +82,19 @@ public void done()
                     keys = new ArrayList<>();
                 }
             }
+
+            @Override
+            public void resultAvailableAfter( long l )
+            {
+              pullAllResponseCollector.resultAvailableAfter( l );
+            }
         };
     }
 
     private StreamCollector newPullAllResponseCollector( Statement statement )
     {
         final SummaryBuilder summaryBuilder = new SummaryBuilder( statement );
+
         return new StreamCollector.NoOperationStreamCollector()
         {
             @Override
@@ -131,6 +138,12 @@ public void done() {
                 summary = summaryBuilder.build();
                 done = true;
             }
+
+            @Override
+            public void resultAvailableAfter(long l)
+            {
+                summaryBuilder.resultAvailableAfter( l );
+            }
         };
     }
 

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

@@ -90,9 +90,18 @@ public void handleSuccessMessage( Map<String,Value> meta )
         collectPlan( collector, meta.get( "plan" ) );
         collectProfile( collector, meta.get( "profile" ) );
         collectNotifications( collector, meta.get( "notifications" ) );
+        collectResultAvailableAfter( collector, meta.get("result_available_after"));
         collector.doneSuccess();
     }
 
+    private void collectResultAvailableAfter( StreamCollector collector, Value resultAvailableAfter )
+    {
+        if (resultAvailableAfter != null)
+        {
+            collector.resultAvailableAfter(resultAvailableAfter.asLong());
+        }
+    }
+
     private void collectNotifications( StreamCollector collector, Value notifications )
     {
         if ( notifications != null )

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

@@ -144,6 +144,12 @@ public void doneIgnored()
         {
             done();
         }
+
+        @Override
+        public void resultAvailableAfter( long l )
+        {
+
+        }
     }
 
     // TODO: This should be modified to simply have head/record/tail methods
@@ -169,5 +175,7 @@ public void doneIgnored()
     void doneFailure( Neo4jException error );
 
     void doneIgnored();
+
+    void resultAvailableAfter( long l );
 }
 

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

@@ -20,18 +20,19 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 import org.neo4j.driver.internal.spi.StreamCollector;
+import org.neo4j.driver.v1.Statement;
+import org.neo4j.driver.v1.Value;
+import org.neo4j.driver.v1.exceptions.ClientException;
 import org.neo4j.driver.v1.exceptions.Neo4jException;
 import org.neo4j.driver.v1.summary.Notification;
 import org.neo4j.driver.v1.summary.Plan;
 import org.neo4j.driver.v1.summary.ProfiledPlan;
 import org.neo4j.driver.v1.summary.ResultSummary;
-import org.neo4j.driver.v1.Statement;
 import org.neo4j.driver.v1.summary.StatementType;
 import org.neo4j.driver.v1.summary.SummaryCounters;
-import org.neo4j.driver.v1.Value;
-import org.neo4j.driver.v1.exceptions.ClientException;
 
 public class SummaryBuilder implements StreamCollector
 {
@@ -42,6 +43,7 @@ public class SummaryBuilder implements StreamCollector
     private Plan plan = null;
     private ProfiledPlan profile;
     private List<Notification> notifications = null;
+    private long resultAvailableAfter;
 
     public SummaryBuilder( Statement statement )
     {
@@ -148,6 +150,12 @@ public void doneIgnored()
         // intentionally empty
     }
 
+    @Override
+    public void resultAvailableAfter( long l )
+    {
+      this.resultAvailableAfter = l;
+    }
+
     public ResultSummary build()
     {
         return new ResultSummary()
@@ -199,6 +207,12 @@ public List<Notification> notifications()
             {
                 return notifications == null ? new ArrayList<Notification>() : notifications;
             }
+
+            @Override
+            public long resultAvailableAfter( TimeUnit timeUnit )
+            {
+                return timeUnit.convert( resultAvailableAfter, TimeUnit.MILLISECONDS );
+            }
         };
     }
 }

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

@@ -19,9 +19,10 @@
 package org.neo4j.driver.v1.summary;
 
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 
-import org.neo4j.driver.v1.util.Immutable;
 import org.neo4j.driver.v1.Statement;
+import org.neo4j.driver.v1.util.Immutable;
 
 /**
  * The result summary of running a statement. The result summary interface can be used to investigate
@@ -90,4 +91,12 @@ public interface ResultSummary
      * notifications produced while executing the statement.
      */
     List<Notification> notifications();
+
+    /**
+     * The time it took the server to make the result available for consumption.
+     *
+     * @param unit The unit of the duration.
+     * @return The time it took for the server to have the result available in the provided time unit.
+     */
+    long resultAvailableAfter( TimeUnit unit );
 }

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

@@ -22,6 +22,8 @@
 import org.junit.Test;
 
 import java.util.List;
+import java.util.concurrent.TimeUnit;
+
 import org.neo4j.driver.v1.StatementResult;
 import org.neo4j.driver.v1.Value;
 import org.neo4j.driver.v1.Values;
@@ -71,6 +73,7 @@ public void shouldContainBasicMetadata() throws Throwable
         assertFalse( summary.hasPlan() );
         assertFalse( summary.hasProfile() );
         assertThat( summary, equalTo( result.consume() ) );
+        assertThat( summary.resultAvailableAfter( TimeUnit.MILLISECONDS ), greaterThan( 0L ) );
     }
 
     @Test