From c7a13f4c10e834b3cab8d5cdbe0625e5eb0543b4 Mon Sep 17 00:00:00 2001 From: ay-eonii Date: Tue, 29 Oct 2024 22:17:27 +0900 Subject: [PATCH] =?UTF-8?q?fix:=20=ED=8A=B8=EB=9E=98=ED=94=BD=20=EC=A4=91?= =?UTF-8?q?=EB=8B=A8=EC=9D=84=20=EC=9C=84=ED=95=9C=20=ED=97=AC=EC=8A=A4?= =?UTF-8?q?=EC=B2=B4=ED=81=AC=20=ED=98=B8=EC=8A=A4=ED=8A=B8=20=EB=B3=80?= =?UTF-8?q?=EA=B2=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../backend/common/HealthCheckController.java | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/backend/src/main/java/mouda/backend/common/HealthCheckController.java b/backend/src/main/java/mouda/backend/common/HealthCheckController.java index 03c668c1..3f6c1717 100644 --- a/backend/src/main/java/mouda/backend/common/HealthCheckController.java +++ b/backend/src/main/java/mouda/backend/common/HealthCheckController.java @@ -13,7 +13,8 @@ @RestController public class HealthCheckController { - private static final String HOST_IP = "127.0.0.1"; + private static final String HOST_IPV4 = "127.0.0.1"; + private static final String HOST_IPV6 = "0:0:0:0:0:0:0:1"; private static final String HOST_NAME = "localhost"; private final AtomicBoolean isTerminating = new AtomicBoolean(false); @@ -28,12 +29,12 @@ public ResponseEntity checkHealth() { @PostMapping("/termination") public ResponseEntity terminate(HttpServletRequest request) { - isTerminating.set(true); - return ResponseEntity.ok().build(); - // String remoteAddr = request.getRemoteAddr(); - - // if (HOST_IP.equals(remoteAddr) || HOST_NAME.equals(remoteAddr)) { - // } - // return ResponseEntity.status(HttpStatus.FORBIDDEN).build(); + String remoteHost = request.getRemoteHost(); + System.out.println(remoteHost); + if (HOST_IPV6.equals(remoteHost) || HOST_IPV4.equals(remoteHost) || HOST_NAME.equals(remoteHost)) { + isTerminating.set(true); + return ResponseEntity.ok().build(); + } + return ResponseEntity.status(HttpStatus.FORBIDDEN).build(); } }