1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.protocol;
import com.yahoo.text.AbstractUtf8Array;
import com.yahoo.text.Utf8Array;
import com.yahoo.text.Utf8String;
import com.yahoo.vespa.config.ConfigPayload;
import com.yahoo.vespa.config.LZ4PayloadCompressor;
import java.util.Objects;
/**
* An immutable config payload
*
* @author hmusum
* @author bratseth
*/
public class Payload {
private final AbstractUtf8Array data;
private final CompressionInfo compressionInfo;
private final static LZ4PayloadCompressor compressor = new LZ4PayloadCompressor();
private Payload(ConfigPayload payload) {
this.data = payload.toUtf8Array(true);
this.compressionInfo = CompressionInfo.create(CompressionType.UNCOMPRESSED, data.getByteLength());
}
private Payload(AbstractUtf8Array payload, CompressionInfo compressionInfo) {
Objects.requireNonNull(payload, "Payload");
Objects.requireNonNull(compressionInfo, "CompressionInfo");
this.data = payload;
this.compressionInfo = compressionInfo;
}
public static Payload from(ConfigPayload payload) {
return new Payload(payload);
}
/** Creates an uncompressed payload from a string */
public static Payload from(String payload) {
return new Payload(new Utf8String(payload), CompressionInfo.uncompressed());
}
public static Payload from(String payload, CompressionInfo compressionInfo) {
return new Payload(new Utf8String(payload), compressionInfo);
}
/** Creates an uncompressed payload from an Utf8Array */
public static Payload from(AbstractUtf8Array payload) {
return new Payload(payload, CompressionInfo.uncompressed());
}
public static Payload from(AbstractUtf8Array payload, CompressionInfo compressionInfo) {
return new Payload(payload, compressionInfo);
}
public AbstractUtf8Array getData() { return data; }
/** Returns a copy of this payload where the data is compressed using the given compression */
public Payload withCompression(CompressionType requestedCompression) {
CompressionType responseCompression = compressionInfo.getCompressionType();
if (requestedCompression == CompressionType.UNCOMPRESSED && responseCompression == CompressionType.LZ4) {
byte[] buffer = compressor.decompress(data.wrap(), compressionInfo.getUncompressedSize());
Utf8Array data = new Utf8Array(buffer);
CompressionInfo info = CompressionInfo.create(CompressionType.UNCOMPRESSED, compressionInfo.getUncompressedSize());
return Payload.from(data, info);
} else if (requestedCompression == CompressionType.LZ4 && responseCompression == CompressionType.UNCOMPRESSED) {
Utf8Array data = new Utf8Array(compressor.compress(this.data.wrap()));
CompressionInfo info = CompressionInfo.create(CompressionType.LZ4, this.data.getByteLength());
return Payload.from(data, info);
} else {
return Payload.from(data, compressionInfo);
}
}
public CompressionInfo getCompressionInfo() { return compressionInfo; }
@Override
public String toString() {
if (compressionInfo.getCompressionType() == CompressionType.UNCOMPRESSED)
return data.toString();
else
return withCompression(CompressionType.UNCOMPRESSED).toString();
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Payload other = (Payload) o;
return this.compressionInfo.equals(other.compressionInfo) && this.data.equals(other.data);
}
@Override
public int hashCode() {
return data.hashCode() + 31 * compressionInfo.hashCode();
}
}
|