com.clianz.spur.Res Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spur-framework Show documentation
Spur server core library
The newest version!
package com.clianz.spur;

import static com.clianz.spur.helpers.HttpMethods.HEAD;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
import java.util.Map;

import org.boon.json.JsonFactory;
import org.boon.json.ObjectMapper;
import org.jboss.logging.Logger;
import org.xnio.channels.StreamSinkChannel;

import io.undertow.server.HttpServerExchange;
import io.undertow.util.Headers;
import io.undertow.util.HttpString;

public class Res {

    private static final Logger LOGGER = Logger.getLogger(Res.class.getName());
    private static final String JSON_CONTENT_TYPE = "application/json";
    private static final ThreadLocal LOCAL_MAPPER = new ThreadLocal<>();
    private HttpServerExchange httpServerExchange;

    protected Res(HttpServerExchange httpServerExchange) {
        this.httpServerExchange = httpServerExchange;
    }

    public HttpServerExchange getRawHttpServerExchange() {
        return httpServerExchange;
    }

    public Res header(String header, String value) {
        httpServerExchange.getResponseHeaders()
                .put(new HttpString(header), value);
        return this;
    }

    public Res headers(Map headers) {
        headers.forEach((header, value) -> httpServerExchange.getResponseHeaders()
                .put(new HttpString(header), value));
        return this;
    }

    public Res status(int statusCode) {
        httpServerExchange.setStatusCode(statusCode);
        return this;
    }

    public void send(String body) {
        if (httpServerExchange.getRequestMethod()
                .equals(HEAD)) {
            httpServerExchange.endExchange();
            return;
        }
        httpServerExchange.getResponseSender()
                .send(body, StandardCharsets.UTF_8);
        httpServerExchange.endExchange();
    }

    public void send() {
        httpServerExchange.endExchange();
    }

    public void send(ByteBuffer byteBuffer) {
        if (httpServerExchange.getRequestMethod()
                .equals(HEAD)) {
            httpServerExchange.endExchange();
            return;
        }
        httpServerExchange.getResponseSender()
                .send(byteBuffer);
        httpServerExchange.endExchange();
    }

    public void send(Object obj) {
        if (httpServerExchange.getRequestMethod()
                .equals(HEAD)) {
            httpServerExchange.endExchange();
            return;
        }
        httpServerExchange.getResponseHeaders()
                .put(Headers.CONTENT_TYPE, JSON_CONTENT_TYPE);

        ObjectMapper objectMapper = LOCAL_MAPPER.get();
        if (objectMapper == null) {
            LOGGER.info("Creating new parser.........");
            objectMapper = JsonFactory.createUseJSONDates();
            LOCAL_MAPPER.set(objectMapper);
        }

        String jsonStr = objectMapper
                .toJson(obj);
        ByteBuffer byteBuffer = ByteBuffer.wrap(jsonStr.getBytes(StandardCharsets.UTF_8));
        StreamSinkChannel responseChannel = httpServerExchange.getResponseChannel();
        int written;
        try {
            do {
                written = responseChannel.write(byteBuffer);
            } while (byteBuffer.hasRemaining() && written > 0);
        } catch (IOException e) {
            LOGGER.error("Can not write response: " + e.getMessage(), e);
        }
    }
}