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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.node.admin.configserver.noderepository;
import com.fasterxml.jackson.databind.JsonNode;
import com.yahoo.component.Version;
import com.yahoo.config.provision.DockerImage;
import com.yahoo.config.provision.WireguardKey;
import com.yahoo.vespa.hosted.node.admin.configserver.noderepository.reports.BaseReport;
import java.time.Instant;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import java.util.TreeMap;
import java.util.stream.Collectors;
import java.util.stream.Stream;
/**
* A node in the node repository is modified by setting which attributes to modify in this class,
* and then patching the node repository node through {@link NodeRepository#updateNodeAttributes(String, NodeAttributes)}.
*
* @author Haakon Dybdahl
* @author Valerij Fredriksen
*/
public class NodeAttributes {
private Optional<String> hostId = Optional.empty();
private Optional<Long> restartGeneration = Optional.empty();
private Optional<Long> rebootGeneration = Optional.empty();
private Optional<DockerImage> dockerImage = Optional.empty();
private Optional<Version> vespaVersion = Optional.empty();
private Optional<Version> currentOsVersion = Optional.empty();
private Optional<Instant> currentFirmwareCheck = Optional.empty();
private List<TrustStoreItem> trustStore = List.of();
private Optional<WireguardKey> wireguardPubkey = Optional.empty();
/** The list of reports to patch. A null value is used to remove the report. */
private Map<String, JsonNode> reports = new TreeMap<>();
public NodeAttributes() { }
public NodeAttributes withHostId(String hostId) {
this.hostId = Optional.of(hostId);
return this;
}
public NodeAttributes withRestartGeneration(Optional<Long> restartGeneration) {
this.restartGeneration = restartGeneration;
return this;
}
public NodeAttributes withRestartGeneration(long restartGeneration) {
return withRestartGeneration(Optional.of(restartGeneration));
}
public NodeAttributes withRebootGeneration(long rebootGeneration) {
this.rebootGeneration = Optional.of(rebootGeneration);
return this;
}
public NodeAttributes withDockerImage(DockerImage dockerImage) {
this.dockerImage = Optional.of(dockerImage);
return this;
}
public NodeAttributes withVespaVersion(Version vespaVersion) {
this.vespaVersion = Optional.of(vespaVersion);
return this;
}
public NodeAttributes withCurrentOsVersion(Version currentOsVersion) {
this.currentOsVersion = Optional.of(currentOsVersion);
return this;
}
public NodeAttributes withCurrentFirmwareCheck(Instant currentFirmwareCheck) {
this.currentFirmwareCheck = Optional.of(currentFirmwareCheck);
return this;
}
public NodeAttributes withTrustStore(List<TrustStoreItem> trustStore) {
this.trustStore = List.copyOf(trustStore);
return this;
}
public NodeAttributes withWireguardPubkey(WireguardKey wireguardPubkey) {
this.wireguardPubkey = Optional.of(wireguardPubkey);
return this;
}
public NodeAttributes withReports(Map<String, JsonNode> nodeReports) {
this.reports = new TreeMap<>(nodeReports);
return this;
}
public NodeAttributes withReport(String reportId, JsonNode jsonNode) {
reports.put(reportId, jsonNode);
return this;
}
public NodeAttributes withReportRemoved(String reportId) {
reports.put(reportId, null);
return this;
}
public Optional<String> getHostId() {
return hostId;
}
public Optional<Long> getRestartGeneration() {
return restartGeneration;
}
public Optional<Long> getRebootGeneration() {
return rebootGeneration;
}
public Optional<DockerImage> getDockerImage() {
return dockerImage;
}
public Optional<Version> getVespaVersion() {
return vespaVersion;
}
public Optional<Version> getCurrentOsVersion() {
return currentOsVersion;
}
public Optional<Instant> getCurrentFirmwareCheck() {
return currentFirmwareCheck;
}
public List<TrustStoreItem> getTrustStore() {
return trustStore;
}
public Optional<WireguardKey> getWireguardPubkey() { return wireguardPubkey; }
public Map<String, JsonNode> getReports() {
return reports;
}
public <T extends BaseReport> Optional<T> getReport(String reportId, Class<T> classInstance) {
return Optional.ofNullable(reports.get(reportId)).map(jn -> BaseReport.fromJsonNode(jn, classInstance));
}
@Override
public int hashCode() {
return Objects.hash(hostId, restartGeneration, rebootGeneration, dockerImage, vespaVersion, currentOsVersion,
currentFirmwareCheck, trustStore, wireguardPubkey, reports);
}
public boolean isEmpty() {
return equals(new NodeAttributes());
}
@Override
public boolean equals(Object o) {
if (!(o instanceof NodeAttributes other)) {
return false;
}
return Objects.equals(hostId, other.hostId)
&& Objects.equals(restartGeneration, other.restartGeneration)
&& Objects.equals(rebootGeneration, other.rebootGeneration)
&& Objects.equals(dockerImage, other.dockerImage)
&& Objects.equals(vespaVersion, other.vespaVersion)
&& Objects.equals(currentOsVersion, other.currentOsVersion)
&& Objects.equals(currentFirmwareCheck, other.currentFirmwareCheck)
&& Objects.equals(trustStore, other.trustStore)
&& Objects.equals(wireguardPubkey, other.wireguardPubkey)
&& Objects.equals(reports, other.reports);
}
@Override
public String toString() {
return Stream.of(hostId.map(id -> "hostId=" + id),
restartGeneration.map(gen -> "restartGeneration=" + gen),
rebootGeneration.map(gen -> "rebootGeneration=" + gen),
dockerImage.map(img -> "dockerImage=" + img.asString()),
vespaVersion.map(ver -> "vespaVersion=" + ver.toFullString()),
currentOsVersion.map(ver -> "currentOsVersion=" + ver.toFullString()),
currentFirmwareCheck.map(at -> "currentFirmwareCheck=" + at),
Optional.ofNullable(trustStore.isEmpty() ? null : "trustStore=" + trustStore),
Optional.ofNullable(wireguardPubkey.isEmpty() ? null : "wireguardPubkey=" + wireguardPubkey),
Optional.ofNullable(reports.isEmpty() ? null : "reports=" + reports))
.filter(Optional::isPresent)
.map(Optional::get)
.collect(Collectors.joining(", ", "{", "}"));
}
}
|
