io.split.client.YamlLocalhostSplitFile Maven / Gradle / Ivy

Go to download
package io.split.client;

import com.google.common.base.Preconditions;
import com.google.common.collect.Maps;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.yaml.snakeyaml.Yaml;

import java.io.FileReader;
import java.io.IOException;
import java.util.List;
import java.util.Map;

public class YamlLocalhostSplitFile extends AbstractLocalhostSplitFile {

    private static final Logger _log = LoggerFactory.getLogger(YamlLocalhostSplitFile.class);

    public YamlLocalhostSplitFile(LocalhostSplitFactory localhostSplitFactory, String directory, String filenameYaml) throws IOException {
        super(localhostSplitFactory, directory, filenameYaml);
    }

    public Map readOnSplits() throws IOException {
        Map onSplits = Maps.newHashMap();
        try {

            Yaml yaml = new Yaml();
            List>> yamlSplits = yaml.load(new FileReader(_file));

            for(Map> aSplit : yamlSplits) {
                // The outter map is a map with one key, the split name
                Map.Entry> splitAndValues = aSplit.entrySet().iterator().next();

                SplitAndKey splitAndKey = null;
                String splitName = splitAndValues.getKey();
                String treatment = (String) splitAndValues.getValue().get("treatment");
                String configurations = splitAndValues.getValue().get("config") != null? (String) splitAndValues.getValue().get("config") : null;
                Object keyOrKeys = splitAndValues.getValue().get("keys");

                if (keyOrKeys == null) {
                    splitAndKey = SplitAndKey.of(splitName); // Key in this line is splitName
                    onSplits.put(splitAndKey, LocalhostSplit.of(treatment, configurations));
                } else {
                    if (keyOrKeys instanceof String) {
                        splitAndKey = SplitAndKey.of(splitName, (String) keyOrKeys);
                        onSplits.put(splitAndKey, LocalhostSplit.of(treatment, configurations));
                    } else {
                        Preconditions.checkArgument(keyOrKeys instanceof List, "'keys' is not a String nor a List.");
                        for (String aKey : (List) keyOrKeys) {
                            splitAndKey = SplitAndKey.of(splitName, aKey);
                            onSplits.put(splitAndKey, LocalhostSplit.of(treatment, configurations));
                        }
                    }
                }
            }
        } catch (Exception e) {
            _log.warn("There was no file named " + _file.getPath() + " found. " +
                    "We created a split client that returns default treatments for all features for all of your users. " +
                    "If you wish to return a specific treatment for a feature, enter the name of that feature name and " +
                    "treatment name separated by whitespace in " + _file.getPath() +
                    "; one pair per line. Empty lines or lines starting with '#' are considered comments", e);
        }

        return onSplits;
    }
}