com.azure.cosmos.CosmosExcludedRegions Maven / Gradle / Ivy

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.azure.cosmos;

import com.azure.cosmos.implementation.guava25.collect.ImmutableSet;

import java.util.Collection;
import java.util.HashSet;
import java.util.Locale;
import java.util.Set;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

import static com.azure.cosmos.implementation.guava25.base.Preconditions.checkArgument;

/**
 * A class which encapsulates a set of excluded regions.
 * */
public final class CosmosExcludedRegions {
    private final Set excludedRegions;
    private final String excludedRegionsAsString;
    private static final Pattern SPACE_PATTERN = Pattern.compile(" ");

    /**
     * Instantiates {@code CosmosExcludedRegions}.
     *
     * @param excludedRegions the set of regions to exclude.
     * @throws IllegalArgumentException if {@code excludedRegions} is set as null.
     * */
    public CosmosExcludedRegions(Set excludedRegions) {

        checkArgument(excludedRegions != null, "excludedRegions cannot be set to null");

        this.excludedRegions = ImmutableSet.copyOf(excludedRegions);
        this.excludedRegionsAsString = stringifyExcludedRegions(this.excludedRegions);
    }

    /**
     * Gets the immutable set of excluded regions.
     *
     * @return an immutable set of excluded regions.
     * */
    public Set getExcludedRegions() {

        return this.excludedRegions;
    }

    @Override
    public String toString() {
        return this.excludedRegionsAsString;
    }

    private static String stringifyExcludedRegions(Set excludedRegions) {
        String substring = "";

        if (excludedRegions == null || excludedRegions.isEmpty()) {
            substring =  "";
        } else {
            substring = excludedRegions
                .stream()
                .map(r -> SPACE_PATTERN.matcher(r.toLowerCase(Locale.ROOT)).replaceAll(""))
                .collect(Collectors.joining(","));
        }

        return "[" + substring + "]";
    }
}