com.vmware.photon.controller.model.PhotonModelServices Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2015-2016 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.model;

import com.vmware.photon.controller.model.monitoring.InMemoryResourceMetricService;
import com.vmware.photon.controller.model.monitoring.ResourceAggregateMetricService;
import com.vmware.photon.controller.model.monitoring.ResourceMetricsService;
import com.vmware.photon.controller.model.resources.ComputeDescriptionService;
import com.vmware.photon.controller.model.resources.ComputeService;
import com.vmware.photon.controller.model.resources.DiskService;
import com.vmware.photon.controller.model.resources.EndpointService;
import com.vmware.photon.controller.model.resources.FirewallService;
import com.vmware.photon.controller.model.resources.ImageService;
import com.vmware.photon.controller.model.resources.NetworkInterfaceDescriptionService;
import com.vmware.photon.controller.model.resources.NetworkInterfaceService;
import com.vmware.photon.controller.model.resources.NetworkService;
import com.vmware.photon.controller.model.resources.ResourceDescriptionService;
import com.vmware.photon.controller.model.resources.ResourceGroupService;
import com.vmware.photon.controller.model.resources.ResourcePoolService;
import com.vmware.photon.controller.model.resources.SecurityGroupService;
import com.vmware.photon.controller.model.resources.SnapshotService;
import com.vmware.photon.controller.model.resources.StorageDescriptionService;
import com.vmware.photon.controller.model.resources.SubnetService;
import com.vmware.photon.controller.model.resources.TagFactoryService;
import com.vmware.photon.controller.model.resources.TagService;
import com.vmware.xenon.common.Operation;
import com.vmware.xenon.common.Service;
import com.vmware.xenon.common.ServiceHost;
import com.vmware.xenon.common.ServiceStats;
import com.vmware.xenon.common.UriUtils;

/**
 * Helper class that starts all the photon model provisioning services
 */
public class PhotonModelServices {

    public static final String[] LINKS = {
            ComputeDescriptionService.FACTORY_LINK,
            ComputeService.FACTORY_LINK,
            ResourcePoolService.FACTORY_LINK,
            ResourceDescriptionService.FACTORY_LINK,
            DiskService.FACTORY_LINK,
            SnapshotService.FACTORY_LINK,
            NetworkInterfaceService.FACTORY_LINK,
            NetworkInterfaceDescriptionService.FACTORY_LINK,
            ResourceGroupService.FACTORY_LINK,
            NetworkService.FACTORY_LINK,
            SecurityGroupService.FACTORY_LINK,
            FirewallService.FACTORY_LINK,
            StorageDescriptionService.FACTORY_LINK,
            ResourceMetricsService.FACTORY_LINK,
            InMemoryResourceMetricService.FACTORY_LINK,
            EndpointService.FACTORY_LINK,
            ImageService.FACTORY_LINK,
            ResourceAggregateMetricService.FACTORY_LINK,
            TagService.FACTORY_LINK };

    public static void startServices(ServiceHost host) throws Throwable {

        host.startFactory(new ComputeDescriptionService());
        host.startFactory(new ComputeService());
        host.startFactory(new ResourcePoolService());
        host.startFactory(new ResourceDescriptionService());
        host.startFactory(new DiskService());
        host.startFactory(new SnapshotService());
        host.startFactory(new NetworkInterfaceService());
        host.startFactory(new NetworkInterfaceDescriptionService());
        host.startFactory(new SubnetService());
        host.startFactory(new ResourceGroupService());
        host.startFactory(new NetworkService());
        host.startFactory(new FirewallService());
        host.startFactory(new SecurityGroupService());
        host.startFactory(new StorageDescriptionService());
        host.startFactory(new EndpointService());
        host.startFactory(new ImageService());
        host.startFactory(new InMemoryResourceMetricService());
        host.startFactory(TagService.class, TagFactoryService::new);
        host.startFactory(ResourceMetricsService.class, ResourceMetricsService::createFactory);
        setFactoryToAvailable(host, ResourceMetricsService.FACTORY_LINK);
        host.startFactory(ResourceAggregateMetricService.class, ResourceAggregateMetricService::createFactory);
        setFactoryToAvailable(host, ResourceAggregateMetricService.FACTORY_LINK);
    }

    /** @see #setFactoryToAvailable(ServiceHost, String, Operation.CompletionHandler) */
    public static void setFactoryToAvailable(ServiceHost host, String factoryPath) {
        setFactoryToAvailable(host, factoryPath, null);
    }

    /**
     * Helper method to explicitly set a factory to be "available". This is usually unnecessary,
     * but currently factories that create {@code ON_DEMAND_LOAD} services are not being set to
     * available... and currently require this work-around.
     *
     * @param host the host
     * @param factoryPath the path of the factory to explicitly set to be available
     * @param handler an optional completion handler
     */
    public static void setFactoryToAvailable(ServiceHost host, String factoryPath, Operation.CompletionHandler handler) {
        ServiceStats.ServiceStat body = new ServiceStats.ServiceStat();
        body.name = Service.STAT_NAME_AVAILABLE;
        body.latestValue = Service.STAT_VALUE_TRUE;

        Operation put = Operation.createPut(UriUtils.buildAvailableUri(host, factoryPath))
                .setBody(body)
                .setCompletion(handler)
                .setReferer(host.getUri());
        host.sendRequest(put);
    }
}