github/overleaf
1
0
Fork 0
mirror of https://github.com/overleaf/overleaf.git synced 2025-04-06 06:52:52 +00:00
Code Issues Projects Releases Packages Wiki Activity

Respond to HEAD request in /status, /health_check

Browse source
This commit is contained in:
Shane Kilkelly 2020-06-26 11:54:18 +01:00
parent a9a7f54a96
commit 5e31a11938
3 changed files with 31 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -28,12 +28,13 @@ public class HealthCheckHandler extends AbstractHandler {
HttpServletRequest request,
HttpServletResponse response
) throws IOException {
String method = baseRequest.getMethod();
if (
"GET".equals(baseRequest.getMethod())
("GET".equals(method) || "HEAD".equals(method))
&& target != null
&& target.matches("^\\/health_check\\/?$")
) {
Log.info("GET <- /health_check");
Log.info(method + " <- /health_check");
baseRequest.setHandled(true);
response.setContentType("text/plain");
if (bridge.healthCheck()) {

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

@ -12,6 +12,7 @@ import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Arrays;
public class StatusHandler extends AbstractHandler {
@ -28,12 +29,13 @@ public class StatusHandler extends AbstractHandler {
HttpServletRequest request,
HttpServletResponse response
) throws IOException {
String method = baseRequest.getMethod();
if (
"GET".equals(baseRequest.getMethod())
("GET".equals(method) || "HEAD".equals(method))
&& target != null
&& target.matches("^\\/status\\/?$")
) {
Log.info("GET <- /status");
Log.info(method + " <- /status");
baseRequest.setHandled(true);
response.setContentType("text/plain");
response.setStatus(200);

24
services/git-bridge/src/test/java/uk/ac/ic/wlgitbridge/application/WLGitBridgeIntegrationTest.java
View file

@ -9,6 +9,7 @@ import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpHead;
import org.apache.http.impl.client.HttpClients;
import org.asynchttpclient.*;
import org.eclipse.jgit.api.errors.GitAPIException;
@ -944,6 +945,29 @@ public class WLGitBridgeIntegrationTest {
assertEquals(200, healthCheckResponse.getStatusLine().getStatusCode());
}
@Test
public void testStatusAndHealthCheckEndpointsWithHead() throws ClientProtocolException, IOException {
int gitBridgePort = 33889;
int mockServerPort = 3889;
server = new MockSnapshotServer(mockServerPort, getResource("/canCloneARepository").toFile());
server.start();
server.setState(states.get("canCloneARepository").get("state"));
wlgb = new GitBridgeApp(new String[] {
makeConfigFile(gitBridgePort, mockServerPort)
});
wlgb.run();
HttpClient client = HttpClients.createDefault();
String urlBase = "http://127.0.0.1:" + gitBridgePort;
// Status
HttpHead statusRequest = new HttpHead(urlBase+"/status");
HttpResponse statusResponse = client.execute(statusRequest);
assertEquals(200, statusResponse.getStatusLine().getStatusCode());
// Health Check
HttpHead healthCheckRequest = new HttpHead(urlBase+"/health_check");
HttpResponse healthCheckResponse = client.execute(healthCheckRequest);
assertEquals(200, healthCheckResponse.getStatusLine().getStatusCode());
}
private String makeConfigFile(
int port,
int apiPort