blob: 79e6a91f5612dff017934afe87fa2fc659b989bb (
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
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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.clustercontroller.core.status;
import com.yahoo.vespa.clustercontroller.core.status.statuspage.StatusPageResponse;
import com.yahoo.vespa.clustercontroller.core.status.statuspage.StatusPageServer;
import com.yahoo.vespa.clustercontroller.core.status.statuspage.StatusPageServerInterface;
import com.yahoo.vespa.clustercontroller.utils.communication.http.HttpRequest;
import com.yahoo.vespa.clustercontroller.utils.communication.http.HttpRequestHandler;
import com.yahoo.vespa.clustercontroller.utils.communication.http.HttpResult;
import java.io.IOException;
import java.io.StringWriter;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.logging.Logger;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class StatusHandler implements HttpRequestHandler {
private final static Logger log = Logger.getLogger(StatusHandler.class.getName());
public interface ClusterStatusPageServerSet {
ContainerStatusPageServer get(String cluster);
Map<String, ContainerStatusPageServer> getAll();
}
public static class ContainerStatusPageServer implements StatusPageServerInterface {
StatusPageServer.HttpRequest request;
StatusPageResponse response;
// Ensure only one use the server at a time
private final Object queueMonitor = new Object();
// Lock safety with fleetcontroller. Wait until completion
private final Object answerMonitor = new Object();
@Override
public int getPort() { return 0; }
@Override
public void shutdown() throws InterruptedException, IOException {}
@Override
public void setPort(int port) {}
@Override
public StatusPageServer.HttpRequest getCurrentHttpRequest() {
synchronized (answerMonitor) {
StatusPageServer.HttpRequest r = request;
request = null;
return r;
}
}
@Override
public void answerCurrentStatusRequest(StatusPageResponse r) {
synchronized (answerMonitor) {
response = r;
answerMonitor.notify();
}
}
StatusPageResponse getStatus(StatusPageServer.HttpRequest req) throws InterruptedException {
synchronized (queueMonitor) {
synchronized (answerMonitor) {
request = req;
while (response == null) {
answerMonitor.wait();
}
StatusPageResponse res = response;
response = null;
return res;
}
}
}
}
private static final Pattern clusterListRequest = Pattern.compile("^/clustercontroller-status/v1/?$");
private static final Pattern statusRequest = Pattern.compile("^/clustercontroller-status/v1/([^/]+)(/.*)?$");
private final ClusterStatusPageServerSet statusClusters;
public StatusHandler(ClusterStatusPageServerSet set) {
statusClusters = set;
}
@Override
public HttpResult handleRequest(HttpRequest httpRequest) throws Exception {
log.fine("Handling status request " + httpRequest);
Matcher matcher = statusRequest.matcher(httpRequest.getPath());
if (matcher.matches()) {
return handleClusterRequest(matcher.group(1), matcher.group(2));
}
matcher = clusterListRequest.matcher(httpRequest.getPath());
if (matcher.matches()) {
return handleClusterListRequest();
}
return new HttpResult().setHttpCode(
404, "No page for request '" + httpRequest.getPath() + "'.");
}
private HttpResult handleClusterRequest(String clusterName, String fleetControllerPath) throws InterruptedException {
ContainerStatusPageServer statusServer = statusClusters.get(clusterName);
if (statusServer == null) {
return new HttpResult().setHttpCode(
404, "No controller exists for cluster '" + clusterName + "'.");
}
if (fleetControllerPath == null || fleetControllerPath.isEmpty()) {
fleetControllerPath = "/";
}
StatusPageServer.HttpRequest req = new StatusPageServer.HttpRequest(fleetControllerPath);
req.setPathPrefix("/clustercontroller-status/v1");
StatusPageResponse response = statusServer.getStatus(req);
HttpResult result = new HttpResult();
if (response.getResponseCode() != null) {
result.setHttpCode(
response.getResponseCode().getCode(),
response.getResponseCode().getMessage());
}
if (response.getContentType() != null) {
result.addHeader("Content-Type", response.getContentType());
}
result.setContent(response.getOutputStream().toString(StandardCharsets.UTF_8));
return result;
}
public HttpResult handleClusterListRequest() {
HttpResult result = new HttpResult();
result.addHeader("Content-Type", "text/html");
StringWriter sw = new StringWriter();
sw.append("<title>clusters</title>\n");
for (String s : statusClusters.getAll().keySet()) {
sw.append("<a href=\"./").append(s).append("\">").append(s)
.append("</a><br>").append("\n");
}
result.setContent(sw.toString());
return result;
}
}
|
