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

Go to download
package io.split.client;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import io.split.client.api.SplitView;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeoutException;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * An implementation of SplitClient that considers all partitions
 * passed in the constructor to be 100% on for all users, and
 * any other split to be 100% off for all users. This implementation
 * is useful for using Codigo in localhost environment.
 *
 * @author adil
 */
public final class LocalhostSplitManager implements SplitManager {

    private Map _splitAndKeyToTreatmentMap;
    private Map> _splitToTreatmentsMap;

    public static LocalhostSplitManager of(Map featureToTreatmentMap) {
        checkNotNull(featureToTreatmentMap, "featureToTreatmentMap must not be null");
        return new LocalhostSplitManager(featureToTreatmentMap, splitsToTreatments(featureToTreatmentMap));
    }

    private static Map> splitsToTreatments(Map splitAndKeyStringMap) {
        Map> splitsToTreatments = Maps.newHashMap();
        for (Map.Entry entry : splitAndKeyStringMap.entrySet()) {
            String split = entry.getKey().split();
            if (!splitsToTreatments.containsKey(split)) {
                splitsToTreatments.put(split, Sets.newHashSet());
            }
            Set treatments = splitsToTreatments.get(split);
            treatments.add(entry.getValue().treatment);
        }
        return splitsToTreatments;
    }

    private LocalhostSplitManager(Map featureToTreatmentMap,  Map> splitToTreatmentsMap) {
        checkNotNull(featureToTreatmentMap, "featureToTreatmentMap must not be null");
        _splitAndKeyToTreatmentMap = featureToTreatmentMap;
        _splitToTreatmentsMap = splitToTreatmentsMap;
    }

    @Override
    public List splits() {
        List result = new ArrayList();

        for (Map.Entry> entry : _splitToTreatmentsMap.entrySet()) {
            result.add(toSplitView(entry.getKey(), entry.getValue()));
        }

        return result;
    }

    @Override
    public List splitNames() {
        Set splits = Sets.newHashSet();
        for (Map.Entry entry : _splitAndKeyToTreatmentMap.entrySet()) {
            splits.add(entry.getKey().split());
        }
        return Lists.newArrayList(splits);
    }

    @Override
    public void blockUntilReady() throws TimeoutException, InterruptedException {
        // It is always ready
    }

    @Override
    public SplitView split(String featureName) {
        if (!_splitToTreatmentsMap.containsKey(featureName)) {
            return null;
        }

        return toSplitView(featureName, _splitToTreatmentsMap.get(featureName));
    }

    void updateFeatureToTreatmentMap(Map featureToTreatmentMap) {
        checkNotNull(featureToTreatmentMap, "featureToTreatmentMap must not be null");
        _splitAndKeyToTreatmentMap = featureToTreatmentMap;
        _splitToTreatmentsMap = splitsToTreatments(_splitAndKeyToTreatmentMap);
    }


    private SplitView toSplitView(String featureName, Set treatments) {
        SplitView view = new SplitView();
        view.name = featureName;
        view.killed = false;
        view.trafficType = null;
        view.changeNumber = 0;
        view.treatments = new ArrayList();
        if (treatments != null) {
            view.treatments.addAll(treatments);
        }

        return view;
    }


}