经过zuul添加或修改请求参数

原文地址:https://blog.csdn.net/J_bean/article/details/81584979

一. 为何要用到这个java

  在基于 springcloud 构建的微服务系统中,一般使用网关zuul来进行一些用户验证等过滤的操做,好比 用户在 header 或者 url 参数中存放了 token ,网关层须要 用该 token 查出用户 的 userId ,并存放于 request 中,以便后续微服务能够直接使用而避免再去用 token 查询。spring

二.基础知识json

    在 zuul 中最大的用法的除了路由以外,就是过滤器了,自定义过滤器需实现接口 ZuulFilter ,在 run() 方法中,能够用app

RequestContext ctx = RequestContext.getCurrentContext();
HttpServletRequest request = ctx.getRequest();

获取到 request,可是在 request 中只有 getParameter() 而没有 setParameter() 方法,因此直接修改 url 参数不可行,另外在 reqeust 中虽然可使用 setAttribute() ,可是可能因为做用域的不一样,在这里设置的 attribute 在后续的微服务中是获取不到的,所以必须考虑另外的方式。ide

三.具体作法微服务

最后肯定的可行的方法是,用url

ctx.setRequest(new HttpServletRequestWrapper(request) {})

的方式,从新构造上下文中的 request ,代码以下:.net

// 例如在请求参数中添加 userId
try {
  InputStream in = ctx.getRequest().getInputStream();
  String body = StreamUtils.copyToString(in, Charset.forName("UTF-8"));
  if(StringUtils.isBlank(body)){
    body = "{}";
  }
  JSONObject jsonObject = JSON.parseObject(body);
  jsonObject.put("userId", 666);
  String newBody = jsonObject.toString();
  final byte[] reqBodyBytes = newBody.getBytes();
  ctx.setRequest(new HttpServletRequestWrapper(request){    
    @Override
    public ServletInputStream getInputStream() throws IOException {
      return new ServletInputStreamWrapper(reqBodyBytes);
    }
    @Override
    public int getContentLength() {
      return reqBodyBytes.length;
    }
    @Override
    public long getContentLengthLong() {
      return reqBodyBytes.length;
    }
  });
} catch (IOException e) {
  e.printStackTrace();
}

思路就是,获取请求的输入流,并重写,即重写json参数。code

在后续的微服务的 controller 中,经过下面的方式获取经过zuul添加或修改的请求参数。blog

InputStream in = request().getInputStream();
String body = StreamUtils.copyToString(in, Charset.forName("UTF-8"));
if(StringUtils.isNotBlank(body)){
  JSONObject jsonObject = JSON.parseObject(body);
  Object userId = jsonObject.get("userId");
}

 

posted @ 2021-11-29 21:57  八方鱼  阅读(322)  评论(0编辑  收藏  举报