SpringBoot拦截器以及源码详析
1、拦截器是什么
java里的拦截器（Interceptor）是动态拦截Action调用的对象，它提供了一种机制可以使开发者在一个Action执行的前后执行一段代码，也可以在一个Action执行前阻止其执行，同时也提供了一种可以提取Action中可重用部分代码的方式。在AOP中，拦截器用于在某个方法或者字段被访问之前进行拦截，然后再之前或者之后加入某些操作。
上面的Action一般指的就是我们Controller层的接口。
2、自定义拦截器
一般自定义一个拦截器分为三步
（1）编写一个拦截器实现HandlerInterceptor接口。
（2）拦截器注册到容器中。
（3）配置拦截规则。
2.1 编写拦截器
我们新建一个SpringBoot项目，然后自定义一个拦截器LoginInterceptor，拦截未登录状态下的某些请求。JDK1.8开始，接口的方法加上default关键字可以有默认实现，所以实现一个接口只需要实现没有加该关键字的方法。
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@Slf4j
public class LoginInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
// 获取请求路径
String requestUrl = request.getRequestURI();
log.info("请求的路径是: {}", requestUrl);
String username = request.getParameter("username");
if (username != null) {
// 放行
return true;
}
request.setAttribute("msg", "请先登录");
// 携带msg跳转到登录页
request.getRequestDispatcher("/").forward(request, response);
return false;
}
@Override
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
log.info("postHandle执行");
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
log.info("afterCompletion执行");
}
}
2.2 注册和配置拦截器
在SpringBoot中，我们需要自定义配置的时候，只需要实现WebMvcConfigurer类重写对应的方法即可。这里我们需要配置拦截器，那么重写它的addInterceptors方法即可。
import com.codeliu.interceptor.LoginInterceptor;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
// 表示这是一个配置类
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new LoginInterceptor())
.addPathPatterns("
boolean applyPreHandle(HttpServletRequest request, HttpServletResponse response) throws Exception {
// 遍历拦截器
for (int i = 0; i < this.interceptorList.size(); i++) {
HandlerInterceptor interceptor = this.interceptorList.get(i);
// 执行当前拦截器的preHandle方法
if (!interceptor.preHandle(request, response, this.handler)) {
// 如果preHandle方法返回为false，则执行当前拦截器的afterCompletion方法
triggerAfterCompletion(request, response, null);
return false;
}
// 记录当前拦截器的下标
this.interceptorIndex = i;
}
return true;
}
通过上面的代码， 我们知道如果当前拦截器的preHandle方法返回为true，则会继续执行下一拦截器的preHandle方法，否则执行拦截器的afterCompletion方法。
那么我们看看triggerAfterCompletion方法的逻辑。
void triggerAfterCompletion(HttpServletRequest request, HttpServletResponse response, @Nullable Exception ex) {
// 反向遍历拦截器
for (int i = this.interceptorIndex; i >= 0; i--) {
HandlerInterceptor interceptor = this.interceptorList.get(i);
try {
// 执行当前拦截器的afterCompletion方法
interceptor.afterCompletion(request, response, this.handler, ex);
}
catch (Throwable ex2) {
logger.error("HandlerInterceptor.afterCompletion threw exception", ex2);
}
}
}
通过上面的代码，我们知道对于拦截器的afterCompletion方法，是反向执行的。
3.3 执行目标方法
如果上面拦截器的所有preHandle方法返回都为true，那么在doDispatch方法内就不会直接return，而是继续往下执行目标方法。如果任何一个拦截器的preHandle方法返回为false，那么执行完拦截器（已经执行过preHandle方法的拦截器）的afterCompletion方法后，在doDispatch方法内会直接return，不会执行目标方法。
通过下面的代码执行目标方法
// Actually invoke the handler.
mv = ha.handle(processedRequest, response, mappedHandler.getHandler());
内部具体怎么执行的就不看了，看看执行完后的逻辑。
3.4 执行拦截器的postHandle方法
目标方法执行完后，代码往下走
mappedHandler.applyPostHandle(processedRequest, response, mv);
查看applyPostHandle的逻辑
void applyPostHandle(HttpServletRequest request, HttpServletResponse response, @Nullable ModelAndView mv)
throws Exception {
// 反向遍历
for (int i = this.interceptorList.size() - 1; i >= 0; i--) {
HandlerInterceptor interceptor = this.interceptorList.get(i);
// 执行当前拦截器的postHandle方法
interceptor.postHandle(request, response, this.handler, mv);
}
}
倒序执行拦截器的postHandle方法
3.5 执行拦截器的afterCompletion方法
继续往下走
processDispatchResult(processedRequest, response, mappedHandler, mv, dispatchException);
进入该方法，该方法会处理执行结果，渲染页面，该方法的最后，执行下面的代码
3.6 异常处理
如果在执行doDispatch方法过程中，抛出了异常，在catch模块，都会触发执行afterCompletion方法
4、总结
上面的过程，大概可以总结为以下几步：
（1）根据当前请求，找到可以处理请求的handler和handler的所有拦截器。
（2）顺序执行所有拦截器的preHandle方法
如果当前拦截器的preHandle方法返回为true，则执行下一拦截器的preHandle方法。如果当前拦截器返回为false，倒序执行所有已经执行了的拦截器的afterCompletion。
（3）如果任何一个拦截器返回false，执行返回，不执行目标方法。
（4）所有拦截器都返回true，执行目标方法。
（5）倒序执行所有拦截器的postHandle方法。
（6）前面的步骤有任何异常都会触发倒序执行afterCompletion方法。
（7）页面成功渲染后，也会倒序执行afterCompletion方法。
到此这篇关于SpringBoot拦截器以及源码的文章就介绍到这了,更多相关SpringBoot拦截器内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
