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

DATAMONGO-2406 - Polishing.

Browse Source 
Add optimization for Mono.

Original pull request: #825.
This commit is contained in:
Mark Paluch
2020-01-29 09:56:54 +01:00
parent 80da9e21ed
commit 0a2ea88f3c
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
spring-data-mongodb/src/main/java/org/springframework/data/mongodb/repository/query/ReactiveMongoQueryExecution.java
View File

@@ -18,8 +18,10 @@ package org.springframework.data.mongodb.repository.query;
import lombok.NonNull;
import lombok.RequiredArgsConstructor;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import org.reactivestreams.Publisher;
import org.springframework.core.convert.converter.Converter;
import org.springframework.data.convert.EntityInstantiators;
import org.springframework.data.domain.Pageable;
@@ -169,8 +171,15 @@ interface ReactiveMongoQueryExecution {
ReturnedType returnedType = processor.getReturnedType();
if (returnsMonoVoid(returnedType)) {
return Flux.from((Publisher) source).then();
if (isVoid(returnedType)) {
if (source instanceof Mono) {
return ((Mono<?>) source).then();
}
if (source instanceof Publisher) {
return Flux.from((Publisher<?>) source).then();
}
}
if (ClassUtils.isPrimitiveOrWrapper(returnedType.getReturnedType())) {
@@ -188,7 +197,7 @@ interface ReactiveMongoQueryExecution {
}
}
static boolean returnsMonoVoid(ReturnedType returnedType) {
return returnedType.getReturnedType() == Void.class;
static boolean isVoid(ReturnedType returnedType) {
return returnedType.getReturnedType().equals(Void.class);
}
}