Springboot 实现拦截器获取header内容
项目中遇到一个需求，对接上游系统是涉及到需要增加请求头，请求头的信息是动态获取的，需要动态从下游拿到之后转给上游。
分析
既然需要动态获取那么只有两种方式：要么每次下游请求过来时从请求头中获取，要么定义统一的拦截器自动获取。
实现
那么我们就先来实现一下吧。
第一种比较简单，直接使用springboot获取请求头的方式，从controller方法入口处使用： @RequestHeader(value = "xxxx",required = false) String appUser的方式获取请求头
代码如下:
@RequestMapping(name = "获取用户详情信息",value = "/getUserDetail",method=GET)
public String getUserDetail(@RequestHeader(value = "app-user",required = false) String appUser, @RequestParam(name = "search") String search){
log.info("------>拿到的请求头是："+appUser);
return "SUCCESS";
}此种方法虽然可以获取，但是太过low b，而且如果有几十个接口那么都要这么写，过于累赘
还是使用拦截器的方式吧
具体如下：
首先，定义一个拦截器，需要实现HandlerInterceptor，其中包含preHandler以及afterCompletion分别对应调用前以及调用完成后
package com.xxxx.cc.admin.interceptor;
import com.alibaba.fastjson.JSONObject;
import com.ddmc.cc.admin.support.ThreadLocalUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
* Copyright (C), 2017-2021
* Author: sjugg
* Date: 2021/5/19 10:39
* FileName: AuthInterceptor
* Description: 交易前置拦截器
*/
@Component
@Slf4j
public class AuthInfoInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
String userInfo = request.getHeader("app-user");
log.info(">>>>>>>拦截到api相关请求头<<<<<<<<"+userInfo);
if(StringUtils.isNotEmpty(userInfo)){
//直接搂下来，放到ThreadLocal中 后续直接从中获取
ThreadLocalUtils.set(MaiCaiApiConstants.MAI_CAI_APP_USER,userInfo);
}
return true;//注意 这里必须是true否则请求将就此终止。
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
//移除app-user
ThreadLocalUtils.remove("app-user");
log.info("移除请求头中的app-user："+ThreadLocalUtils.get("app-user"));
}
}后面只要从ThreadLocal中按key去拿就可以了。
另外需要做的是
添加web过滤器
1、添加web 过滤器，管理过滤的类，以及过滤的请求路径
2、继承WebMvcConfigurationSupport 类 和上面一样Generate,复写父类的方法 addInterceptors
3、通过registry.addInterceptor拦截类，addPathPatterns拦截路径，excludePathPatterns添加允许路径
这里我拦截的是以xxxx开头的请求。
package com.ddmc.cc.admin.config;
import com.ddmc.cc.admin.interceptor.AuthInfoInterceptor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
/**
*
*
* @author sjugg
* @since 2021/5/13 4:43 下午
*/
@Configuration
public class WebConfig implements WebMvcConfigurer {
@Autowired
private AuthInfoInterceptor authInfoInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(authInfoInterceptor).addPathPatterns("/xxxx/*");
}
}执行效果如下：
请求头里设置app-user信息
可以看到Controller中已经拿到请求头里的信息
获取head的代码：
@RequestMapping(name = "获取用户详情信息",value = "/getUserDetail",method=GET)
public String getUserDetail(@RequestParam(name = "search") String search){
log.info("-----> heard info here: "+ThreadLocalUtils.get("app-user"));
return "SUCCESS";
} 
方法执行完之后对应的threadlocal变量也被清除
