aboutsummaryrefslogtreecommitdiffstats
path: root/config/src/main/java/com/yahoo/vespa/config/protocol/SlimeConfigResponse.java
blob: a5549ab180d3f55fff4693e1830c2fe098c1c850 (plain) (blame)
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
// 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.vespa.config.ConfigPayload;
import com.yahoo.vespa.config.PayloadChecksums;

import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;

/**
 * Class for serializing config responses based on {@link com.yahoo.slime.Slime} implementing the {@link ConfigResponse} interface.
 *
 * @author Ulf Lilleengen
 */
public class SlimeConfigResponse implements ConfigResponse {

    private final AbstractUtf8Array payload;
    private final CompressionInfo compressionInfo;
    private final long generation;
    private final boolean applyOnRestart;
    private final PayloadChecksums payloadChecksums;

    public static SlimeConfigResponse fromConfigPayload(ConfigPayload payload,
                                                        long generation,
                                                        boolean applyOnRestart,
                                                        PayloadChecksums payloadChecksums) {
        AbstractUtf8Array data = payload.toUtf8Array(true);
        return new SlimeConfigResponse(data,
                                       generation,
                                       applyOnRestart,
                                       payloadChecksums,
                                       CompressionInfo.create(CompressionType.UNCOMPRESSED, data.getByteLength()));
    }

    public SlimeConfigResponse(AbstractUtf8Array payload,
                               long generation,
                               boolean applyOnRestart,
                               PayloadChecksums payloadChecksums,
                               CompressionInfo compressionInfo) {
        this.payload = payload;
        this.generation = generation;
        this.applyOnRestart = applyOnRestart;
        this.payloadChecksums = payloadChecksums;
        this.compressionInfo = compressionInfo;
    }

    @Override
    public AbstractUtf8Array getPayload() {
        return payload;
    }

    @Override
    public long getGeneration() {
        return generation;
    }

    @Override
    public boolean applyOnRestart() { return applyOnRestart; }

    @Override
    public void serialize(OutputStream os, CompressionType type) throws IOException {
        ByteBuffer buf = Payload.from(payload, compressionInfo).withCompression(type).getData().wrap();
        os.write(buf.array(), buf.arrayOffset()+buf.position(), buf.remaining());
    }

    @Override
    public String toString() {
        return "generation=" + generation +  "\n" +
                "checksums=" + payloadChecksums +  "\n" +
                Payload.from(payload, compressionInfo).withCompression(CompressionType.UNCOMPRESSED);
    }

    @Override
    public CompressionInfo getCompressionInfo() { return compressionInfo; }

    @Override
    public PayloadChecksums getPayloadChecksums() { return payloadChecksums; }
}