Added includeHeaders flag in ExceptionResolverResult

This commit is contained in:
Denis Kazantsev 2022-03-30 12:42:45 +03:00
parent fb8afe2b14
commit 93ba1ab526
2 changed files with 3 additions and 2 deletions

View File

@ -8,6 +8,7 @@ data class ExceptionResolverResult(
val apiError: ApiError,
val status: HttpStatus,
val exception: Exception?,
val includeHeaders: Boolean = false,
) {
companion object {

View File

@ -31,10 +31,10 @@ class ExceptionHandlerAdvice(
val body = exceptionResponseBodyCreator(result.apiError)
val headers = HttpHeaders().apply {
val headers = if (result.includeHeaders) HttpHeaders().apply {
set("X-Error-Code", result.apiError.errorCode.toString())
set("X-Error-Message", result.apiError.errorMessage)
}
} else null
return ResponseEntity(body, headers, result.status)
}