Allocate the optional buffer lazily

1 files changed, 8 insertions, 2 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/dispatch/rpc/MapConverter.java b/container-search/src/main/java/com/yahoo/search/dispatch/rpc/MapConverter.java
index c71a06a5a49..4bea68dc403 100644
--- a/container-search/src/main/java/com/yahoo/search/dispatch/rpc/MapConverter.java
+++ b/container-search/src/main/java/com/yahoo/search/dispatch/rpc/MapConverter.java
@@ -19,10 +19,13 @@ import java.util.function.Consumer;
 public class MapConverter {
 
     public static void convertMapTensors(Map<String, Object> map, Consumer<TensorProperty.Builder> inserter) {
-        GrowableByteBuffer buffer = new GrowableByteBuffer(4096);
+        GrowableByteBuffer buffer = null;
         for (var entry : map.entrySet()) {
             var value = entry.getValue();
             if (value instanceof Tensor tensor) {
+                if (buffer == null) {
+                    buffer = new GrowableByteBuffer(4096);
+                }
                 buffer.clear();
                 TypedBinaryFormat.encode(tensor, buffer);
                 inserter.accept(TensorProperty.newBuilder().setName(entry.getKey()).setValue(ByteString.copyFrom(buffer.getByteBuffer().flip())));
@@ -51,7 +54,7 @@ public class MapConverter {
     public static void convertMultiMap(Map<String, List<Object>> map,
                                        Consumer<StringProperty.Builder> stringInserter,
                                        Consumer<TensorProperty.Builder> tensorInserter) {
-        GrowableByteBuffer buffer = new GrowableByteBuffer(4096);
+        GrowableByteBuffer buffer = null;
         for (var entry : map.entrySet()) {
             if (entry.getValue() != null) {
                 var key = entry.getKey();
@@ -59,6 +62,9 @@ public class MapConverter {
                 for (var value : entry.getValue()) {
                     if (value != null) {
                         if (value instanceof Tensor tensor) {
+                            if (buffer == null) {
+                                buffer = new GrowableByteBuffer(4096);
+                            }
                             buffer.clear();
                             TypedBinaryFormat.encode(tensor, buffer);
                             tensorInserter.accept(TensorProperty.newBuilder().setName(key).setValue(ByteString.copyFrom(buffer.getByteBuffer().flip())));