自定义拦截器HandlerInterceptor

@Component
public class MyInterceptor implements HandlerInterceptor {

    private static Logger logger = LoggerFactory.getLogger(MyInterceptor.class);

    @Autowired
    private CacheService cacheService;

    private static final String UNAUTHORIZED_URI = "/unauthorized";

    private static final String AUTHORIZATION_INVALID_URI = "/auth/invalid";

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        String referer = request.getHeader("referer");
        try {

            String isDevModel = cacheService.isDevModel();

            if (StringUtil.isBlank(referer)) {
                // 非测试模式,无访问来源
                if (!Constants.DEV_MODEL.equalsIgnoreCase(isDevModel)) {
                    logger.warn(">>>>>> referer is blank:" +
                            ",RequestURL=" + request.getRequestURL() +
                            ",QueryString=" + request.getQueryString() +
                            ",RemoteAddr=" + request.getRemoteAddr());
                    return false;
                }
            }

            // 非开发模式需要校验访问来源
            if (!Constants.DEV_MODEL.equalsIgnoreCase(isDevModel)) {
                // 域名
                String requestUrl = referer.split("/")[2];

                // 白名单
                List<String> whiteList = cacheService.getwhiteList();

                boolean isInWhiteList = false;
                if (whiteList .contains(requestUrl)) {
                    isInWhiteList = true;
                }

                if (isInWhiteList == false) {
                    logger.warn(">>>>>> Access source addresses are not on the whitelist:referer=" + referer +
                            ",RequestURL=" + request.getRequestURL() +
                            ",QueryString=" + request.getQueryString() +
                            ",RemoteAddr=" + request.getRemoteAddr());
                    response.sendRedirect(UNAUTHORIZED_URI);
                    return false;
                }
            }

            String requestURI = request.getRequestURI();

            Class<?> clazz = ((HandlerMethod) handler).getBean().getClass();
            String className = ((HandlerMethod) handler).getBean().getClass().getName();
            String methodName = ((HandlerMethod) handler).getMethod().getName();

            logger.info(">>>>>> clazz:" + clazz + ", className :" + className + ", methodName :" + methodName );

            if (LoginController.class == clazz) {
                return true;
            } else {

                String token = request.getHeader(Constants.ACCESS_TOKEN);
                if (StringUtil.isBlank(token)) {
                    token = request.getParameter(Constants.ACCESS_TOKEN);
                }

                if (StringUtil.isBlank(token)) {
                    logger.warn(">>>>>> The token is blank:  referer=" + referer +
                            ",RequestURL=" + request.getRequestURL() +
                            ",QueryString=" + request.getQueryString() +
                            ",RemoteAddr=" + request.getRemoteAddr());
                    response.sendRedirect(UNAUTHORIZED_URI);
                    return false;
                }

                if (!ActionContextUtil.isExists(token)) {
                    logger.warn(">>>>>> The token is invalid:  eferer=" + referer +
                            ",RequestURL=" + request.getRequestURL() +
                            ",QueryString=" + request.getQueryString() +
                            ",RemoteAddr=" + request.getRemoteAddr());
                    response.sendRedirect(AUTHORIZATION_INVALID_URI);
                    return false;
                }

                ActionContext context = ActionContextUtil.getActionContext(token);
                ActionContextUtil.setContext(context);
                return true;
            }
        } catch (Exception e) {
            ResultBean resultBean = new ResultBean();
            resultBean.setStatus(StatusEnum.SYSTEM_ERROR);
            response.getOutputStream().write(JSONUtil.convertObjectToJSON(resultBean).getBytes(StandardCharsets.UTF_8));
            logger.error(">>>>>> error referer=" + referer +
                    ",RequestURL=" + request.getRequestURL() +
                    ",QueryString=" + request.getQueryString() +
                    ",RemoteAddr=" + request.getRemoteAddr(), e);
            return false;
        }
    }

    @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 {

    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Java的拦截HandlerInterceptor允许你在请求处理之前或之后自定义返回结果。你可以通过实现HandlerInterceptor接口并重写其中的三个方法来实现自定义拦截: ``` public interface HandlerInterceptor { // 请求处理之前调用 boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception; // 请求处理之后调用,但是在视图被渲染之前(Controller方法调用之后) void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception; // 在整个请求结束之后调用,也就是在DispatcherServlet渲染了对应的视图之后执行(主要是用于进行资源清理工作) void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception; } ``` 你可以在这些方法中自定义返回结果。例如,你可以在`preHandle`方法中设置响应头,或者在`postHandle`方法中设置响应正文。你也可以在`afterCompletion`方法中执行资源清理操作。 要使用自定义拦截,你需要将其配置到Spring MVC的DispatcherServlet中。你可以通过实现WebMvcConfigurer接口并重写`addInterceptors`方法来完成此操作。 ``` @Configuration public class WebMvcConfig implements WebMvcConfigurer { @Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(new MyInterceptor()) .addPathPatterns("/**") .excludePathPatterns("/login"); } } ``` 在上面的例子中,我们将

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值