configserver/src/main/java/com/yahoo/vespa/config/server/http/StaticResponse.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

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.http;

import com.yahoo.container.jdisc.HttpResponse;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;

public class StaticResponse extends HttpResponse {
    private final String contentType;
    private final InputStream body;

    /**
     * @param body    Ownership is passed to StaticResponse (is responsible for closing it)
     */
    public StaticResponse(int status, String contentType, InputStream body) {
        super(status);
        this.contentType = contentType;
        this.body = body;
    }

    public StaticResponse(int status, String contentType, String body) {
        this(status, contentType, new ByteArrayInputStream(body.getBytes(StandardCharsets.UTF_8)));
    }

    @Override
    public void render(OutputStream outputStream) throws IOException {
        body.transferTo(outputStream);
        body.close();
    }

    @Override
    public String getContentType() {
        return contentType;
    }
}