springboot 添加请求头（反射实现）

我会带着你远行 2023-07-17 07:22 96阅读 0赞

springboot 添加请求头（反射实现）

#

******************************

相关类及接口

org.apache.catalina.connector.RequestFacade

  1. public class RequestFacade implements HttpServletRequest {
  2.     protected Request request = null;
  3.     protected static final StringManager sm = StringManager.getManager(RequestFacade.class);
  5.     public RequestFacade(Request request) {
  6.         this.request = request;
  7.     }

org.apache.catalina.connector.Request

  1. public class Request implements HttpServletRequest {
  2.     private static final Log log = LogFactory.getLog(Request.class);
  3.     protected org.apache.coyote.Request coyoteRequest;

org.apache.coyote.Request

  1. public final class Request {
  3.     private final MimeHeaders headers = new MimeHeaders();

org.apache.tomcat.util.http.MimeHeaders

  1. public class MimeHeaders {
  3. ***************
  4. 相关方法
  6.     public void removeHeader(String name) {
  7.     public MessageBytes addValue(String name) {

org.apache.tomcat.util.buf.MessageBytes

  1. public final class MessageBytes implements Cloneable, Serializable {
  3.     public void setString(String s) {

******************************

示例

**********************

filter 层

CustomFilter

  1. @Component
  2. public class CustomFilter implements HandlerInterceptor {
  4.     @Override
  5.     public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
  6.         Map<String,String> map=new HashMap<>();
  7.         map.put("name","瓜田李下");
  8.         addHeader(request,map);
  10.         return true;
  11.     }
  13.     private void addHeader(HttpServletRequest request, Map<String,String> headerMap){
  14.         if (headerMap==null||headerMap.isEmpty()){
  15.             return;
  16.         }
  18.         Class<? extends HttpServletRequest> c=request.getClass();
  19.         System.out.println(c.getName());
  21.         try{
  22.             Field requestField=c.getDeclaredField("request");
  23.             requestField.setAccessible(true);
  25.             Object o=requestField.get(request);
  26.             Field coyoteRequest=o.getClass().getDeclaredField("coyoteRequest");
  27.             coyoteRequest.setAccessible(true);
  29.             Object o2=coyoteRequest.get(o);
  30.             Field headers=o2.getClass().getDeclaredField("headers");
  31.             headers.setAccessible(true);
  33.             MimeHeaders mimeHeaders=(MimeHeaders) headers.get(o2);
  34.             for (Map.Entry<String,String> entry:headerMap.entrySet()){
  35.                 mimeHeaders.removeHeader(entry.getKey());
  36.                 mimeHeaders.addValue(entry.getKey()).setString(entry.getValue());
  37.             }
  39.         }catch (Exception e){
  40.             e.printStackTrace();
  41.         }
  42.     }
  43. }

**********************

config 层

WebMvcConfig

  1. @Configuration
  2. public class WebMvcConfig implements WebMvcConfigurer {
  4.     @Resource
  5.     private CustomFilter customFilter;
  7.     @Override
  8.     public void addInterceptors(InterceptorRegistry registry) {
  9.         registry.addInterceptor(customFilter).addPathPatterns("/hello");
  10.     }
  11. }

**********************

controller 层

HelloController

  1. @RestController
  2. public class HelloController {
  4.     @RequestMapping("/hello")
  5.     public String hello3(HttpServletRequest request){
  6.         String name=request.getHeader("name");
  7.         if (name!=null){
  8.             System.out.println(name);
  9.         }
  11.         return "hello world";
  12.     }
  14. }

************************

控制台输出

  1. org.apache.catalina.connector.RequestFacade
  2. 瓜田李下

发表评论

表情：
评论列表 （有 0 条评论，96人围观）

还没有评论，来说两句吧...

相关阅读