vespa-http-client/src/main/java/com/yahoo/vespa/http/client/core/api/MultiClusterSessionOutputStream.java


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

// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.http.client.core.api;

import com.yahoo.vespa.http.client.core.Document;
import com.yahoo.vespa.http.client.core.operationProcessor.OperationProcessor;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.time.Clock;

/**
 * Class for wiring up the Session API. It is the return value of stream() in the Session API.
 * 
 * @author dybis
*/
class MultiClusterSessionOutputStream extends ByteArrayOutputStream {

    private final CharSequence documentId;
    private final OperationProcessor operationProcessor;
    private final Object context;
    private final Clock clock;

    public MultiClusterSessionOutputStream(CharSequence documentId,
                                           OperationProcessor operationProcessor,
                                           Object context,
                                           Clock clock) {
        this.documentId = documentId;
        this.context = context;
        this.operationProcessor = operationProcessor;
        this.clock = clock;
    }

    @Override
    public void close() throws IOException {
        Document document = new Document(documentId.toString(), toByteArray(), context, clock.instant());
        operationProcessor.sendDocument(document);
        super.close();
    }

}