如何重写HttpServletRequest的Header
场景
在使用网关对请求头中的token进行鉴权之后,希望将鉴权后的用户信息,写入到request的header中;此时,一般的HttpServletRequest是没有提供直接写入header的方法的。
此时或许有人会退而求其次,不将信息放到header中,而放到key-value参数里面;但如果我们的服务使用的是JSON格式接收数据,相关信息就不太好处理了。所以建议还是放到请求头里去。
方案
虽然HttpServletRequest没有直接提供addHeader的方法,但是HttpServletRequestWrapper已经实现了HttpServletRequest的接口,故我们可以通过自定义子类继承HttpServletRequestWrapper,重写getHeaders的方法,再将自定义子类充当 HttpServletRequest返回或 放入上下文RequestContext中即可。
自定义子类
/**
* 自定义子类,核心点在于自定义header的map,再重写获取header的方法,将自定义的map也一起返回
* 重写HttpServletRequest的请求头
* @author: songwenguang
* @time: 2021/3/17 14:11
*/
@Slf4j
public class HeaderMapRequestWrapper extends HttpServletRequestWrapper {
/**
* 构造方法,可保持原有的request内容不变
*
*
版权声明:本文为q410654146原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。