♻️ :: authenticationEntryPoint 수정

team-xquare · Jul 11, 2023 · 9cbac3c · 9cbac3c
1 parent 8f7793f
commit 9cbac3c
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/git-infrastructure/src/main/kotlin/com/xquare/git/global/config/SecurityConfiguration.kt b/git-infrastructure/src/main/kotlin/com/xquare/git/global/config/SecurityConfiguration.kt
@@ -5,8 +5,12 @@ import org.springframework.context.annotation.Configuration
 import org.springframework.http.HttpStatus
 import org.springframework.security.config.annotation.web.reactive.EnableWebFluxSecurity
 import org.springframework.security.config.web.server.ServerHttpSecurity
+import org.springframework.security.core.AuthenticationException
 import org.springframework.security.web.server.SecurityWebFilterChain
-import org.springframework.security.web.server.authentication.HttpStatusServerEntryPoint
+import org.springframework.security.web.server.ServerAuthenticationEntryPoint
+import org.springframework.stereotype.Component
+import org.springframework.web.server.ServerWebExchange
+import reactor.core.publisher.Mono
 
 @Configuration
 @EnableWebFluxSecurity
@@ -21,8 +25,16 @@ class SecurityConfiguration {
             .anyExchange().authenticated()
             .and()
             .httpBasic()
-            .authenticationEntryPoint(HttpStatusServerEntryPoint(HttpStatus.UNAUTHORIZED))
+            .authenticationEntryPoint(CustomAuthenticationEntryPoint())
             .and()
             .build()
     }
+}
+
+@Component
+class CustomAuthenticationEntryPoint : ServerAuthenticationEntryPoint {
+    override fun commence(exchange: ServerWebExchange, ex: AuthenticationException): Mono<Void> {
+        exchange.response.statusCode = HttpStatus.UNAUTHORIZED
+        return exchange.response.setComplete()
+    }
 }