node-repository/src/main/java/com/yahoo/vespa/hosted/provision/node/filter/CloudAccountFilter.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

package com.yahoo.vespa.hosted.provision.node.filter;

import com.yahoo.config.provision.CloudAccount;
import com.yahoo.vespa.hosted.provision.Node;

import java.util.Collection;
import java.util.Objects;
import java.util.function.Predicate;

/**
 * Filters nodes based on their cloud account.
 *
 * @author gjoranv
 */
public class CloudAccountFilter {

    private CloudAccountFilter() { }

    /** Creates a node filter which removes the nodes from the given cloud accounts */
    public static Predicate<Node> from(Collection<CloudAccount> unwantedAccounts, boolean enabled) {
        Objects.requireNonNull(unwantedAccounts, "unwantedAccounts cannot be null");
        return node -> {
            if (unwantedAccounts.isEmpty()) return true;
            if (! enabled) return true;
            return ! unwantedAccounts.contains(node.cloudAccount());
        };
    }

}