1 files changed, 8 insertions, 5 deletions
diff --git a/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/api/MultiClusterSessionOutputStream.java b/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/api/MultiClusterSessionOutputStream.java
index bf55a46277d..e09cecf7161 100644
--- a/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/api/MultiClusterSessionOutputStream.java
+++ b/vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/api/MultiClusterSessionOutputStream.java
@@ -6,6 +6,7 @@ import com.yahoo.vespa.http.client.core.operationProcessor.OperationProcessor;
 
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.time.Clock;
 
 /**
  * Class for wiring up the Session API. It is the return value of stream() in the Session API.
@@ -17,19 +18,21 @@ class MultiClusterSessionOutputStream extends ByteArrayOutputStream {
     private final CharSequence documentId;
     private final OperationProcessor operationProcessor;
     private final Object context;
+    private final Clock clock;
 
-    public MultiClusterSessionOutputStream(
-            CharSequence documentId,
-            OperationProcessor operationProcessor,
-            Object context) {
+    public MultiClusterSessionOutputStream(CharSequence documentId,
+                                           OperationProcessor operationProcessor,
+                                           Object context,
+                                           Clock clock) {
         this.documentId = documentId;
         this.context = context;
         this.operationProcessor = operationProcessor;
+        this.clock = clock;
     }
 
     @Override
     public void close() throws IOException {
-        Document document = new Document(documentId.toString(), toByteArray(), context);
+        Document document = new Document(documentId.toString(), toByteArray(), context, clock.instant());
         operationProcessor.sendDocument(document);
         super.close();
     }