Refactor HealthService to use readyz and livez instead of liveness an…

…d readyness endpoints Signed-off-by: Alfredo Gutierrez <[email protected]>
hashgraph · Aug 27, 2024 · 321662d · 321662d
1 parent c8e0aeb
commit 321662d
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 19 deletions.
diff --git a/charts/hedera-block-node/values.yaml b/charts/hedera-block-node/values.yaml
@@ -84,6 +84,6 @@ blockNode:
     PRIVATE_KEY: "fake_private_key"
   health:
     readiness:
-      endpoint: "/healthz/readiness"
+      endpoint: "/healthz/readyz"
     liveness:
-      endpoint: "/healthz/liveness"
+      endpoint: "/healthz/livez"
diff --git a/server/src/main/java/com/hedera/block/server/health/HealthService.java b/server/src/main/java/com/hedera/block/server/health/HealthService.java
@@ -37,7 +37,7 @@ public interface HealthService extends HttpService {
      * @param req the server request
      * @param res the server response
      */
-    void handleLiveness(@NonNull final ServerRequest req, @NonNull final ServerResponse res);
+    void handleLivez(@NonNull final ServerRequest req, @NonNull final ServerResponse res);
 
     /**
      * Handles the request for readiness endpoint, that it most be defined on routing
@@ -46,5 +46,5 @@ public interface HealthService extends HttpService {
      * @param req the server request
      * @param res the server response
      */
-    void handleReadiness(@NonNull final ServerRequest req, @NonNull final ServerResponse res);
+    void handleReadyz(@NonNull final ServerRequest req, @NonNull final ServerResponse res);
 }
diff --git a/server/src/main/java/com/hedera/block/server/health/HealthServiceImpl.java b/server/src/main/java/com/hedera/block/server/health/HealthServiceImpl.java
@@ -28,8 +28,8 @@
 @Singleton
 public class HealthServiceImpl implements HealthService {
 
-    private static final String LIVENESS_PATH = "/liveness";
-    private static final String READINESS_PATH = "/readiness";
+    private static final String LIVEZ_PATH = "/livez";
+    private static final String READYZ_PATH = "/readyz";
 
     private final ServiceStatus serviceStatus;
 
@@ -56,9 +56,7 @@ public String getHealthRootPath() {
      */
     @Override
     public void routing(@NonNull final HttpRules httpRules) {
-        httpRules
-                .get(LIVENESS_PATH, this::handleLiveness)
-                .get(READINESS_PATH, this::handleReadiness);
+        httpRules.get(LIVEZ_PATH, this::handleLivez).get(READYZ_PATH, this::handleReadyz);
     }
 
     /**
@@ -68,7 +66,7 @@ public void routing(@NonNull final HttpRules httpRules) {
      * @param res the server response
      */
     @Override
-    public final void handleLiveness(
+    public final void handleLivez(
             @NonNull final ServerRequest req, @NonNull final ServerResponse res) {
         if (serviceStatus.isRunning()) {
             res.status(200).send("OK");
@@ -85,7 +83,7 @@ public final void handleLiveness(
      * @param res the server response
      */
     @Override
-    public final void handleReadiness(
+    public final void handleReadyz(
             @NonNull final ServerRequest req, @NonNull final ServerResponse res) {
         if (serviceStatus.isRunning()) {
             res.status(200).send("OK");

diff --git a/server/src/test/java/com/hedera/block/server/health/HealthServiceTest.java b/server/src/test/java/com/hedera/block/server/health/HealthServiceTest.java
@@ -42,63 +42,63 @@ class HealthServiceTest {
     @Mock ServerResponse serverResponse;
 
     @Test
-    public void testHandleLiveness() {
+    public void testHandleLivez() {
         // given
         when(serviceStatus.isRunning()).thenReturn(true);
         when(serverResponse.status(200)).thenReturn(serverResponse);
         doNothing().when(serverResponse).send("OK");
         HealthService healthService = new HealthServiceImpl(serviceStatus);
 
         // when
-        healthService.handleLiveness(serverRequest, serverResponse);
+        healthService.handleLivez(serverRequest, serverResponse);
 
         // then
         verify(serverResponse, times(1)).status(200);
         verify(serverResponse, times(1)).send("OK");
     }
 
     @Test
-    public void testHandleLiveness_notRunning() {
+    public void testHandleLivez_notRunning() {
         // given
         when(serviceStatus.isRunning()).thenReturn(false);
         when(serverResponse.status(503)).thenReturn(serverResponse);
         doNothing().when(serverResponse).send("Service is not running");
         HealthService healthService = new HealthServiceImpl(serviceStatus);
 
         // when
-        healthService.handleLiveness(serverRequest, serverResponse);
+        healthService.handleLivez(serverRequest, serverResponse);
 
         // then
         verify(serverResponse, times(1)).status(503);
         verify(serverResponse, times(1)).send("Service is not running");
     }
 
     @Test
-    public void testHandleReadiness() {
+    public void testHandleReadyz() {
         // given
         when(serviceStatus.isRunning()).thenReturn(true);
         when(serverResponse.status(200)).thenReturn(serverResponse);
         doNothing().when(serverResponse).send("OK");
         HealthService healthService = new HealthServiceImpl(serviceStatus);
 
         // when
-        healthService.handleReadiness(serverRequest, serverResponse);
+        healthService.handleReadyz(serverRequest, serverResponse);
 
         // then
         verify(serverResponse, times(1)).status(200);
         verify(serverResponse, times(1)).send("OK");
     }
 
     @Test
-    public void testHandleReadiness_notRunning() {
+    public void testHandleReadyz_notRunning() {
         // given
         when(serviceStatus.isRunning()).thenReturn(false);
         when(serverResponse.status(503)).thenReturn(serverResponse);
         doNothing().when(serverResponse).send("Service is not running");
         HealthService healthService = new HealthServiceImpl(serviceStatus);
 
         // when
-        healthService.handleReadiness(serverRequest, serverResponse);
+        healthService.handleReadyz(serverRequest, serverResponse);
 
         // then
         verify(serverResponse, times(1)).status(503);