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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.api.integration.configserver;
import com.yahoo.component.Version;
import com.yahoo.config.provision.NodeType;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
/**
* Target versions for node upgrades in a zone.
*
* @author mpolden
*/
public class TargetVersions {
public static final TargetVersions EMPTY = new TargetVersions(Map.of(), Map.of());
private final Map<NodeType, Version> vespaVersions;
private final Map<NodeType, Version> osVersions;
public TargetVersions(Map<NodeType, Version> vespaVersions, Map<NodeType, Version> osVersions) {
this.vespaVersions = Map.copyOf(Objects.requireNonNull(vespaVersions, "vespaVersions must be non-null"));
this.osVersions = Map.copyOf(Objects.requireNonNull(osVersions, "osVersions must be non-null"));
}
/** Returns a copy of this with Vespa version set for given node type */
public TargetVersions withVespaVersion(NodeType nodeType, Version version) {
var vespaVersions = new HashMap<>(this.vespaVersions);
vespaVersions.put(nodeType, version);
return new TargetVersions(vespaVersions, osVersions);
}
/** Returns a copy of this with OS version set for given node type */
public TargetVersions withOsVersion(NodeType nodeType, Version version) {
var osVersions = new HashMap<>(this.osVersions);
osVersions.put(nodeType, version);
return new TargetVersions(vespaVersions, osVersions);
}
/** Returns the target OS version of given node type, if any */
public Optional<Version> osVersion(NodeType type) {
return Optional.ofNullable(osVersions.get(type));
}
/** Returns the target Vespa version of given node type, if any */
public Optional<Version> vespaVersion(NodeType type) {
return Optional.ofNullable(vespaVersions.get(type));
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
TargetVersions that = (TargetVersions) o;
return vespaVersions.equals(that.vespaVersions) &&
osVersions.equals(that.osVersions);
}
@Override
public int hashCode() {
return Objects.hash(vespaVersions, osVersions);
}
}
|
