com.github.davidmoten.aws.lw.client.Response Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aws-lightweight-client-java Show documentation
Lightweight client for all AWS services (but still with useful builders and XML parser)
The newest version!
package com.github.davidmoten.aws.lw.client;

import java.net.HttpURLConnection;
import java.nio.charset.StandardCharsets;
import java.time.Instant;
import java.time.ZonedDateTime;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

import com.github.davidmoten.aws.lw.client.internal.util.Preconditions;

public final class Response {

    private final Map> headers;
    private final Map> headersLowerCaseKey;
    private final byte[] content;
    private final int statusCode;

    public Response(Map> headers, byte[] content, int statusCode) {
        this.headers = headers;
        this.headersLowerCaseKey = lowerCaseKey(headers);
        this.content = content;
        this.statusCode = statusCode;
    }

    public Map> headers() {
        return headers;
    }
    
    public Map> headersLowerCaseKey() {
        return headersLowerCaseKey;
    }

    public Optional firstHeader(String name) {
        List h = headersLowerCaseKey.get(lowerCase(name));
        if (h == null || h.isEmpty()) {
            return Optional.empty();
        } else {
            return Optional.of(h.get(0));
        }
    }

    public Optional firstHeaderFullDate(String name) {
        return firstHeader(name) //
                .map(x -> ZonedDateTime.parse(x, Formats.FULL_DATE).toInstant());
    }

    /**
     * Returns those headers that start with {@code x-amz-meta-} (and removes that
     * prefix).
     * 
     * @return headers that start with {@code x-amz-meta-} (and removes that prefix)
     */
    public Metadata metadata() {
        return new Metadata(headersLowerCaseKey //
                .entrySet() //
                .stream() //
                .filter(x -> x.getKey() != null) //
                .filter(x -> x.getKey().startsWith("x-amz-meta-")) //
                .collect(Collectors.toMap( //
                        x -> x.getKey().substring(11), //
                        x -> x.getValue().get(0))));
    }

    public Optional metadata(String name) {
        Preconditions.checkNotNull(name);
        // value() method does conversion of name to lower case
        return metadata().value(name);
    }

    public byte[] content() {
        return content;
    }

    public String contentUtf8() {
        return new String(content, StandardCharsets.UTF_8);
    }

    public int statusCode() {
        return statusCode;
    }

    public boolean isOk() {
        return statusCode >= 200 && statusCode <= 299;
    }

    /**
     * Returns true if and only if status code is 2xx. Returns false if status code
     * is 404 (NOT_FOUND) and throws a {@link ServiceException} otherwise.
     * 
     * @return true if status code 2xx, false if 404 otherwise throws
     *         ServiceException
     * @throws ServiceException if status code other than 2xx or 404
     */
    public boolean exists() {
        if (statusCode >= 200 && statusCode <= 299) {
            return true;
        } else if (statusCode == HttpURLConnection.HTTP_NOT_FOUND) {
            return false;
        } else {
            throw new ServiceException(statusCode, "call failed");
        }
    }

    private static Map> lowerCaseKey(Map> m) {
        return m.entrySet().stream().collect( //
                Collectors.toMap( //
                        entry -> lowerCase(entry.getKey()), //
                        entry -> entry.getValue()));
    }
    
    private static String lowerCase(String s) {
        return s == null ? s : s.toLowerCase(Locale.ENGLISH);
    }
    
    // TODO add toString method
}