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.clustercontroller.core.restapiv2.requests;
import ai.vespa.metrics.StorageMetrics;
import com.yahoo.vespa.clustercontroller.core.NodeInfo;
import com.yahoo.vespa.clustercontroller.core.RemoteClusterControllerTask;
import com.yahoo.vespa.clustercontroller.core.hostinfo.Metrics;
import com.yahoo.vespa.clustercontroller.core.restapiv2.Id;
import com.yahoo.vespa.clustercontroller.core.restapiv2.Request;
import com.yahoo.vespa.clustercontroller.core.restapiv2.Response;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.errors.MissingResourceException;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.errors.StateRestApiException;
public class NodeStateRequest extends Request<Response.NodeResponse> {
private final Id.Node id;
public NodeStateRequest(Id.Node id) {
super(MasterState.MUST_BE_MASTER);
this.id = id;
}
@Override
public Response.NodeResponse calculateResult(RemoteClusterControllerTask.Context context) throws StateRestApiException {
Response.NodeResponse result = new Response.NodeResponse();
NodeInfo info = context.cluster.getNodeInfo(id.getNode());
if (info == null) {
throw new MissingResourceException("node " + id.getNode());
}
if (info.getGroup() != null) {
result.addAttribute("hierarchical-group", info.getGroup().getPath());
}
result.addState("generated", new Response.UnitStateImpl(context.currentConsolidatedState.getNodeState(id.getNode())));
result.addState("unit", new Response.UnitStateImpl(info.getReportedState()));
result.addState("user", new Response.UnitStateImpl(info.getWantedState()));
if (info.isStorage()) {
fillInMetrics(context.cluster.getNodeInfo(id.getNode()).getHostInfo().getMetrics(), result);
}
return result;
}
private static void fillInMetrics(Metrics metrics, Response.NodeResponse result) {
for (Metrics.Metric metric: metrics.getMetrics()) {
fillInMetricValue(metric.getName(), metric.getValue(), result);
}
}
private static void fillInMetricValue(String name, Metrics.Value value, Response.NodeResponse result) {
if (name.equals(StorageMetrics.VDS_DATASTORED_ALLDISKS_DOCS.baseName())) {
if (value.getLast() == null) {
return;
}
result.addMetric("unique-document-count", value.getLast());
} else if (name.equals(StorageMetrics.VDS_DATASTORED_ALLDISKS_BYTES.baseName())) {
if (value.getLast() == null) {
return;
}
result.addMetric("unique-document-total-size", value.getLast());
} else if (name.equals(StorageMetrics.VDS_DATASTORED_ALLDISKS_BUCKETS.baseName())) {
if (value.getLast() == null) {
return;
}
result.addMetric("bucket-count", value.getLast());
}
}
}
|
