如何拦截所有请求请求头中的信息

开启拦截器

在启动类上加上注解@ServletComponentScan

//basePackages配置拦截器所在的包名
@ServletComponentScan(basePackages = "com.xxx.xxx")

编写RequestHeaderInfo(用于存储请求中带的一些通用请求头)

@Data
public class RequestHeaderInfo {
    /**
     * 客户端ip
     */
    private String clientIp;
    /**
     * 发起请求的用户id
     */
    private Integer userId;
    /**
     * 调用url
     */
    private String url;
}

编写ThreadLocal类

import com.xxx.model.common.RequestHeaderInfo;
public class RequestHeaderInfoThreadLocal {

    private final static ThreadLocal<RequestHeaderInfo> REQUEST_HEADER_INFO_THREAD_LOCAL = new ThreadLocal<>();

    /**
     * 将请求头中带的信息放在ThreadLocal中
     * @param requestHeaderInfo
     */
    public static void setRequestHeaderInfo(RequestHeaderInfo requestHeaderInfo){
        REQUEST_HEADER_INFO_THREAD_LOCAL.set(requestHeaderInfo);
    }

    /**
     * 获取请求头中带的信息
     * @return
     */
    public static RequestHeaderInfo getRequestHeaderInfo(){
        return REQUEST_HEADER_INFO_THREAD_LOCAL.get();
    }

    /**
     * 清除ThreadLocal
     */
    public static void clearRequestHeaderInfo(){
        REQUEST_HEADER_INFO_THREAD_LOCAL.remove();
    }
}

编写拦截器类

import cn.hutool.core.util.StrUtil;
import com.xxx.common.RequestHeaderInfoThreadLocal;
import com.xxx.model.common.RequestHeaderInfo;
import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
@WebFilter(filterName = "requestHeaderInfoFilter", urlPatterns = "/*")
public class RequestHeaderInfoFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    /**
     * 提取请求头中的客户端ip信息和用户信息, 放在ThreadLocal中
     * @param request
     * @param response
     * @param chain
     * @throws IOException
     * @throws ServletException
     */
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {

        HttpServletRequest httpServletRequest = (HttpServletRequest) request;
        String clientIp = httpServletRequest.getHeader("clientIp");
        String userIdHeaderValue = httpServletRequest.getHeader("userId");
        Integer userId = StrUtil.isEmpty(userIdHeaderValue) ? null : Integer.valueOf(userIdHeaderValue);
        String url = httpServletRequest.getServletPath();
        RequestHeaderInfo requestHeaderInfo = new RequestHeaderInfo();
        requestHeaderInfo.setClientIp(clientIp);
        requestHeaderInfo.setUserId(userId);
        requestHeaderInfo.setUrl(url);
        RequestHeaderInfoThreadLocal.setRequestHeaderInfo(requestHeaderInfo);
        try {
            chain.doFilter(request, response);
        } finally {
            clearRequestHeaderInfo();
        }
    }

    private void clearRequestHeaderInfo() {
        RequestHeaderInfoThreadLocal.clearRequestHeaderInfo();
    }

    @Override
    public void destroy() {

    }
}

获取请求头信息

/*从请求头中获取用户id*/
Integer userId = RequestHeaderInfoThreadLocal.getRequestHeaderInfo().getUserId();

自此完成了拦截请求并获取请求头信息然后再放心请求的所有操作。

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值