com.netflix.fabricator.archaius.ArchaiusComponentConfiguration Maven / Gradle / Ivy

Go to download
package com.netflix.fabricator.archaius;

import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;

import org.apache.commons.configuration.AbstractConfiguration;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Function;
import com.google.common.base.Joiner;
import com.google.common.base.Supplier;
import com.google.common.collect.Lists;
import com.netflix.config.ConfigurationManager;
import com.netflix.config.DynamicProperty;
import com.netflix.fabricator.ConfigurationNode;
import com.netflix.fabricator.properties.AbstractPropertiesComponentConfiguration;
import com.netflix.fabricator.supplier.ListenableSupplier;

public class ArchaiusComponentConfiguration extends AbstractPropertiesComponentConfiguration {
    private static final Logger LOG = LoggerFactory.getLogger(ArchaiusComponentConfiguration.class);
    
    private final AbstractConfiguration config;
    
    public static ArchaiusComponentConfiguration forPrefix(String prefix) {
        AbstractConfiguration config = ConfigurationManager.getConfigInstance();
        String type = config.getString(prefix + "type");
        String id = StringUtils.substringAfterLast(prefix, ".");
        return new ArchaiusComponentConfiguration(id, type, config, prefix);
    }
    
    public ArchaiusComponentConfiguration(String id, String type, AbstractConfiguration config, String prefix) {
        super(id, type, prefix);
        this.config = config;
    }

    public ArchaiusComponentConfiguration(String id, String type, AbstractConfiguration config) {
        super(id, type);
        this.config = config;
    }

    public static abstract class DynamicListenableSupplier implements ListenableSupplier {
        private final DynamicProperty prop;
        
        DynamicListenableSupplier(DynamicProperty prop) {
            this.prop = prop;
        }
        
        @Override
        public void onChange(final Function func) {
            prop.addCallback(new Runnable() {
                @Override
                public void run() {
                    func.apply(get());
                }
            });
        }
    }
    
    @SuppressWarnings("unchecked")
    @Override
    public  ListenableSupplier getDynamicValue(Class type) {
        final DynamicProperty prop = DynamicProperty.getInstance(getFullName());
        if ( String.class.isAssignableFrom(type) ) {
            return (ListenableSupplier) new DynamicListenableSupplier(prop) {
                @Override
                public String get() {
                    return prop.getString();
                }
            };
        }
        else if ( Boolean.class.isAssignableFrom(type) || 
                  Boolean.TYPE.isAssignableFrom(type) || 
                  boolean.class.equals(type)) {
            return (ListenableSupplier) new DynamicListenableSupplier(prop) {
                @Override
                public Boolean get() {
                    return prop.getBoolean();
                }
            };
        }
        else if ( Integer.class.isAssignableFrom(type) 
                  || Integer.TYPE.isAssignableFrom(type) 
                  || int.class.equals(type))
        {
            return (ListenableSupplier) new DynamicListenableSupplier(prop) {
                @Override
                public Integer get() {
                    return prop.getInteger();
                }
            };
        }
        else if ( Long.class.isAssignableFrom(type) 
                  || Long.TYPE.isAssignableFrom(type) 
                  || long.class.equals(type))
        {
            return (ListenableSupplier) new DynamicListenableSupplier(prop) {
                @Override
                public Long get() {
                    return prop.getLong();
                }
            };
        }
        else if ( Double.class.isAssignableFrom(type) 
                  || Double.TYPE.isAssignableFrom(type) 
                  || double.class.equals(type))
        {
            return (ListenableSupplier) new DynamicListenableSupplier(prop) {
                @Override
                public Double get() {
                    return prop.getDouble();
                }
            };
        }
        else if ( Properties.class.isAssignableFrom(type)) {
            return (ListenableSupplier) new DynamicListenableSupplier(prop) {
                @Override
                public Properties get() {
                    if (config.containsKey(getFullName())) {
                        throw new RuntimeException(getFullName() + " is not a root for a properties structure");
                    }
                    
                    String prefix = getFullName() + ".";
                    Properties result = new Properties();
                    for (String prop : Lists.newArrayList(config.getKeys(prefix))) {
                        if (prop.startsWith(prefix)) {
                            result.setProperty(prop.substring(prefix.length()), config.getProperty(prop).toString());
                        }
                    }
                    return result;
                }
            };
        }
        else {
            LOG.warn(String.format("Unknown type '%s' for property '%s'", type.getCanonicalName(), getFullName()));
        }

        return null;
    }

    @Override
    public ConfigurationNode getChild(String name) {
        String fullName = Joiner.on(".").skipNulls().join(getFullName(), name);
        
        return new ArchaiusComponentConfiguration(
                name, 
                config.getString(Joiner.on(".").skipNulls().join(fullName, "type")),
                config, 
                fullName);
    }

    @Override
    public boolean isSingle() {
        if (config.containsKey(getFullName())) {
            return true;
        }
        return false;   // TODO: Look for sub properties
    }

    @Override
    public boolean hasChild(String propertyName) {
        return true;
    }

    @Override
    public Set getUnknownProperties(Set supportedProperties) {
        // TODO:
        return Collections.emptySet();
    }

    @Override
    public  T getValue(Class type) {
        Supplier supplier = this.getDynamicValue(type);
        if (supplier != null)
            return supplier.get();
        return null;
    }
}