Limit to 16MB outstanding visit response data

1 files changed, 3 insertions, 1 deletions

diff --git a/vespaclient-container-plugin/src/main/java/com/yahoo/document/restapi/resource/DocumentV1ApiHandler.java b/vespaclient-container-plugin/src/main/java/com/yahoo/document/restapi/resource/DocumentV1ApiHandler.java
index f1d6b4825c6..865aed221c5 100644
--- a/vespaclient-container-plugin/src/main/java/com/yahoo/document/restapi/resource/DocumentV1ApiHandler.java
+++ b/vespaclient-container-plugin/src/main/java/com/yahoo/document/restapi/resource/DocumentV1ApiHandler.java
@@ -9,6 +9,7 @@ import com.yahoo.concurrent.DaemonThreadFactory;
 import com.yahoo.container.core.HandlerMetricContextUtil;
 import com.yahoo.container.core.documentapi.VespaDocumentAccess;
 import com.yahoo.container.jdisc.ContentChannelOutputStream;
+import com.yahoo.container.jdisc.MaxPendingContentChannelOutputStream;
 import com.yahoo.document.Document;
 import com.yahoo.document.DocumentId;
 import com.yahoo.document.DocumentOperation;
@@ -582,7 +583,8 @@ public class DocumentV1ApiHandler extends AbstractRequestHandler {
 
         private JsonResponse(ResponseHandler handler, boolean streaming) throws IOException {
             this.handler = handler;
-            out = new ContentChannelOutputStream(buffer);
+            out = streaming ? new MaxPendingContentChannelOutputStream(buffer, 1 << 24)
+                            : new ContentChannelOutputStream(buffer);
             json = jsonFactory.createGenerator(out);
             json.writeStartObject();
         }