com.yahoo.vespa.config.server.application.PendingRestarts Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of configserver Show documentation

There is a newer version: 8.441.21

package com.yahoo.vespa.config.server.application;

import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;

import static java.util.Collections.unmodifiableMap;
import static java.util.Collections.unmodifiableSet;

/**
 * @author jonmv
 */
public class PendingRestarts {

    private static final PendingRestarts empty = new PendingRestarts(Map.of());

    private final Map> generationsForRestarts;

    public static PendingRestarts empty() {
        return empty;
    }

    PendingRestarts(Map> generationsForRestarts) {
        Map> builder = new LinkedHashMap<>();
        generationsForRestarts.forEach((generation, hostnames) -> builder.put(generation, unmodifiableSet(new LinkedHashSet<>(hostnames))));
        this.generationsForRestarts = unmodifiableMap(builder);
    }

    public Map> generationsForRestarts() { return generationsForRestarts; }

    public boolean isEmpty() { return generationsForRestarts.isEmpty(); }

    public PendingRestarts withRestarts(long atGeneration, Collection hostnames) {
        Map> newRestarts = new LinkedHashMap<>(generationsForRestarts);
        newRestarts.put(atGeneration, new LinkedHashSet<>(newRestarts.getOrDefault(atGeneration, Set.of())) {{ addAll(hostnames); }});
        return new PendingRestarts(newRestarts);
    }

    public PendingRestarts withoutPreviousGenerations(long generation) {
        Map> newRestarts = new LinkedHashMap<>(generationsForRestarts);
        newRestarts.keySet().removeIf(g -> g <= generation);
        return new PendingRestarts(newRestarts);
    }

    public Set hostnames() {
        return restartsReadyAt(Long.MAX_VALUE);
    }

    public Set restartsReadyAt(long generation) {
        LinkedHashSet ready = new LinkedHashSet<>();
        generationsForRestarts.forEach((g, hosts) -> { if (g <= generation) ready.addAll(hosts); });
        return ready;
    }

}