io.cloudsoft.tosca.a4c.brooklyn.util.EntitySpecs Maven / Gradle / Ivy

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

import com.google.common.base.Optional;
import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
import com.google.common.collect.Iterables;
import org.apache.brooklyn.api.entity.Application;
import org.apache.brooklyn.api.entity.EntitySpec;
import org.apache.brooklyn.camp.brooklyn.BrooklynCampConstants;
import org.apache.brooklyn.config.ConfigKey;
import org.apache.brooklyn.util.guava.SerializablePredicate;

import javax.annotation.Nullable;

public class EntitySpecs {

    public static EntitySpec findChildEntitySpecByPlanId(EntitySpec app, String planId){

        Optional> result = Iterables.tryFind(app.getChildren(),
                configSatisfies(BrooklynCampConstants.PLAN_ID, planId));

        if (result.isPresent()) {
            return result.get();
        }
        //TODO: better NoSuchElementException? return null?
        throw new IllegalStateException("Entity with planId  " + planId + " is not contained on" +
                " ApplicationSpec "+ app +" children");
    }

    public static  Predicate configSatisfies(final ConfigKey configKey, final T val) {
        return new ConfigKeySatisfies(configKey, Predicates.equalTo(val));
    }

    protected static class ConfigKeySatisfies implements SerializablePredicate {
        protected final ConfigKey configKey;
        protected final Predicate condition;

        private ConfigKeySatisfies(ConfigKey configKey, Predicate condition) {
            this.configKey = configKey;
            this.condition = condition;
        }

        @Override
        @SuppressWarnings("unchecked")
        public boolean apply(@Nullable EntitySpec input) {
            return (input != null) && condition.apply((T)input.getConfig().get(configKey));
        }

        @Override
        public String toString() {
            return "configKeySatisfies("+configKey.getName()+","+condition+")";
        }
    }
}