no.finn.unleash.strategy.RemoteAddressStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of unleash-client-java Show documentation

A client library for Unleash

There is a newer version: 4.4.1

package no.finn.unleash.strategy;

import no.finn.unleash.UnleashContext;
import no.finn.unleash.util.IpAddressMatcher;

import java.util.Arrays;
import java.util.Map;
import java.util.Optional;
import java.util.regex.Pattern;
import java.util.stream.Stream;

public final class RemoteAddressStrategy implements Strategy {
    static final String PARAM = "IPs";
    private static final String STRATEGY_NAME = "remoteAddress";
    private static final Pattern SPLITTER = Pattern.compile(",");

    @Override
    public String getName() {
        return STRATEGY_NAME;
    }

    @Override
    public boolean isEnabled(Map parameters) {
        return false;
    }

    @Override
    public boolean isEnabled(Map parameters, UnleashContext context) {
        return Optional.ofNullable(parameters.get(PARAM))
                .map(ips -> Arrays.asList(SPLITTER.split(ips, -1)))
                .map(ips -> ips.stream()
                        .flatMap(ipAddress -> buildIpAddressMatcher(ipAddress)
                                .map(Stream::of)
                                .orElseGet(Stream::empty))
                        .map(subnet -> context.getRemoteAddress()
                                .map(subnet::matches)
                                .orElse(false))
                        .anyMatch(Boolean.TRUE::equals))
                .orElse(false);
    }

    private Optional buildIpAddressMatcher(String ipAddress) {
        try {
            return Optional.of(new IpAddressMatcher(ipAddress));
        } catch (IllegalArgumentException ex) {
            return Optional.empty();
        }
    }
}