com.netflix.discovery.shared.transport.EurekaHttpResponse Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 Netflix, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.netflix.discovery.shared.transport;

import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * @author Tomasz Bak
 */
public class EurekaHttpResponse {
    private final int statusCode;
    private final T entity;
    private final Map headers;
    private final URI location;

    protected EurekaHttpResponse(int statusCode, T entity) {
        this.statusCode = statusCode;
        this.entity = entity;
        this.headers = null;
        this.location = null;
    }

    private EurekaHttpResponse(EurekaHttpResponseBuilder builder) {
        this.statusCode = builder.statusCode;
        this.entity = builder.entity;
        this.headers = builder.headers;

        if (headers != null) {
            String locationValue = headers.get(HttpHeaders.LOCATION);
            try {
                this.location = locationValue == null ? null : new URI(locationValue);
            } catch (URISyntaxException e) {
                throw new TransportException("Invalid Location header value in response; cannot complete the request (location="
                        + locationValue + ')', e);
            }
        } else {
            this.location = null;
        }
    }

    public int getStatusCode() {
        return statusCode;
    }

    public URI getLocation() {
        return location;
    }

    public Map getHeaders() {
        return headers == null ? Collections.emptyMap() : headers;
    }

    public T getEntity() {
        return entity;
    }

    public static EurekaHttpResponse status(int status) {
        return new EurekaHttpResponse<>(status, null);
    }

    public static EurekaHttpResponseBuilder anEurekaHttpResponse(int statusCode) {
        return new EurekaHttpResponseBuilder<>(statusCode);
    }

    public static  EurekaHttpResponseBuilder anEurekaHttpResponse(int statusCode, Class entityType) {
        return new EurekaHttpResponseBuilder(statusCode);
    }

    public static  EurekaHttpResponseBuilder anEurekaHttpResponse(int statusCode, T entity) {
        return new EurekaHttpResponseBuilder(statusCode).entity(entity);
    }

    public static class EurekaHttpResponseBuilder {

        private final int statusCode;
        private T entity;
        private Map headers;

        private EurekaHttpResponseBuilder(int statusCode) {
            this.statusCode = statusCode;
        }

        public EurekaHttpResponseBuilder entity(T entity) {
            this.entity = entity;
            return this;
        }

        public EurekaHttpResponseBuilder entity(T entity, MediaType contentType) {
            return entity(entity).type(contentType);
        }

        public EurekaHttpResponseBuilder type(MediaType contentType) {
            headers(HttpHeaders.CONTENT_TYPE, contentType.toString());
            return this;
        }

        public EurekaHttpResponseBuilder headers(String name, Object value) {
            if (headers == null) {
                headers = new HashMap<>();
            }
            headers.put(name, value.toString());
            return this;
        }

        public EurekaHttpResponseBuilder headers(Map headers) {
            this.headers = headers;
            return this;
        }

        public EurekaHttpResponse build() {
            return new EurekaHttpResponse(this);
        }
    }
}