From 89ef54df80995ef56f7e31fa926749d0ae62ece6 Mon Sep 17 00:00:00 2001
From: Mona Mohamadinia <mona.mohamadinia@gmail.com>
Date: Fri, 14 Aug 2020 03:26:35 +0430
Subject: [PATCH] Added a HttpCodeStatusMapper impl.

---
 .../health/CustomStatusCodeMapper.java        | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 spring-boot-modules/spring-boot-actuator/src/main/java/com/baeldung/health/CustomStatusCodeMapper.java

diff --git a/spring-boot-modules/spring-boot-actuator/src/main/java/com/baeldung/health/CustomStatusCodeMapper.java b/spring-boot-modules/spring-boot-actuator/src/main/java/com/baeldung/health/CustomStatusCodeMapper.java
new file mode 100644
index 0000000000..0ff6773240
--- /dev/null
+++ b/spring-boot-modules/spring-boot-actuator/src/main/java/com/baeldung/health/CustomStatusCodeMapper.java
@@ -0,0 +1,26 @@
+package com.baeldung.health;
+
+import org.springframework.boot.actuate.health.HttpCodeStatusMapper;
+import org.springframework.boot.actuate.health.Status;
+import org.springframework.stereotype.Component;
+
+@Component
+public class CustomStatusCodeMapper implements HttpCodeStatusMapper {
+
+    @Override
+    public int getStatusCode(Status status) {
+        if (status == Status.DOWN) {
+            return 500;
+        }
+
+        if (status == Status.OUT_OF_SERVICE) {
+            return 503;
+        }
+
+        if (status == Status.UNKNOWN) {
+            return 500;
+        }
+
+        return 200;
+    }
+}