com.aventstack.chaintest.http.JsonMappedBodyHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of chaintest-core Show documentation

Core Java client library for ChainTest framework

The newest version!

package com.aventstack.chaintest.http;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.UncheckedIOException;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;

public class JsonMappedBodyHandler implements HttpResponse.BodyHandler {

    private static final Logger log = LoggerFactory.getLogger(JsonMappedBodyHandler.class);

    private final Class _clazz;
    private final ObjectMapper _objectMapper;

    public JsonMappedBodyHandler(final Class clazz, final ObjectMapper objectMapper) {
        _clazz = clazz;
        _objectMapper = objectMapper;
    }

    @Override
    public HttpResponse.BodySubscriber apply(final HttpResponse.ResponseInfo responseInfo) {
        final HttpResponse.BodySubscriber bodySubscriber = HttpResponse.BodySubscribers.ofString(StandardCharsets.UTF_8);
        return HttpResponse.BodySubscribers.mapping(bodySubscriber,
            (final String body) -> {
                log.debug("Received response from service: {}", body);
                // We can safely check if the response type is ErrorResponse using the below
                // since neither elements appear in Build or Test entities
                if (body.contains("\"error\":") && body.contains("\"status\":")) {
                    log.error("Service returned ErrorResponse: {}", body);
                }
                try {
                    return _objectMapper.readValue(body, _clazz);
                } catch (final JsonProcessingException e) {
                    log.error("Failed to map expected response of type {}. Received response: {}",
                            _clazz.getSimpleName(), body, e);
                    throw new UncheckedIOException(e);
                }
            });
    }

}