Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Replace ReadCancellationException with takeWhile

Browse Source 
Closes gh-24125
This commit is contained in:
Rossen Stoyanchev
2019-12-04 18:17:46 +00:00
parent b866d4209c
commit 3a48682226
1 changed files with 6 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
spring-webflux/src/main/java/org/springframework/web/reactive/function/BodyExtractors.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2018 the original author or authors.
* Copyright 2002-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -264,18 +264,11 @@ public abstract class BodyExtractors {
() -> consumeAndCancel(message).then(Mono.empty()) : Mono::empty;
}
private static Mono<Void> consumeAndCancel(ReactiveHttpInputMessage message) {
return message.getBody()
.map(buffer -> {
DataBufferUtils.release(buffer);
throw new ReadCancellationException();
})
.onErrorResume(ReadCancellationException.class, ex -> Mono.empty())
.then();
}
@SuppressWarnings("serial")
private static class ReadCancellationException extends RuntimeException {
private static Flux<DataBuffer> consumeAndCancel(ReactiveHttpInputMessage message) {
return message.getBody().takeWhile(buffer -> {
DataBufferUtils.release(buffer);
return false;
});
}
}