java - 部分消耗 Publisher

Question

我正在编写一个org.springframework.cloud.gateway.filter.GatewayFilter(spring-cloud-gateway)，它尝试使用遵循特定任意协议(protocol)的响应。响应正文的内容如下:

200\n
header1:value1\n
header2:header2\n
\n
<the actual body>

这个想法是，来自下游服务的整个响应元数据都在响应主体中(很像 SOAP 信封封装真实主体的方式)。网关的客户端应该收到修改后的解包响应。

我一直在遵循 ModifyRequestBodyGatewayFilterFactory 和 ModifyResponseBodyGatewayFilterFactory 作为指导，但我认为它们不适合我的用例。

我相信我可以通过返回ServerHttpResponseDecorator的子类来实现这一点，但我还不能完全理解如何实现:

@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
   // unwrap status code
   // unwrap headers
   // let it continue as usual 
}

Answer 1

我能够通过以下方式实现这一目标:

@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
   return Mono.from(body).flatMap { buffer -> 
      val size = readUtf8Line(buffer)
      // recompute 'Content-Length', if applicable

      super.writeWith(Mono.just(buffer))
   }
}

readUtf8Line 只是我自己的方法，它消耗 DataBuffer 直到到达换行符。返回读取的字节数(即 size)也很重要，因为需要重新计算 Content-Length(如果存在)。