当前位置:   article > 正文

Gateway网关拦截器的ServerWebExchange

serverwebexchange

 ServerWebExcahnge的解释如下:

Contract for an HTTP request-response interaction. Provides access to the HTTP request and response and also exposes additional server-side processing related properties and features such as request attributes.

大概是说:ServerWebExchange是一个HTTP请求-响应交互的契约。提供对HTTP请求和响应的访问,并公开额外的服务器端处理相关属性和特性,如请求属性,有点像Context的角色。

在 HTTP 协议中,请求头和请求体是请求的重要组成部分,ServerWebExchange 提供了丰富的方法来对其进行处理。

请求数据读取

例如,可以通过 ServerWebExchange.getRequest().getHeaders() 方法获取请求头:

  1. ServerHttpRequest request = exchange.getRequest();
  2. HttpHeaders headers = request.getHeaders();
  3. String userAgent = headers.getFirst(HttpHeaders.USER_AGENT);

可以通过 ServerWebExchange.getRequest().getBody() 方法获取请求体:

  1. ServerHttpRequest request = exchange.getRequest();
  2. Mono<String> requestBody = request.getBody().map(dataBuffer -> {
  3. byte[] buffer = new byte[dataBuffer.readableByteCount()];
  4. dataBuffer.read(buffer);
  5. DataBufferUtils.release(dataBuffer);
  6. Charset charset = Charset.forName("UTF-8");
  7. return new String(buffer, charset);
  8. });

这里使用 BodyExtractors.toMono(String.class) 也可以获取请求体,但是这种方式只适用于请求体是 JSON、XML 等文本格式的情况。而如果请求体是二进制文件,则需要使用上面的方式来获取。

简单的理解是,ServerWebExcahnge包含了http请求的请求数据(ServerHttpRequest),返回数据(ServerHttpResponse)。

从新封装请求

       ServerHttpRequest是一个只读类,对于读多写少的场景,这种设计模式是值得借鉴的。但是有时候我们也需要修改。则可以通过以下实现修改:

  1. ServerHttpRequest serverHttpRequest = exchange.getRequest().mutate().path(newPath).method(HttpMethod.GET).build();
  2. ServerWebExchange serverWebExchange = exchange.mutate().request(serverHttpRequest).build();
  3. return chain.filter(exchange);

     但是请求中的body参数流(inputstream)只能被读取一次,但是,我们通常由很多个过滤器,所以可能需要读取多次消息。解决这个办法需要用到官方提供的:ContentCachingRequestWrapper类。

  1. public ContentCachingRequestWrapper(HttpServletRequest request) {
  2.     super(request);
  3.     int contentLength = request.getContentLength();
  4.     this.cachedContent = new ByteArrayOutputStream(contentLength >= 0 ? contentLength : 1024);
  5.     this.contentCacheLimit = null;
  6. }

包装ContentCachingRequestWrapper,主要就做了一件事儿,感知request里面内容的长度,为自己开一个同样大小的输出流cachedContent。然后重写其中的方法:

  1. //重写方法getInputStream
  2. @Override
  3. public ServletInputStream getInputStream() throws IOException {
  4.     if (this.inputStream == null) {
  5.         this.inputStream = new ContentCachingInputStream(getRequest().getInputStream());
  6.     }
  7.     return this.inputStream;
  8. }
  9. //重写java.io.InputStream#read()方法
  10. @Override
  11. public int read() throws IOException {
  12. int ch = this.is.read();
  13. if (ch != -1 && !this.overflow) {
  14. if (contentCacheLimit != null && cachedContent.size() == contentCacheLimit) {
  15. this.overflow = true;
  16. handleContentOverflow(contentCacheLimit);
  17. }
  18. else {
  19. cachedContent.write(ch);
  20. }
  21. }
  22. return ch;
  23. }
  24. //重写方法inputStream中的read(byte b[], int off, int len)
  25. @Override
  26. public int read(final byte[] b, final int off, final int len) throws IOException {
  27. int count = this.is.read(b, off, len);
  28. writeToCache(b, off, count);
  29. return count;
  30. }

总之,ContentCachingRequestWrapper通过覆盖inpuStream方法,通过移花接木的方式,将每次调用inputStream读的时候都将其放置到输出流当中。这样后续inputStream流被关闭了,我们还可以通过使用cachedContent内容读取入参内容。

构造返回数据

  1. private void doResponse(ServletResponse servletResponse) throws IOException{
  2. servletResponse.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
  3. HttpServletResponse httpServletResponse = (HttpServletResponse)servletResponse;
  4. httpServletResponse.setStatus(401);
  5. PrintWriter writer = servletResponse.getWriter();
  6. writer.write("xxxxx");
  7. }
  8. private void doFileResponse(ServletResponse servletResponse) throws IOException{
  9. //1. 用于读取文件
  10. FileInputStream fis = new FileInputStream("d://xxx/xxx/a.jpg");
  11. //2. 获取response字节输出流
  12. ServletOutputStream os = servletResponse.getOutputStream();
  13. //3. 完成流的复制
  14. byte[] buff = new byte[1024];
  15. int len = 0;
  16. while ((len = fis.read(buff))!= -1){
  17. os.write(buff,0,len);
  18. }
  19. fis.close();
  20. }

读取返回数据:

https://www.cnblogs.com/fdzang/p/11812348.html

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/301410
推荐阅读
相关标签
  

闽ICP备14008679号