com.vmware.photon.controller.api.client.resource.ClusterApi.java Source code

Introduction

Here is the source code for com.vmware.photon.controller.api.client.resource.ClusterApi.java

Source

/*
 * Copyright 2015 VMware, Inc. All Rights Reserved.
 *
 * 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.vmware.photon.controller.api.client.resource;

import com.vmware.photon.controller.api.client.RestClient;
import com.vmware.photon.controller.api.model.Cluster;
import com.vmware.photon.controller.api.model.ClusterResizeOperation;
import com.vmware.photon.controller.api.model.ResourceList;
import com.vmware.photon.controller.api.model.Task;
import com.vmware.photon.controller.api.model.Vm;

import com.fasterxml.jackson.core.type.TypeReference;
import com.google.common.util.concurrent.FutureCallback;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;

import javax.annotation.Nullable;

import java.io.IOException;

/**
 * Cluster Api.
 */
public class ClusterApi extends ApiBase {
    public ClusterApi(RestClient restClient) {
        super(restClient);
    }

    @Override
    public String getBasePath() {
        return "/clusters";
    }

    /**
     * Get details about the specified cluster.
     *
     * @param clusterId
     * @return
     * @throws IOException
     */
    public Cluster getCluster(String clusterId) throws IOException {
        String path = String.format("%s/%s", getBasePath(), clusterId);

        HttpResponse httpResponse = this.restClient.perform(RestClient.Method.GET, path, null);
        this.restClient.checkResponse(httpResponse, HttpStatus.SC_OK);

        return this.restClient.parseHttpResponse(httpResponse, new TypeReference<Cluster>() {
        });
    }

    /**
     * Get details about the specified cluster.
     *
     * @param clusterId
     * @param responseCallback
     * @throws IOException
     */
    public void getClusterAsync(final String clusterId, final FutureCallback<Cluster> responseCallback)
            throws IOException {
        final String path = String.format("%s/%s", getBasePath(), clusterId);

        getObjectByPathAsync(path, responseCallback, new TypeReference<Cluster>() {
        });
    }

    /**
     * Delete the specified cluster.
     *
     * @param clusterId
     * @return
     * @throws IOException
     */
    public Task delete(String clusterId) throws IOException {
        String path = String.format("%s/%s", getBasePath(), clusterId);

        HttpResponse response = this.restClient.perform(RestClient.Method.DELETE, path, null);
        this.restClient.checkResponse(response, HttpStatus.SC_CREATED);

        return parseTaskFromHttpResponse(response);
    }

    /**
     * Delete the specified cluster.
     *
     * @param clusterId
     * @param responseCallback
     * @throws IOException
     */
    public void deleteAsync(final String clusterId, final FutureCallback<Task> responseCallback)
            throws IOException {
        deleteObjectAsync(clusterId, responseCallback);
    }

    /**
     * Resize the specified cluster.
     *
     * @param clusterId
     * @param size
     * @return
     * @throws IOException
     */
    public Task resize(String clusterId, int size) throws IOException {
        String path = String.format("%s/%s/resize", getBasePath(), clusterId);

        ClusterResizeOperation op = new ClusterResizeOperation();
        op.setNewSlaveCount(size);

        HttpResponse response = this.restClient.perform(RestClient.Method.POST, path, serializeObjectAsJson(op));
        this.restClient.checkResponse(response, HttpStatus.SC_CREATED);

        return parseTaskFromHttpResponse(response);
    }

    /**
     * Resize the specified cluster.
     *
     * @param clusterId
     * @param size
     * @param responseCallback
     * @throws IOException
     */
    public void resizeAsync(final String clusterId, final int size, final FutureCallback<Task> responseCallback)
            throws IOException {
        String path = String.format("%s/%s/resize", getBasePath(), clusterId);

        ClusterResizeOperation op = new ClusterResizeOperation();
        op.setNewSlaveCount(size);

        createObjectAsync(path, serializeObjectAsJson(op), responseCallback);
    }

    /**
     * Get a list of vms in the specified cluster.
     * @param clusterId
     * @return
     * @throws IOException
     */
    public ResourceList<Vm> getVmsInCluster(String clusterId) throws IOException {
        String path = String.format("%s/%s/vms", getBasePath(), clusterId);

        ResourceList<Vm> vmResourceList = new ResourceList<>();
        ResourceList<Vm> resourceList = getVmResourceList(path);
        vmResourceList.setItems(resourceList.getItems());
        while (resourceList.getNextPageLink() != null && !resourceList.getNextPageLink().isEmpty()) {
            resourceList = getVmResourceList(resourceList.getNextPageLink());
            vmResourceList.getItems().addAll(resourceList.getItems());
        }

        return vmResourceList;
    }

    /**
     * Get all Vms at specified path.
     *
     * @param path
     * @return
     * @throws IOException
     */
    private ResourceList<Vm> getVmResourceList(String path) throws IOException {
        HttpResponse httpResponse = this.restClient.perform(RestClient.Method.GET, path, null);
        this.restClient.checkResponse(httpResponse, HttpStatus.SC_OK);
        return this.restClient.parseHttpResponse(httpResponse, new TypeReference<ResourceList<Vm>>() {
        });
    }

    /**
     * Get a list of vms in the specified cluster.
     *
     * @param clusterId
     * @param responseCallback
     * @throws IOException
     */
    public void getVmsInClusterAsync(final String clusterId,
            final FutureCallback<ResourceList<Vm>> responseCallback) throws IOException {
        String path = String.format("%s/%s/vms", getBasePath(), clusterId);

        ResourceList<Vm> vmResourceList = new ResourceList<>();
        FutureCallback<ResourceList<Vm>> callback = new FutureCallback<ResourceList<Vm>>() {
            @Override
            public void onSuccess(@Nullable ResourceList<Vm> result) {
                if (vmResourceList.getItems() == null) {
                    vmResourceList.setItems(result.getItems());
                } else {
                    vmResourceList.getItems().addAll(result.getItems());
                }
                if (result.getNextPageLink() != null && !result.getNextPageLink().isEmpty()) {
                    try {
                        getObjectByPathAsync(result.getNextPageLink(), this, new TypeReference<ResourceList<Vm>>() {
                        });
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                } else {
                    responseCallback.onSuccess(vmResourceList);
                }
            }

            @Override
            public void onFailure(Throwable t) {
                responseCallback.onFailure(t);
            }
        };

        getObjectByPathAsync(path, callback, new TypeReference<ResourceList<Vm>>() {
        });
    }
}