Merge branch '6.1.x'

sbrannen · Apr 15, 2024 · fe83003 · fe83003
2 parents 5012843 + 1e54916
commit fe83003
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/...ng-web/src/main/java/org/springframework/web/server/adapter/DefaultServerWebExchange.java b/...ng-web/src/main/java/org/springframework/web/server/adapter/DefaultServerWebExchange.java
@@ -24,7 +24,9 @@
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.Function;
+import java.util.stream.Collectors;
 
+import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
 import org.springframework.context.ApplicationContext;
@@ -249,20 +251,21 @@ public Mono<MultiValueMap<String, Part>> getMultipartData() {
 	public Mono<Void> cleanupMultipart() {
 		return Mono.defer(() -> {
 			if (this.multipartRead) {
-				return getMultipartData()
-						.onErrorComplete()
-						.flatMapIterable(Map::values)
-						.flatMapIterable(Function.identity())
-						.flatMap(part -> part.delete()
-									.onErrorComplete())
-						.then();
+				return Mono.usingWhen(getMultipartData().onErrorComplete().map(this::collectParts),
+						parts -> Mono.empty(),
+						parts -> Flux.fromIterable(parts).flatMap(part -> part.delete().onErrorComplete())
+				);
 			}
 			else {
 				return Mono.empty();
 			}
 		});
 	}
 
+	private List<Part> collectParts(MultiValueMap<String, Part> multipartData) {
+		return multipartData.values().stream().flatMap(List::stream).collect(Collectors.toList());
+	}
+
 	@Override
 	public LocaleContext getLocaleContext() {
 		return this.localeContextResolver.resolveLocaleContext(this);