使用@ExceptionHandler根据请求动态返回HTTP状态代码

Question

我希望根据响应对象错误动态返回HTTPStatus代码，如400、400、404等。我被提到这个问题--使用Spring3RESTful以编程方式更改http响应状态，但没有帮助。

我有一个带有@exceptionHandler方法的控制器类

@ExceptionHandler(CustomException.class)
    @ResponseBody
    public ResponseEntity<?> handleException(CustomException e) {
        return new ResponseEntity<MyErrorResponse>(
                new MyErrorResponse(e.getCode(), ExceptionUtility.getMessage(e.getMessage())), 
                ExceptionUtility.getHttpCode(e.getCode()));
    }

ExceptionUtility是一个类，其中有上面使用的两个方法(GetMessage和GetCode)。

public class ExceptionUtility {
    public static String getMessage(String message) {
        return message;
    }

    public static HttpStatus getHttpCode(String code) {
        return HttpStatus.NOT_FOUND; //how to return status code dynamically here ?
    }
}

我不想签入if条件并相应地返回响应代码，有没有其他更好的方法来做到这一点？

Answer 1

第一种方法：

您可以在customException类中获取HTTPStatus字段。例如。

    class CustomException {
         HttpStatus httpStatus;
         ....other fields
    }

您可以抛出如下错误：

throw new CustomException(otherField , HttpStatus.CREATED);

@ExceptionHandler(CustomException.class)
    @ResponseBody
    public ResponseEntity<?> handleException(CustomException e) {
        return new ResponseEntity<MyErrorResponse>(
                new MyErrorResponse(e.getCode(), ExceptionUtility.getMessage(e.getMessage())), 
                e.getHttpStatus());
    }

public enum ErrorCode {


    MyErrorCode(HttpStatus.OK);

    HttpStatus status;

//getter for httpstatus
}

@ExceptionHandler(CustomException.class)
        @ResponseBody
        public ResponseEntity<?> handleException(CustomException e) {
            return new ResponseEntity<MyErrorResponse>(
                    new MyErrorResponse(e.getCode(), ExceptionUtility.getMessage(e.getMessage())), 
                    e.getHttpStatus());
        }

HttpStatus getHttpStatus(String code) {
   return ErrorCode.valueOf(code.toUpperCase()).getHttpStatus();
}

对不起，我错过了空指针异常在上面的方法，但您可以修改根据您的需要，只是给出一个想法。：）