SpringMVC基于注解的异常处理（二）

@ControllerAdvice
public class GlobalExceptionHandler {

  @ExceptionHandler({InvalidEmailException.class, InvalidPasswordException.class})
  public ResponseEntity<String> handleInvalidInputException(Exception ex) {
    return ResponseEntity
            .status(HttpStatus.BAD_REQUEST)
            .body(ex.getMessage());
  }

  @ExceptionHandler(UserNotFoundException.class)
  public ResponseEntity<String> handleUserNotFoundException(UserNotFoundException ex) {
    return ResponseEntity
            .status(HttpStatus.NOT_FOUND)
            .body(ex.getMessage());
  }

  @ExceptionHandler(Exception.class)
  public ResponseEntity<String> handleException(Exception ex) {
    return ResponseEntity
            .status(HttpStatus.INTERNAL_SERVER_ERROR)
            .body("An unexpected error occurred.");
  }
}

@Controller
public class UserController {

  @PostMapping("/register")
  public ResponseEntity<String> registerUser(@RequestParam String email, @RequestParam String password) throws InvalidEmailException, InvalidPasswordException {
    if(!isValidEmail(email)) {
      throw new InvalidEmailException("Invalid email address.");
    }
    if(!isValidPassword(password)) {
      throw new InvalidPasswordException("Password must be at least 8 characters long.");
    }
    // 用户注册逻辑
    return ResponseEntity.ok("User registered successfully.");
  }

  private boolean isValidEmail(String email) {
    // 验证电子邮件地址是否有效
    return true;
  }

  private boolean isValidPassword(String password) {
    // 验证密码是否有效
    return password.length() >= 8;
  }

  public static void main(String[] args) {
    SpringApplication.run(UserController.class, args);
  }
}

class InvalidEmailException extends Exception {
  public InvalidEmailException(String message) {
    super(message);
  }
}

class InvalidPasswordException extends Exception {
  public InvalidPasswordException(String message) {
    super(message);
  }
}

class UserNotFoundException extends Exception {
  public UserNotFoundException(String message) {
    super(message);
  }
}