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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.clustercontroller.utils.staterestapi.server;
import com.yahoo.vespa.clustercontroller.utils.communication.http.HttpRequest;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.errors.InvalidContentException;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.requests.SetUnitStateRequest;
import com.yahoo.vespa.clustercontroller.utils.staterestapi.response.UnitState;
import org.codehaus.jettison.json.JSONArray;
import org.codehaus.jettison.json.JSONObject;
import java.time.Duration;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
public class JsonReader {
public static final Duration DEFAULT_TIMEOUT = Duration.ofSeconds(30);
public static final Duration MAX_TIMEOUT = Duration.ofHours(1);
private static final long MICROS_IN_SECOND = TimeUnit.SECONDS.toMillis(1);
private static class UnitStateImpl implements UnitState {
private final String id;
private final String reason;
public UnitStateImpl(String id, String reason) {
this.id = id;
this.reason = reason;
}
@Override
public String getId() {
return id;
}
@Override
public String getReason() {
return reason;
}
}
static class SetRequestData {
final Map<String, UnitState> stateMap;
final SetUnitStateRequest.Condition condition;
final SetUnitStateRequest.ResponseWait responseWait;
final Optional<Duration> timeout;
public SetRequestData(Map<String, UnitState> stateMap,
SetUnitStateRequest.Condition condition,
SetUnitStateRequest.ResponseWait responseWait,
Optional<Duration> timeout) {
this.stateMap = stateMap;
this.condition = condition;
this.responseWait = responseWait;
this.timeout = timeout;
}
}
public SetRequestData getStateRequestData(HttpRequest request) throws Exception {
JSONObject json = new JSONObject(request.getPostContent().toString());
final SetUnitStateRequest.Condition condition;
if (json.has("condition")) {
condition = SetUnitStateRequest.Condition.fromString(json.getString("condition"));
} else {
condition = SetUnitStateRequest.Condition.FORCE;
}
final SetUnitStateRequest.ResponseWait responseWait = json.has("response-wait")
? SetUnitStateRequest.ResponseWait.fromString(json.getString("response-wait"))
: SetUnitStateRequest.ResponseWait.WAIT_UNTIL_CLUSTER_ACKED;
Map<String, UnitState> stateMap = new HashMap<>();
if (!json.has("state")) {
throw new InvalidContentException("Set state requests must contain a state object");
}
Object o = json.get("state");
if (!(o instanceof JSONObject)) {
throw new InvalidContentException("value of state is not a json object");
}
JSONObject state = (JSONObject) o;
JSONArray stateTypes = state.names();
for (int i=0; i<stateTypes.length(); ++i) {
o = stateTypes.get(i);
String type = (String) o;
o = state.get(type);
if (!(o instanceof JSONObject)) {
throw new InvalidContentException("value of state->" + type + " is not a json object");
}
JSONObject userState = (JSONObject) o;
String code = "up";
if (userState.has("state")) {
o = userState.get("state");
if (!(o instanceof String)) {
throw new InvalidContentException("value of state->" + type + "->state is not a string");
}
code = o.toString();
}
String reason = "";
if (userState.has("reason")) {
o = userState.get("reason");
if (!(o instanceof String)) {
throw new InvalidContentException("value of state->" + type + "->reason is not a string");
}
reason = o.toString();
}
stateMap.put(type, new UnitStateImpl(code, reason));
}
final Optional<Duration> timeout = parseTimeout(request.getOption("timeout", null));
return new SetRequestData(stateMap, condition, responseWait, timeout);
}
public static Optional<Duration> parseTimeout(String timeoutOption) throws InvalidContentException {
if (timeoutOption == null) {
return Optional.empty();
} else {
float timeoutSeconds;
try {
timeoutSeconds = Float.parseFloat(timeoutOption);
} catch (NumberFormatException e) {
throw new InvalidContentException("value of timeout->" + timeoutOption + " is not a float");
}
if (timeoutSeconds <= 0.0) {
return Optional.of(Duration.ZERO);
} else if (timeoutSeconds <= MAX_TIMEOUT.getSeconds()) {
long micros = Math.round(timeoutSeconds * MICROS_IN_SECOND);
long nanoAdjustment = TimeUnit.MILLISECONDS.toNanos(micros % MICROS_IN_SECOND);
return Optional.of(Duration.ofSeconds(micros / MICROS_IN_SECOND, nanoAdjustment));
} else {
throw new InvalidContentException("value of timeout->" + timeoutOption + " exceeds max timeout " + MAX_TIMEOUT);
}
}
}
}
|
