io.polyglotted.common.web.HttpResponder Maven / Gradle / Ivy

Go to download
package io.polyglotted.common.web;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

import java.io.OutputStream;
import java.util.Map;

import static com.google.common.base.Strings.isNullOrEmpty;
import static io.polyglotted.common.util.BaseSerializer.serialize;
import static io.polyglotted.common.util.MapBuilder.immutableMap;
import static io.polyglotted.common.web.GatewayResponse.sendResult;
import static java.util.Base64.getEncoder;
import static org.apache.http.HttpHeaders.CONTENT_LENGTH;
import static org.apache.http.HttpHeaders.CONTENT_TYPE;
import static org.apache.http.HttpStatus.SC_OK;

@SuppressWarnings("WeakerAccess")
@Slf4j @RequiredArgsConstructor
public final class HttpResponder {
    final boolean isLoadBalanced;
    final OutputStream outputStream;

    public void sendBytes(int status, byte[] bytes) {
        sendResult(isLoadBalanced, outputStream, status, getEncoder().encodeToString(bytes), true,
            immutableMap(CONTENT_LENGTH, String.valueOf(bytes.length), CONTENT_TYPE, "application/octet-stream"));
    }

    public void sendObject(Object result) { sendObject(SC_OK, result); }

    public void sendObject(int status, Object result) { sendJson(status, serialize(result)); }

    public void sendJson(String json) { sendJson(SC_OK, json); }

    public void sendJson(int status, String json) { sendString(status, json, "application/json"); }

    public void sendString(String data, String contentType) { sendString(SC_OK, data, contentType); }

    public void sendString(int status, String data, String contentType) {
        if (isNullOrEmpty(data)) { sendStatus(status); return; }
        sendResult(isLoadBalanced, outputStream, status, data, false, immutableMap(CONTENT_TYPE, contentType));
    }

    public void sendStatus(int status) { sendStatus(status, immutableMap()); }

    public void sendStatus(int status, Map headers) { sendResult(isLoadBalanced, outputStream, status, null, false, headers); }
}