org.swisspush.gateleen.routing.RuleFeaturesProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gateleen-routing Show documentation
The newest version!
package org.swisspush.gateleen.routing;

import com.google.common.collect.ImmutableMap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.*;
import java.util.regex.Pattern;

/**
 * Class RuleFeaturesProvider.
 * This class extracts features from {@link Rule} objects and collect them in a List of {@link RuleFeatures} objects.
 *
 * @author https://github.com/mcweba [Marc-Andre Weber]
 */
public final class RuleFeaturesProvider {

    private Logger log = LoggerFactory.getLogger(RuleFeaturesProvider.class);
    List featuresList;

    public RuleFeaturesProvider(List rules) {
        featuresList = collectRuleFeatures(rules);
    }

    public List getFeaturesList() {
        if(featuresList == null){
            featuresList = new ArrayList<>();
        }
        return featuresList;
    }

    /**
     * Checks the whether the provided uri matches the provided {@link RuleFeatures.Feature}.
     *
     * @param feature the feature to check against
     * @param uri the uri to check
     * @return returns true when the uri matches the feature, else returns false.
     */
    public boolean isFeatureRequest(RuleFeatures.Feature feature, String uri){
        for (RuleFeatures features : getFeaturesList()) {
            if(features.getUrlPattern().matcher(uri).matches()){
                if(features.hasFeature(feature)){
                    log.debug(String.format("Don't expand the uri: %s for the pattern %s since it's a %s request", uri, features.getUrlPattern().pattern(), feature.name()));
                    return true;
                }
                return false;
            }
        }
        return false;
    }

    private List collectRuleFeatures(List rules) {
        List featuresList = new ArrayList<>();
        for (Rule rule : rules) {
            boolean isStorageExpand = rule.isStorageExpand();
            boolean isExpandOnBackend = rule.isExpandOnBackend();
            boolean isDeltaOnBackend = rule.isDeltaOnBackend();
            try {
                Pattern pattern = Pattern.compile(rule.getUrlPattern());
                featuresList.add(new RuleFeatures(pattern, ImmutableMap.of(
                        RuleFeatures.Feature.STORAGE_EXPAND, isStorageExpand,
                        RuleFeatures.Feature.EXPAND_ON_BACKEND, isExpandOnBackend,
                        RuleFeatures.Feature.DELTA_ON_BACKEND, isDeltaOnBackend)));
                log.info(String.format("Collected features for rule url pattern %s storageExpand:%s expandOnBackend:%s deltaOnBackend:%s", rule.getUrlPattern(), isStorageExpand, isExpandOnBackend, isDeltaOnBackend));
            } catch (Exception e) {
                log.error("Could not compile the regex:{} to a pattern. Cannot collect feature information of this rule", rule.getUrlPattern());
            }
        }
        return featuresList;
    }
}