Use an interface that is more flexible.

2 files changed, 3 insertions, 4 deletions

diff --git a/config/src/main/java/com/yahoo/vespa/config/LZ4PayloadCompressor.java b/config/src/main/java/com/yahoo/vespa/config/LZ4PayloadCompressor.java
index a70a2861cc8..ba2caaf3e91 100644
--- a/config/src/main/java/com/yahoo/vespa/config/LZ4PayloadCompressor.java
+++ b/config/src/main/java/com/yahoo/vespa/config/LZ4PayloadCompressor.java
@@ -26,8 +26,8 @@ public class LZ4PayloadCompressor {
         return compressor.compressUnconditionally(input);
     }
 
-    public void decompress(byte[] input, byte[] output) {
-        compressor.decompressUnconditionally(input, output);
+    public byte [] decompress(byte[] input, int uncompressedLen) {
+        return compressor.decompressUnconditionally(input, 0, uncompressedLen);
     }
 
 }
diff --git a/config/src/main/java/com/yahoo/vespa/config/protocol/Payload.java b/config/src/main/java/com/yahoo/vespa/config/protocol/Payload.java
index 88e8b21347f..13f9602f70c 100644
--- a/config/src/main/java/com/yahoo/vespa/config/protocol/Payload.java
+++ b/config/src/main/java/com/yahoo/vespa/config/protocol/Payload.java
@@ -60,8 +60,7 @@ public class Payload {
     public Payload withCompression(CompressionType requestedCompression) {
         CompressionType responseCompression = compressionInfo.getCompressionType();
         if (requestedCompression == CompressionType.UNCOMPRESSED && responseCompression == CompressionType.LZ4) {
-            byte[] buffer = new byte[compressionInfo.getUncompressedSize()];
-            compressor.decompress(data.getBytes(), buffer);
+            byte[] buffer = compressor.decompress(data.getBytes(), compressionInfo.getUncompressedSize());
             Utf8Array data = new Utf8Array(buffer);
             CompressionInfo info = CompressionInfo.create(CompressionType.UNCOMPRESSED, compressionInfo.getUncompressedSize());
             return Payload.from(data, info);