GroupSpec.java

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.facebook.presto.router.spec;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.collect.ImmutableList;

import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

import static java.util.Collections.nCopies;
import static java.util.Objects.requireNonNull;

public class GroupSpec
{
    private final String name;
    private List<URI> members;
    private final Optional<List<URI>> membersDiscoveryUri;
    private final Optional<List<Integer>> weights;

    @JsonCreator
    public GroupSpec(
            @JsonProperty("name") String name,
            @JsonProperty("members") List<URI> members,
            @JsonProperty("membersDiscoveryURI") Optional<List<URI>> membersDiscoveryUri,
            @JsonProperty("weights") Optional<List<Integer>> weights)
    {
        this.name = requireNonNull(name, "name is null");
        this.members = ImmutableList.copyOf(requireNonNull(members, "members is null"));
        this.membersDiscoveryUri = requireNonNull(membersDiscoveryUri, "members is null");
        this.weights = requireNonNull(weights, "weights is null");
    }

    @JsonProperty
    public String getName()
    {
        return name;
    }

    @JsonProperty
    public List<URI> getMembers()
    {
        return members;
    }

    @JsonProperty
    public List<URI> getMembersDiscoveryURI()
    {
        return membersDiscoveryUri.orElse(members);
    }

    @JsonProperty
    public List<Integer> getWeights()
    {
        return weights.orElseGet(() -> new ArrayList<>(nCopies(members.size(), 1)));
    }
}