core/store/dist/src/main/java/org/onosproject/store/consistent/impl/DatabaseDefinition.java

/*
 * Copyright 2015 Open Networking Laboratory
 *
 * 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 org.onosproject.store.consistent.impl;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Maps;
import org.onosproject.store.cluster.impl.NodeInfo;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

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

/**
 * Partitioned database configuration.
 */
public class DatabaseDefinition {
    private Map<String, Set<NodeInfo>> partitions;
    private Set<NodeInfo> nodes;

    /**
     * Creates a new DatabaseDefinition.
     *
     * @param partitions partition map
     * @param nodes      set of nodes
     * @return database definition
     */
    public static DatabaseDefinition from(Map<String, Set<NodeInfo>> partitions,
                                          Set<NodeInfo> nodes) {
        checkNotNull(partitions);
        checkNotNull(nodes);
        DatabaseDefinition definition = new DatabaseDefinition();
        definition.partitions = ImmutableMap.copyOf(partitions);
        definition.nodes = ImmutableSet.copyOf(nodes);
        return definition;
    }

    /**
     * Creates a new DatabaseDefinition using default partitions.
     *
     * @param nodes set of nodes
     * @return database definition
     */
    public static DatabaseDefinition from(Set<NodeInfo> nodes) {
        return from(generateDefaultPartitions(nodes), nodes);
    }

    /**
     * Returns the map of database partitions.
     *
     * @return db partition map
     */
    public Map<String, Set<NodeInfo>> getPartitions() {
        return partitions;
    }

    /**
     * Returns the set of nodes.
     *
     * @return nodes
     */
    public Set<NodeInfo> getNodes() {
        return nodes;
    }


    /**
     * Generates set of default partitions using permutations of the nodes.
     *
     * @param nodes information about cluster nodes
     * @return default partition map
     */
    private static Map<String, Set<NodeInfo>> generateDefaultPartitions(Set<NodeInfo> nodes) {
        List<NodeInfo> sorted = new ArrayList<>(nodes);
        Collections.sort(sorted, (o1, o2) -> o1.getId().compareTo(o2.getId()));
        Map<String, Set<NodeInfo>> partitions = Maps.newHashMap();

        int length = nodes.size();
        int count = 3;
        for (int i = 0; i < length; i++) {
            Set<NodeInfo> set = new HashSet<>(count);
            for (int j = 0; j < count; j++) {
                set.add(sorted.get((i + j) % length));
            }
            partitions.put("p" + (i + 1), set);
        }
        return partitions;
    }

}