HttpStatusCodeException 수정(json방식 응답 추가)

main
이범준 12 months ago
parent e420217d11
commit 62b49426d5

@ -5,13 +5,24 @@ public class HttpStatusCodeException extends RuntimeException {
private static final long serialVersionUID = 1L;
private int status;
private boolean contentTypeJson;
public int getStatus() {
return this.status;
}
public boolean isJsonResp() {
return this.contentTypeJson;
}
public HttpStatusCodeException(int statusCode, String message){
super(message);
this.status = statusCode;
}
public HttpStatusCodeException(int statusCode, String message, boolean jsonResp){
super(message);
this.status = statusCode;
this.contentTypeJson = jsonResp;
}
}

@ -18,9 +18,17 @@ public class HttpStatusCodeExceptionControllerAdvice extends ResponseEntityExcep
public ResponseEntity<Object> handleHttpStatusCodeException(
final HttpStatusCodeException exception, final WebRequest webRequest) {
int status = exception.getStatus();
boolean isJsonResp = exception.isJsonResp();
String body = exception.getMessage();
HttpHeaders header = new HttpHeaders();
header.setContentType(new MediaType(MediaType.TEXT_PLAIN, StandardCharsets.UTF_8));
if(isJsonResp) {
header.setContentType(new MediaType(MediaType.APPLICATION_JSON, StandardCharsets.UTF_8));
} else {
header.setContentType(new MediaType(MediaType.TEXT_PLAIN, StandardCharsets.UTF_8));
}
return handleExceptionInternal(exception, body, header, HttpStatus.valueOf(status), webRequest);
}

Loading…
Cancel
Save