root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Allow repeatable writes in Client Interceptors 

See gh-31449
This commit is contained in:
Arjen Poutsma 2024-02-14 10:05:26 +01:00
parent 48da9524c3
commit b4bec4ca61
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
spring-web/src/main/java/org/springframework/http/client/InterceptingClientHttpRequest.java
View File

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2022 the original author or authors. * Copyright 2002-2024 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -17,6 +17,7 @@
package org.springframework.http.client; package org.springframework.http.client;
import java.io.IOException; import java.io.IOException;
import java.io.OutputStream;
import java.net.URI; import java.net.URI;
import java.util.Iterator; import java.util.Iterator;
import java.util.List; import java.util.List;
@ -92,7 +93,17 @@ class InterceptingClientHttpRequest extends AbstractBufferingClientHttpRequest {
request.getHeaders().forEach((key, value) -> delegate.getHeaders().addAll(key, value)); request.getHeaders().forEach((key, value) -> delegate.getHeaders().addAll(key, value));
if (body.length > 0) { if (body.length > 0) {
if (delegate instanceof StreamingHttpOutputMessage streamingOutputMessage) { if (delegate instanceof StreamingHttpOutputMessage streamingOutputMessage) {
streamingOutputMessage.setBody(outputStream -> StreamUtils.copy(body, outputStream)); streamingOutputMessage.setBody(new StreamingHttpOutputMessage.Body() {
@Override
public void writeTo(OutputStream outputStream) throws IOException {
StreamUtils.copy(body, outputStream);
}
@Override
public boolean repeatable() {
return true;
}
});
} }
else { else {
StreamUtils.copy(body, delegate.getBody()); StreamUtils.copy(body, delegate.getBody());