Spring Boot 中的拦截器 Interceptors

博客主页: 南来_北往

系列专栏:Spring Boot实战


前言

Spring Boot中的拦截器(Interceptor)是一种用于拦截和处理HTTP请求的机制,它基于Spring MVC框架中的HandlerInterceptor接口实现。拦截器允许在请求到达控制器(Controller)之前或之后执行一些共享的逻辑。

自定义拦截器案例

下面是一个自定义拦截器的示例,用于记录请求的开始时间和结束时间:

java 复制代码
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class LogInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        long startTime = System.currentTimeMillis();
        request.setAttribute("startTime", startTime);
        return true; // 继续执行后续处理
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        long endTime = System.currentTimeMillis();
        long startTime = (Long) request.getAttribute("startTime");
        long executionTime = endTime - startTime;
        System.out.println("Request URL: " + request.getRequestURL());
        System.out.println("Execution time: " + executionTime + "ms");
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        // 可选操作,在请求完成后执行一些清理工作
    }
}

接下来,我们需要将这个拦截器注册到Spring MVC的拦截器链中。可以通过实现WebMvcConfigurer接口并重写addInterceptors方法来实现:

java 复制代码
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 WebConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LogInterceptor())
                .addPathPatterns("/api/**") // 指定拦截路径模式
                .excludePathPatterns("/api/login"); // 排除登录路径
    }
}

在上面的代码中,我们创建了一个名为LogInterceptor的拦截器类,并在WebConfig类中将其注册到拦截器链中。通过addPathPatterns方法指定了需要拦截的URL路径模式,而excludePathPatterns方法则排除了不需要拦截的路径。在这个例子中,我们将拦截所有以"/api/"开头的请求,但排除了"/api/login"路径。

当有请求匹配到指定的路径模式时,LogInterceptor中的preHandle方法会在请求到达控制器之前被调用,postHandle方法会在请求处理完成后被调用。在这些方法中,我们可以添加自定义的逻辑来记录请求的开始时间和结束时间,并计算请求的处理时间。

注册拦截器案例

下面是一个注册拦截器的示例,用于检查用户是否已登录:

java 复制代码
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class LoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 检查用户是否已登录
        if (request.getSession().getAttribute("user") == null) {
            // 如果用户未登录,重定向到登录页面
            response.sendRedirect("/login");
            return false; // 阻止请求继续执行后续处理
        }
        return true; // 继续执行后续处理
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        // 可选操作,在请求完成后执行一些清理工作
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        // 可选操作,在请求完成后执行一些清理工作
    }
}

接下来,我们需要将这个拦截器注册到Spring MVC的拦截器链中。可以通过实现WebMvcConfigurer接口并重写addInterceptors方法来实现:

java 复制代码
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 WebConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LoginInterceptor())
                .addPathPatterns("/api/**") // 指定拦截路径模式
                .excludePathPatterns("/api/login", "/api/register"); // 排除登录和注册路径
    }
}

在上面的代码中,我们创建了一个名为LoginInterceptor的拦截器类,并在WebConfig类中将其注册到拦截器链中。通过addPathPatterns方法指定了需要拦截的URL路径模式,而excludePathPatterns方法则排除了不需要拦截的路径。在这个例子中,我们将拦截所有以"/api/"开头的请求,但排除了"/api/login"和"/api/register"路径。

当有请求匹配到指定的路径模式时,LoginInterceptor中的preHandle方法会在请求到达控制器之前被调用。在这个方法中,我们检查用户的登录状态,如果用户未登录,则重定向到登录页面并阻止请求继续执行后续处理。如果用户已登录,则允许请求继续执行后续处理。

总结

总结来说,Spring Boot中的拦截器是一种强大的工具,可以在请求处理的不同阶段执行自定义逻辑。通过实现HandlerInterceptor接口并注册到拦截器链中,可以轻松地添加日志记录、权限检查、性能监控等功能。

相关推荐
Algorithm157616 分钟前
谈谈接口和抽象类有什么区别?
java·开发语言
细心的莽夫1 小时前
SpringCloud 微服务复习笔记
java·spring boot·笔记·后端·spring·spring cloud·微服务
IT瘾君2 小时前
JavaWeb:Html&Css
前端·html
264玫瑰资源库2 小时前
问道数码兽 怀旧剧情回合手游源码搭建教程(反查重优化版)
java·开发语言·前端·游戏
pwzs2 小时前
Java 中 String 转 Integer 的方法与底层原理详解
java·后端·基础
喝拿铁写前端2 小时前
从圣经Babel到现代编译器:没开玩笑,普通程序员也能写出自己的编译器!
前端·架构·前端框架
东阳马生架构2 小时前
Nacos简介—2.Nacos的原理简介
java
HED3 小时前
VUE项目发版后用户访问的仍然是旧页面?原因和解决方案都在这啦!
前端·vue.js
普if加的帕3 小时前
java Springboot使用扣子Coze实现实时音频对话智能客服
java·开发语言·人工智能·spring boot·实时音视频·智能客服
拉不动的猪3 小时前
前端自做埋点,我们应该要注意的几个问题
前端·javascript·面试