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

#321 - Polishing.

Browse Source 
Add optimizations for void projections.
This commit is contained in:
Mark Paluch
2020-03-19 10:43:12 +01:00
parent b0348e272a
commit 048bcb8baa
1 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/main/java/org/springframework/data/r2dbc/repository/query/R2dbcQueryExecution.java
View File

@@ -15,6 +15,11 @@
*/ */
package org.springframework.data.r2dbc.repository.query; package org.springframework.data.r2dbc.repository.query;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import org.reactivestreams.Publisher;
import org.springframework.core.convert.converter.Converter; import org.springframework.core.convert.converter.Converter;
import org.springframework.data.convert.EntityInstantiators; import org.springframework.data.convert.EntityInstantiators;
import org.springframework.data.mapping.context.MappingContext; import org.springframework.data.mapping.context.MappingContext;
@@ -82,13 +87,31 @@ interface R2dbcQueryExecution {
@Override @Override
public Object convert(Object source) { public Object convert(Object source) {
ReturnedType returnedType = this.processor.getReturnedType(); ReturnedType returnedType = processor.getReturnedType();
if (Void.class == returnedType.getReturnedType() if (ClassUtils.isPrimitiveOrWrapper(returnedType.getReturnedType())) {
|| ClassUtils.isPrimitiveOrWrapper(returnedType.getReturnedType())) {
return source; return source;
} }
if (!mappingContext.hasPersistentEntityFor(returnedType.getReturnedType())) {
return source;
}
if (Void.class == returnedType.getReturnedType()) {
if (source instanceof Mono) {
return ((Mono<?>) source).then();
}
if (source instanceof Publisher) {
return Flux.from((Publisher<?>) source).then();
}
if (ClassUtils.isPrimitiveOrWrapper(returnedType.getReturnedType())) {
return source;
}
}
Converter<Object, Object> converter = new DtoInstantiatingConverter(returnedType.getReturnedType(), Converter<Object, Object> converter = new DtoInstantiatingConverter(returnedType.getReturnedType(),
this.mappingContext, this.instantiators); this.mappingContext, this.instantiators);