diff options
Diffstat (limited to 'container-search')
| -rw-r--r-- | container-search/src/main/java/com/yahoo/search/dispatch/rpc/MapConverter.java | 10 |
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()))); |