上海网站建设门户,在线教育网站建设投标书,一般网站建设需求有哪些方面,wordpress cpu负载过高在SpringBoot的开发中#xff0c;为了提高程序运行的鲁棒性#xff0c;我们经常需要对各种程序异常进行处理#xff0c;但是如果在每个出异常的地方进行单独处理的话#xff0c;这会引入大量业务不相关的异常处理代码#xff0c;增加了程序的耦合#xff0c;同时未来想改… 在SpringBoot的开发中为了提高程序运行的鲁棒性我们经常需要对各种程序异常进行处理但是如果在每个出异常的地方进行单独处理的话这会引入大量业务不相关的异常处理代码增加了程序的耦合同时未来想改变异常的处理逻辑也变得比较困难。这篇文章带大家了解一下如何优雅的进行全局异常处理。 为了实现全局拦截这里使用到了Spring中提供的两个注解RestControllerAdvice和ExceptionHandler结合使用可以拦截程序中产生的异常并且根据不同的异常类型分别处理。下面我会先介绍如何利用这两个注解优雅的完成全局异常的处理接着解释这背后的原理。
1. 如何实现全局拦截
1.1 自定义异常处理类
在下面的例子中我们继承了 ResponseEntityExceptionHandler并使用RestControllerAdvice注解了这个类接着结合ExceptionHandler针对不同的异常类型来定义不同的异常处理方法。这里可以看到我处理的异常是自定义异常后续我会展开介绍。 ResponseEntityExceptionHandler中包装了各种SpringMVC在处理请求时可能抛出的异常的处理处理结果都是封装成一个ResponseEntity对象。ResponseEntityExceptionHandler是一个抽象类通常我们需要定义一个用来处理异常的使用RestControllerAdvice注解标注的异常处理类来继承自ResponseEntityExceptionHandler。ResponseEntityExceptionHandler中为每个异常的处理都单独定义了一个方法如果默认的处理不能满足你的需求则可以重写对某个异常的处理。 java复制代码Log4j2
RestControllerAdvice
public class GlobalExceptionHandler extends ResponseEntityExceptionHandler { /** * 定义要捕获的异常 可以多个 ExceptionHandler({}) * * param request request * param e exception * param response response * return 响应结果 */ ExceptionHandler(AuroraRuntimeException.class) public GenericResponse customExceptionHandler(HttpServletRequest request, final Exception e, HttpServletResponse response) { AuroraRuntimeException exception (AuroraRuntimeException) e; if (exception.getCode() ResponseCode.USER_INPUT_ERROR) { response.setStatus(HttpStatus.BAD_REQUEST.value()); } else if (exception.getCode() ResponseCode.FORBIDDEN) { response.setStatus(HttpStatus.FORBIDDEN.value()); } else { response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value()); } return new GenericResponse(exception.getCode(), null, exception.getMessage()); } ExceptionHandler(NotLoginException.class) public GenericResponse tokenExceptionHandler(HttpServletRequest request, final Exception e, HttpServletResponse response) { log.error(token exception, e); response.setStatus(HttpStatus.FORBIDDEN.value()); return new GenericResponse(ResponseCode.AUTHENTICATION_NEEDED); } }1.2 定义异常码
这里定义了常见的几种异常码主要用在抛出自定义异常时对不同的情形进行区分。
java复制代码Getter
public enum ResponseCode { SUCCESS(0, Success), INTERNAL_ERROR(1, 服务器内部错误), USER_INPUT_ERROR(2, 用户输入错误), AUTHENTICATION_NEEDED(3, Token过期或无效), FORBIDDEN(4, 禁止访问), TOO_FREQUENT_VISIT(5, 访问太频繁请休息一会儿); private final int code; private final String message; private final Response.Status status; ResponseCode(int code, String message, Response.Status status) { this.code code; this.message message; this.status status; } ResponseCode(int code, String message) { this(code, message, Response.Status.INTERNAL_SERVER_ERROR); } }1.3 自定义异常类
这里我定义了一个AuroraRuntimeException的异常就是在上面的异常处理函数中用到的异常。每个异常实例会有一个对应的异常码也就是前面刚定义好的。
java复制代码Getter
public class AuroraRuntimeException extends RuntimeException { private final ResponseCode code; public AuroraRuntimeException() { super(String.format(%s, ResponseCode.INTERNAL_ERROR.getMessage())); this.code ResponseCode.INTERNAL_ERROR; } public AuroraRuntimeException(Throwable e) { super(e); this.code ResponseCode.INTERNAL_ERROR; } public AuroraRuntimeException(String msg) { this(ResponseCode.INTERNAL_ERROR, msg); } public AuroraRuntimeException(ResponseCode code) { super(String.format(%s, code.getMessage())); this.code code; } public AuroraRuntimeException(ResponseCode code, String msg) { super(msg); this.code code; } }1.4 自定义返回类型
为了保证各个接口的返回统一这里专门定义了一个返回类型。
java复制代码Getter
Setter
public class GenericResponseT { private int code; private T data; private String message; public GenericResponse() {}; public GenericResponse(int code, T data) { this.code code; this.data data; } public GenericResponse(int code, T data, String message) { this(code, data); this.message message; } public GenericResponse(ResponseCode responseCode) { this.code responseCode.getCode(); this.data null; this.message responseCode.getMessage(); } public GenericResponse(ResponseCode responseCode, T data) { this(responseCode); this.data data; } public GenericResponse(ResponseCode responseCode, T data, String message) { this(responseCode, data); this.message message; }
}实际测试异常
下面的例子中我们想获取到用户的信息如果用户的信息不存在可以直接抛出一个异常这个异常会被我们上面定义的全局异常处理方法所捕获然后根据不同的异常编码完成不同的处理和返回。
java复制代码public User getUserInfo(Long userId) { // some logicUser user daoFactory.getExtendedUserMapper().selectByPrimaryKey(userId); if (user null) { throw new AuroraRuntimeException(ResponseCode.USER_INPUT_ERROR, 用户id不存在); }// some logic....
}以上就完成了整个全局异常的处理过程接下来重点说说为什么RestControllerAdvice和ExceptionHandler结合使用可以拦截程序中产生的异常
全局拦截的背后原理 下面会提到ControllerAdvice注解简单地说RestControllerAdvice与ControllerAdvice的区别就和RestController与Controller的区别类似RestControllerAdvice注解包含了ControllerAdvice注解和ResponseBody注解。 接下来我们深入Spring源码看看是怎么实现的首先DispatcherServlet对象在创建时会初始化一系列的对象这里重点关注函数 initHandlerExceptionResolvers(context);.
java复制代码public class DispatcherServlet extends FrameworkServlet {// ......protected void initStrategies(ApplicationContext context) {initMultipartResolver(context);initLocaleResolver(context);initThemeResolver(context);initHandlerMappings(context);initHandlerAdapters(context);// 重点关注initHandlerExceptionResolvers(context);initRequestToViewNameTranslator(context);initViewResolvers(context);initFlashMapManager(context);}// ......
}在 initHandlerExceptionResolvers(context)方法中会取得所有实现了HandlerExceptionResolver接口的bean并保存起来其中就有一个类型为ExceptionHandlerExceptionResolver的bean这个bean在应用启动过程中会获取所有被ControllerAdvice注解标注的bean对象做进一步处理关键代码在这里
java复制代码public class ExceptionHandlerExceptionResolver extends AbstractHandlerMethodExceptionResolverimplements ApplicationContextAware, InitializingBean {// ......private void initExceptionHandlerAdviceCache() {// ......ListControllerAdviceBean adviceBeans ControllerAdviceBean.findAnnotatedBeans(getApplicationContext());AnnotationAwareOrderComparator.sort(adviceBeans);for (ControllerAdviceBean adviceBean : adviceBeans) {ExceptionHandlerMethodResolver resolver new ExceptionHandlerMethodResolver(adviceBean.getBeanType());if (resolver.hasExceptionMappings()) {// 找到所有ExceptionHandler标注的方法并保存成一个ExceptionHandlerMethodResolver类型的对象缓存起来this.exceptionHandlerAdviceCache.put(adviceBean, resolver);if (logger.isInfoEnabled()) {logger.info(Detected ExceptionHandler methods in adviceBean);}}// ......}}// ......
}当Controller抛出异常时DispatcherServlet通过 ExceptionHandlerExceptionResolver来解析异常而ExceptionHandlerExceptionResolver又通过ExceptionHandlerMethodResolver 来解析异常 ExceptionHandlerMethodResolver 最终解析异常找到适用的ExceptionHandler标注的方法是这里
java复制代码public class ExceptionHandlerMethodResolver {// ......private Method getMappedMethod(Class? extends Throwable exceptionType) {ListClass? extends Throwable matches new ArrayListClass? extends Throwable();// 找到所有适用于Controller抛出异常的处理方法,例如Controller抛出的异常// 是AuroraRuntimeException(继承自RuntimeException),那么ExceptionHandler(AuroraRuntimeException.class)和// ExceptionHandler(Exception.class)标注的方法都适用此异常for (Class? extends Throwable mappedException : this.mappedMethods.keySet()) {if (mappedException.isAssignableFrom(exceptionType)) {matches.add(mappedException);}}if (!matches.isEmpty()) {/* 这里通过排序找到最适用的方法,排序的规则依据抛出异常相对于声明异常的深度,例如Controller抛出的异常是是AuroraRuntimeException(继承自RuntimeException),那么AuroraRuntimeException相对于ExceptionHandler(AuroraRuntimeException.class)声明的AuroraRuntimeException.class其深度是0,相对于ExceptionHandler(Exception.class)声明的Exception.class其深度是2,所以ExceptionHandler(BizException.class)标注的方法会排在前面 */Collections.sort(matches, new ExceptionDepthComparator(exceptionType));return this.mappedMethods.get(matches.get(0));}else {return null;}}// ......
}整个RestControllerAdvice处理的流程就是这样结合ExceptionHandler就完成了对不同异常的灵活处理。
