io.cloudsoft.tosca.a4c.brooklyn.spec.Alien4CloudEntitySpecFactory Maven / Gradle / Ivy

Go to download
package io.cloudsoft.tosca.a4c.brooklyn.spec;

import javax.inject.Inject;

import org.apache.brooklyn.api.catalog.CatalogItem;
import org.apache.brooklyn.api.entity.Entity;
import org.apache.brooklyn.api.entity.EntitySpec;
import org.apache.brooklyn.api.mgmt.ManagementContext;
import org.apache.brooklyn.core.catalog.internal.CatalogUtils;
import org.apache.brooklyn.core.entity.Attributes;
import org.apache.brooklyn.core.sensor.Sensors;
import org.apache.brooklyn.enricher.stock.Enrichers;
import org.apache.brooklyn.entity.software.base.SameServerEntity;
import org.apache.brooklyn.entity.software.base.SoftwareProcess;
import org.apache.brooklyn.entity.software.base.VanillaSoftwareProcess;
import org.apache.brooklyn.entity.stock.BasicApplication;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import com.google.common.base.Functions;

import io.cloudsoft.tosca.a4c.brooklyn.Alien4CloudApplication;
import io.cloudsoft.tosca.a4c.brooklyn.Alien4CloudFacade;
import io.cloudsoft.tosca.a4c.brooklyn.ToscaApplication;
import io.cloudsoft.tosca.a4c.brooklyn.ToscaFacade;

@Component
public class Alien4CloudEntitySpecFactory implements EntitySpecFactory {

    private static final Logger LOG = LoggerFactory.getLogger(Alien4CloudEntitySpecFactory.class);

    private final ManagementContext mgmt;
    private ToscaFacade alien4CloudFacade;

    @Inject
    public Alien4CloudEntitySpecFactory(ManagementContext mgmt, ToscaFacade alien4CloudFacade) {
        this.mgmt = mgmt;
        this.alien4CloudFacade = alien4CloudFacade;
    }

    @Override
    public EntitySpec create(String nodeId, Alien4CloudApplication toscaApplication) {
        // TODO: decide on how to behave if indexedNodeTemplate.getElementId is abstract.
        // Currently we create a VanillaSoftwareProcess.

        EntitySpec spec;
        String type = toscaApplication.getNodeTemplate(nodeId).getType();
        CatalogItem catalogItem = CatalogUtils.getCatalogItemOptionalVersion(mgmt, type);
        if (catalogItem != null) {
            LOG.info("Found Brooklyn catalog item that match node type: " + type);
            spec = (EntitySpec) mgmt.getCatalog().createSpec(catalogItem);

        } else if (isComputeType(nodeId, toscaApplication)) {
            spec = EntitySpec.create(SameServerEntity.class);
        } else {
            try {
                LOG.info("Found Brooklyn entity that match node type: " + type);
                spec = EntitySpec.create((Class) Class.forName(type));

            } catch (ClassNotFoundException e) {
                LOG.info("Cannot find any Brooklyn catalog item nor Brooklyn entities that match node type: " +
                        type + ". Defaulting to a VanillaSoftwareProcess");
                spec = EntitySpec.create(VanillaSoftwareProcess.class);
            }
        }

        return spec;
    }

    private boolean isComputeType(String nodeId, Alien4CloudApplication toscaApplication) {
        return alien4CloudFacade.isDerivedFrom(nodeId, toscaApplication, Alien4CloudFacade.COMPUTE_TYPE);
    }
}