打印请求头

	Map<String,Object> headerMap = new HashMap<>();
				Enumeration<String> headerNames = request.getHeaderNames();

				while (headerNames.hasMoreElements()) {
					String headerName = headerNames.nextElement();
					Enumeration<String> headers = request.getHeaders(headerName);
					while (headers.hasMoreElements()) {
						String headerValue = headers.nextElement();
						headerMap.put(headerName, headerValue);
					}

				}
  1. 请求头拦截器Interceptor
     implements HandlerInterceptor  重写方法
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import java.lang.reflect.Field;
import java.util.Enumeration;

@Slf4j
public class MyInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        MyHeader myHeader = new MyHeader();
        MyHeaderEnum[] myHeaderEnums =  MyHeaderEnum.values();
        for (MyHeaderEnum myHeaderEnum : myHeaderEnums){
            String headerValue = request.getHeader(myHeaderEnum.getVal());
            setMyHeaderFiled(myHeader,myHeaderEnum,headerValue);
        }
        log.info("myHeader:{}",myHeader);
        SecurityUtils.setMyHeader(myHeader);
        return true;
    }

    private void setMyHeaderFiled(MyHeader myHeader, MyHeaderEnum myHeaderEnum, String headerValue) throws Exception {
        String filedName = myHeaderEnum.getVal();
        Class<?> clazz = MyHeader.class;
        Field field = clazz.getDeclaredField(filedName);
        field.setAccessible(true);
        field.set(myHeader,headerValue);

    }

    private String getRequestHeaders(HttpServletRequest request){
        Enumeration<String> headerNames = request.getHeaderNames();
        StringBuilder headers = new StringBuilder();
        while (headerNames.hasMoreElements()){
            String headerName = headerNames.nextElement();
            String headerValue = request.getHeader(headerName);
            headers.append(headerName).append(": ").append(headerValue).append(", ");
        }
        if (headers.length() > 0){
            headers.setLength(headers.length() - 2);
        }
        return headers.toString();
    }
    @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 {
        SecurityUtils.clearMyHeader();
    }
}

在配置类中注入拦截器

WebConfig
@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(clientRequestInterceptor());
    }

    @Bean
    @Scope(value = "request", proxyMode = ScopedProxyMode.TARGET_CLASS)
    public MyHeader myHeader(){
        return new MyHeader();
    }

    @Bean
    public MyInterceptor clientRequestInterceptor(){
        return new MyInterceptor();
    }
}

请求头的枚举类、实体类

@Getter
@AllArgsConstructor
public enum MyHeaderEnum {
    USERID("userId"),
    ORGID("orgId");

    private final String val;
}

存进线程池


public class SecurityUtils {
    private static final ThreadLocal<MyHeader> myHeaderThreadLocal = new ThreadLocal<>();

    public static MyHeader getMyHeader(){
        return myHeaderThreadLocal.get();
    }

    public static void setMyHeader(MyHeader myHeader){
        myHeaderThreadLocal.set(myHeader);
    }

    public static void clearMyHeader(){
        myHeaderThreadLocal.remove();
    }
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值