com.ecsteam.cloudlaunch.services.statistics.CloudFoundryStatisticsProvider.java Source code

Introduction

Here is the source code for com.ecsteam.cloudlaunch.services.statistics.CloudFoundryStatisticsProvider.java

Source

/*
 * Copyright 2015 ECS Team, 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.ecsteam.cloudlaunch.services.statistics;

import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.List;

import org.cloudfoundry.client.lib.CloudCredentials;
import org.cloudfoundry.client.lib.CloudFoundryClient;
import org.cloudfoundry.client.lib.CloudFoundryOperations;
import org.cloudfoundry.client.lib.domain.ApplicationStats;
import org.cloudfoundry.client.lib.domain.InstanceStats;
import org.cloudfoundry.client.lib.domain.InstanceStats.Usage;
import org.springframework.beans.factory.annotation.Value;

import com.ecsteam.cloudlaunch.services.statistics.model.ApplicationInstance;
import com.ecsteam.cloudlaunch.services.statistics.model.ApplicationStatistics;
import com.ecsteam.cloudlaunch.services.statistics.model.VcapApplication;
import com.ecsteam.cloudlaunch.services.statistics.model.VcapApplication.Limits;
import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * Use the Cloud Controller API to determine statistics for the current application
 * 
 * @author Josh Ghiloni
 *
 */
public class CloudFoundryStatisticsProvider implements ApplicationStatisticsProvider {
    private VcapApplication application;

    @Value("${ecs.cc.user:}")
    private String cloudControllerUserId;

    @Value("${ecs.cc.password:}")
    private String cloudControllerPassword;

    @Value("${ecs.cc.url:}")
    private String url;

    @Value("${ecs.cc.trustSelfSigned:false}")
    private Boolean trustSelfSignedCerts;

    private static long MEGS_TO_BYTES = 1048576L;

    @Override
    public ApplicationStatistics getCurrentStatistics() throws IOException {
        String vcapApplicationString = System.getenv("VCAP_APPLICATION");
        application = (new ObjectMapper()).readValue(vcapApplicationString, VcapApplication.class);

        CloudFoundryOperations client = cloudFoundryClient();

        ApplicationStats cfAppStats = client.getApplicationStats(application.getApplicationName());

        Limits limits = application.getLimits();
        if (limits == null) {
            limits = new Limits();
            limits.setDisk(-1);
            limits.setFds(-1);
            limits.setMem(-1);

        }

        ApplicationStatistics stats = new ApplicationStatistics();
        stats.setName(application.getApplicationName());
        stats.setId(application.getApplicationId());
        stats.setDiskLimit(limits.getDisk() * MEGS_TO_BYTES);
        stats.setRamLimit(limits.getMem() * MEGS_TO_BYTES);
        stats.setHost(application.getHost());
        stats.setPort(application.getPort());
        stats.setActiveInstance(application.getActiveInstanceIndex());

        addInstanceInfo(stats, cfAppStats);

        return stats;
    }

    private void addInstanceInfo(ApplicationStatistics statistics, ApplicationStats cfAppStats) {
        List<InstanceStats> infos = cfAppStats.getRecords();
        List<ApplicationInstance> appInstances = statistics.getInstances();

        if (infos != null) {
            statistics.setInstanceCount(infos.size());
        }

        for (int index = 0; index < infos.size(); ++index) {
            InstanceStats instance = infos.get(index);

            ApplicationInstance appInstance = new ApplicationInstance();
            appInstance.setState(instance.getState());

            Usage usage = instance.getUsage();

            if (usage != null) {
                appInstance.setCpu(usage.getCpu());
                appInstance.setDisk(usage.getDisk());
                appInstance.setMemory(usage.getMem());
            }

            appInstances.add(appInstance);
        }
    }

    private CloudFoundryOperations cloudFoundryClient() throws MalformedURLException {
        URL cfUrl = new URL(url);
        CloudCredentials creds = new CloudCredentials(cloudControllerUserId, cloudControllerPassword);

        CloudFoundryOperations client = new CloudFoundryClient(creds, cfUrl, trustSelfSignedCerts.booleanValue());

        return client;
    }
}