SpringBoot 拦截器

Mashiro

2022 年 10 月 06 日

546 次浏览

暂无评论

2213字数

博客

# SpringBoot 使用拦截器

## 引导类

继承 `WebMvcConfigurer` 接口，重写 `addInterceptors` 方法

```java
@SpringBootApplication
public class ClassScheduleApplication implements WebMvcConfigurer {

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

@Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LoginInterceptor());
    }
}
```

## 拦截器类 Interceptor

继承 `HandlerInterceptor` 接口

```java
public class LoginInterceptor implements HandlerInterceptor {
    private static final String USER_CODE = "userCode";
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String uri = request.getRequestURI();
        Cookie[] cookies = request.getCookies();
        String userCode = null;
        for (Cookie cookie : cookies) {
            if (USER_CODE.equals(cookie.getName())) {
                userCode = cookie.getValue();
            }
        }
        String[] uris = {"/Login.html", "/Register.html", "/users/register", "/users/login"};
        boolean flag = true;
        for (String s : uris) {
            if (s.equals(uri)) {
                flag = false;
                break;
            }
        }
        if (flag) {
            if (request.getSession().getAttribute(USER_CODE) != null && request.getSession().getAttribute(USER_CODE).equals(userCode)) {
                return true;
            }
            response.sendRedirect("/Login.html");
        } else {
            return true;
        }
        return false;
    }

@Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
    }

@Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        HandlerInterceptor.super.afterCompletion(request, response, handler, ex);
    }
}
```

最后修改：2022 年 10 月 06 日