From 166e14cf7195f5923027864181e2167a0369ad24 Mon Sep 17 00:00:00 2001
From: Kavin <20838718+FireMasterK@users.noreply.github.com>
Date: Mon, 14 Feb 2022 02:38:02 +0000
Subject: [PATCH] Add dedicated route for health check (#189)

---
 src/main/java/me/kavin/piped/ServerLauncher.java | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/java/me/kavin/piped/ServerLauncher.java b/src/main/java/me/kavin/piped/ServerLauncher.java
index 8a738fc..cf060c9 100644
--- a/src/main/java/me/kavin/piped/ServerLauncher.java
+++ b/src/main/java/me/kavin/piped/ServerLauncher.java
@@ -24,7 +24,6 @@ import org.xml.sax.InputSource;
 
 import java.io.ByteArrayInputStream;
 import java.net.InetSocketAddress;
-import java.nio.charset.StandardCharsets;
 import java.util.Objects;
 import java.util.concurrent.Executor;
 
@@ -32,6 +31,7 @@ import static io.activej.config.converter.ConfigConverters.ofInetSocketAddress;
 import static io.activej.http.HttpHeaders.*;
 import static io.activej.http.HttpMethod.GET;
 import static io.activej.http.HttpMethod.POST;
+import static java.nio.charset.StandardCharsets.UTF_8;
 
 public class ServerLauncher extends MultithreadedHttpServerLauncher {
 
@@ -44,6 +44,7 @@ public class ServerLauncher extends MultithreadedHttpServerLauncher {
     AsyncServlet mainServlet(Executor executor) {
 
         RoutingServlet router = RoutingServlet.create()
+                .map(GET, "/healthcheck", request -> getRawResponse("OK".getBytes(UTF_8), "text/plain", "no-cache"))
                 .map(HttpMethod.OPTIONS, "/*", request -> HttpResponse.ofCode(200))
                 .map(GET, "/webhooks/pubsub", request -> HttpResponse.ok200().withPlainText(Objects.requireNonNull(request.getQueryParameter("hub.challenge"))))
                 .map(POST, "/webhooks/pubsub", AsyncServlet.ofBlocking(executor, request -> {
@@ -70,7 +71,7 @@ public class ServerLauncher extends MultithreadedHttpServerLauncher {
                     try {
                         return getJsonResponse(
                                 SponsorBlockUtils.getSponsors(request.getPathParameter("videoId"),
-                                        request.getQueryParameter("category")).getBytes(StandardCharsets.UTF_8),
+                                        request.getQueryParameter("category")).getBytes(UTF_8),
                                 "public, max-age=3600");
                     } catch (Exception e) {
                         return getErrorResponse(e, request.getPath());