blob: 36ca052d9ab59b027f6d047fc8f9307e83781dea (
plain) (
blame)
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.modelfactory;
import com.yahoo.collections.Pair;
import com.yahoo.component.Version;
import com.yahoo.config.provision.AllocatedHosts;
import com.yahoo.config.provision.ClusterSpec;
import com.yahoo.config.provision.HostSpec;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
/**
* Different model versions being built may disagree on the set of hosts that should be allocated.
* This keeps track of the right set of hosts to finally allocate given such disagreement:
* The superset of hosts allocated on all model versions, with the spec of the newest version allocating it.
*
* @author bratseth
*/
public class AllocatedHostsFromAllModels {
/** All hosts with the newest model version that allocated it, indexed on hostname */
private final Map<String, Pair<HostSpec, Version>> hosts = new LinkedHashMap<>();
/** Adds the nodes allocated for a particular model version. */
public void add(AllocatedHosts allocatedHosts, Version version) {
for (var newHost : allocatedHosts.getHosts()) {
var presentHost = hosts.get(newHost.hostname());
if (presentHost == null || version.isAfter(presentHost.getSecond()))
hosts.put(newHost.hostname(), new Pair<>(newHost, version));
}
}
public AllocatedHosts toAllocatedHosts() {
// Preserve add order for tests
Set<HostSpec> hostSet = new LinkedHashSet<>();
for (var host : hosts.values())
hostSet.add(host.getFirst());
return AllocatedHosts.withHosts(hostSet);
}
}
|