mesosphere.dcos.client.DCOSClient Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of marathon-client Show documentation

A Java API client for Mesosphere's Marathon.

There is a newer version: 0.6.3

package mesosphere.dcos.client;

import feign.Feign;
import feign.Response;
import feign.codec.ErrorDecoder;
import feign.gson.GsonDecoder;
import feign.gson.GsonEncoder;
import mesosphere.client.common.HttpClientUtils;
import mesosphere.client.common.ModelUtils;
import mesosphere.dcos.client.model.DCOSAuthCredentials;
import org.apache.commons.io.IOUtils;

import java.io.IOException;

public class DCOSClient {
    public static DCOS getInstance(String endpoint) {
        return getInstance(endpoint, Config.builder().build());
    }

    public static DCOS getInstance(String endpoint, final DCOSAuthCredentials authCredentials) {
        return getInstance(endpoint, Config.builder().withCredentials(authCredentials).build());
    }

    public static DCOS getInstance(String endpoint, final Config config) {
        GsonDecoder decoder = new GsonDecoder(ModelUtils.GSON);
        GsonEncoder encoder = new GsonEncoder(ModelUtils.GSON);

        Feign.Builder builder = Feign.builder()
                .client(HttpClientUtils.createHttpClient(config))
                .encoder(encoder)
                .decoder(decoder)
                .errorDecoder(new DCOSErrorDecoder());

        if (config.getCredentials() != null) {
            // Have to create a DCOS instance without authentication credentials that will be responsible for refreshing the authentication token.
            final Config noAuthConfig = Config.builder(config).withCredentials(null).build();
            builder.requestInterceptor(new DCOSAuthTokenHeaderInterceptor(config.getCredentials(), getInstance(endpoint, noAuthConfig)));
        }

        builder.requestInterceptor(new DCOSAPIInterceptor());

        return builder.target(DCOS.class, endpoint);
    }

    private static class DCOSErrorDecoder implements ErrorDecoder {
        @Override
        public Exception decode(String methodKey, Response response) {

            String details;
            try {
                details = IOUtils.toString(response.body().asInputStream(), "UTF-8");
            } catch (NullPointerException | IOException e) {
                details = "Unable to read response body";
            }
            return new DCOSException(response.status(), response.reason(), methodKey, details);
        }
    }
}