ReservationControllerAdvice.java

package com.prgms.kokoahairshop.reservation.controller;

import com.prgms.kokoahairshop.reservation.exception.HairshopClosedDayException;
import com.prgms.kokoahairshop.reservation.exception.ReservationCancelTimeoutException;
import com.prgms.kokoahairshop.reservation.exception.ReservationNotFoundException;
import com.prgms.kokoahairshop.reservation.exception.ReservationNotReservedException;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

@RestControllerAdvice("com.prgms.kokoahairshop.reservation.controller")
public class ReservationControllerAdvice {

    @ExceptionHandler(ReservationNotFoundException.class)
    ResponseEntity<Object> handleReservationNotFoundException(ReservationNotFoundException e) {
        return ResponseEntity.notFound().build();
    }

    @ExceptionHandler(ReservationNotReservedException.class)
    ResponseEntity<String> handleReservationNotReservedException(
        ReservationNotReservedException e) {
        return ResponseEntity.badRequest().body(e.getMessage());
    }

    @ExceptionHandler(ReservationCancelTimeoutException.class)
    ResponseEntity<String> handleReservationCancelTimeoutException(
        ReservationCancelTimeoutException e) {
        return ResponseEntity.badRequest().body(e.getMessage());
    }
}