github/overleaf
1
0
Fork 0
mirror of https://github.com/overleaf/overleaf.git synced 2024-11-21 20:47:08 -05:00
Code Issues Projects Releases Packages Wiki Activity

Use ServletHolder lifecycle more carefully

Browse source 
Fixes metrics with Jetty > 9.4.20
This commit is contained in:
Simon Detheridge 2021-06-22 15:38:43 +01:00
parent 70f4e5801d
commit 63ca055637
2 changed files with 32 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
services/git-bridge/src/main/java/uk/ac/ic/wlgitbridge/server/GitBridgeServer.java
View file

@ -127,7 +127,7 @@ public class GitBridgeServer {
handlers.addHandler(new StatusHandler(bridge));
handlers.addHandler(new HealthCheckHandler(bridge));
handlers.addHandler(new GitLfsHandler(bridge));
handlers.addHandler(new PrometheusHandler(bridge));
handlers.addHandler(new PrometheusHandler());
base.setHandler(handlers);
return base;
}

40
services/git-bridge/src/main/java/uk/ac/ic/wlgitbridge/server/PrometheusHandler.java
View file

@ -18,17 +18,30 @@ import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Arrays;
/**
* Wrapper for the MetricsServlet from the Prometheus client.
*
* We wrap this in a handler here, as we use a wildcard servlet context on /*, and adding
* an additional servlet context collides with it. Adding a servlet context on /metrics
* causes a redirect to /metrics/ which breaks things when jetty doesn't know the full
* public URL.
*
* There may still be a better way to do this, but it works.
**/
public class PrometheusHandler extends AbstractHandler {
private final Bridge bridge;
private final MetricsServlet metricsServlet;
private final ServletHolder holder;
public PrometheusHandler(Bridge bridge) {
this.bridge = bridge;
this.metricsServlet = new MetricsServlet();
this.holder = new ServletHolder(metricsServlet);
public PrometheusHandler() {
DefaultExports.initialize();
this.holder = new ServletHolder(new MetricsServlet());
try {
this.holder.initialize();
} catch (Exception e) {
Log.error("Unable to initialise metrics servlet: " + e.getMessage());
}
}
@Override
@ -45,11 +58,20 @@ public class PrometheusHandler extends AbstractHandler {
&& target.matches("^/metrics/?$")
) {
Log.info(method + " <- /metrics");
response.setContentType("application/vnd.git-lfs+json");
response.setStatus(200);
if (!this.holder.isAvailable()) {
try {
this.holder.start();
} catch (Exception e) {
Log.error("Unable to start metrics servlet: " + e.getMessage());
response.setStatus(500);
baseRequest.setHandled(true);
return;
}
}
this.holder.handle(baseRequest, request, response);
baseRequest.setHandled(true);
}
}
}