新建一个类做拦截器,记住一点要加@Component注解
import com.alibaba.fastjson.JSON;
import org.springframework.http.HttpStatus;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
import sample.hgwl.common.utils.JsonResult;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
@Component
public class LoginInterceptor implements HandlerInterceptor {
/**
* 预处理回调方法,实现处理器预处理
* 返回值:true表示继续流程;false表示流程中断,不会继续调用其他的拦截器或者处理器
*/
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
System.out.println("我是拦截器,我被执行了");
Cookie[] cookies = request.getCookies();
if (StringUtils.isEmpty(cookies)){
JsonResult data = JsonResult.error(HttpStatus.UNAUTHORIZED.value() + "", "UNAUTHORIZED", HttpStatus.UNAUTHORIZED.name());
get(response, data);
return false;
}
return true;
}
private void get(HttpServletResponse response, Object data) {
PrintWriter writer = null;
try {
writer = response.getWriter();
writer.print(JSON.toJSONString(data));
writer.flush();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (writer != null) {
writer.close();
}
}
}
/**
* 后处理回调方法,实现处理器(controller)的后处理,但在渲染视图之前
* 此时我们可以通过modelAndView对模型数据进行处理或对视图进行处理
*/
@Override
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable ModelAndView modelAndView) throws Exception {
}
/**
* 整个请求处理完毕回调方法,即在视图渲染完毕时回调,
* 如性能监控中我们可以在此记录结束时间并输出消耗时间,
* 还可以进行一些资源清理,类似于try-catch-finally中的finally,
* 但仅调用处理器执行链中
*/
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable Exception ex) throws Exception {
}
}
我们在启动类中把拦截器配置进去
addInterceptor用于添加你自定义的拦截器实例
addPathPatterns用于添加要拦截的url,可以写多个。
excludePathPatterns用于添加不需要拦截的url,可以写多个
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
public static void main(String[] args) {
SpringApplication.run(EnterpriseApplication.class, args);
}
@Autowired
private CookieInterceptorAdapter cookieInterceptorAdapter;
@Autowired
private LoginInterceptor loginInterceptor;
@Override
protected void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(cookieInterceptorAdapter).addPathPatterns("/**")
.excludePathPatterns("/swagger-resources/**", "/webjars/**", "/v2/**", "/swagger-ui.html/**");
registry.addInterceptor(loginInterceptor).addPathPatterns("/**");
}