io.github.openunirest.request.ResponseBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of unirest-java Show documentation

Simplified, lightweight HTTP client library

There is a newer version: 2.5.03

package io.github.openunirest.request;


import io.github.openunirest.http.BodyData;
import io.github.openunirest.http.HttpResponse;
import io.github.openunirest.http.JsonNode;
import io.github.openunirest.http.ObjectMapper;
import io.github.openunirest.http.exceptions.UnirestException;
import io.github.openunirest.http.options.Option;
import io.github.openunirest.http.options.Options;

import java.io.InputStream;
import java.io.UnsupportedEncodingException;

import static io.github.openunirest.http.BodyData.from;


class ResponseBuilder {

    public HttpResponse asJson(org.apache.http.HttpResponse response) {
        return new HttpResponse<>(response, from(response.getEntity(), b -> toJson(b)));
    }

    public HttpResponse asBinary(org.apache.http.HttpResponse response){
        return new HttpResponse<>(response, from(response.getEntity(), BodyData::getRawInput));
    }

    public  HttpResponse asObject(org.apache.http.HttpResponse response, Class aClass) {
        return new HttpResponse<>(response, from(response.getEntity(), b -> toObject(b, aClass)));
    }

    public  HttpResponse asObject(org.apache.http.HttpResponse response, GenericType genericType) {
        return new HttpResponse<>(response, from(response.getEntity(), b -> toObject(b, genericType)));
    }

    public HttpResponse asString(org.apache.http.HttpResponse response) {
        return new HttpResponse<>(response, from(response.getEntity(), this::toString));
    }

    private  T toObject(BodyData b, GenericType genericType) {
        ObjectMapper o = getObjectMapper();
        return o.readValue(toString(b), genericType);
    }

    private  T toObject(BodyData b, Class aClass) {
        ObjectMapper o = getObjectMapper();
        return o.readValue(toString(b), aClass);
    }

    private String toString(BodyData b) {
        try {
            return new String(b.getRawBytes(), b.getCharset());
        } catch (UnsupportedEncodingException e) {
            throw new ParsingException(e);
        }
    }

    private JsonNode toJson(BodyData b) {
        String jsonString = toString(b);
        return new JsonNode(jsonString);
    }

    private ObjectMapper getObjectMapper() {
        return Options.tryGet(Option.OBJECT_MAPPER, ObjectMapper.class)
                .orElseThrow(() -> new UnirestException("No Object Mapper Configured. Please configure one with Unirest.setObjectMapper"));
    }

    public static class ParsingException extends RuntimeException {

        public ParsingException(Exception e){
            super(e);
        }
    }
}