Polish gh-1

evgeniycheban · Aug 3, 2022 · f81c424 · f81c424
1 parent cdbe6c0
commit f81c424
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/org/springframework/data/reindexer/repository/query/ReindexerQueryMethod.java b/src/main/java/org/springframework/data/reindexer/repository/query/ReindexerQueryMethod.java
@@ -55,9 +55,9 @@ public final class ReindexerQueryMethod extends QueryMethod {
 	public ReindexerQueryMethod(Method method, RepositoryMetadata metadata, ProjectionFactory factory) {
 		super(method, metadata, factory);
 		this.isIteratorQuery = Lazy.of(() -> Iterator.class.isAssignableFrom(getReturnedObjectType()));
-		this.isOptionalQuery = Lazy.of(() -> Optional.class.isAssignableFrom(getReturnedObjectType()));
-		this.isListQuery = Lazy.of(() -> List.class.isAssignableFrom(getReturnedObjectType()));
-		this.isSetQuery = Lazy.of(() -> Set.class.isAssignableFrom(getReturnedObjectType()));
+		this.isOptionalQuery = Lazy.of(() -> Optional.class.isAssignableFrom(method.getReturnType()));
+		this.isListQuery = Lazy.of(() -> List.class.isAssignableFrom(method.getReturnType()));
+		this.isSetQuery = Lazy.of(() -> Set.class.isAssignableFrom(method.getReturnType()));
 		this.queryAnnotationExtractor = Lazy.of(() -> method.getAnnotation(Query.class));
 	}