diff --git a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/CassandraTemplate.java b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/CassandraTemplate.java index a8a046c65..aa5390bf1 100644 --- a/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/CassandraTemplate.java +++ b/spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/CassandraTemplate.java @@ -19,7 +19,6 @@ import java.util.List; import java.util.function.Consumer; import java.util.function.Function; import java.util.stream.Stream; -import java.util.stream.StreamSupport; import org.springframework.beans.BeansException; import org.springframework.context.ApplicationContext; @@ -374,10 +373,8 @@ public class CassandraTemplate implements CassandraOperations, ApplicationEventP Assert.notNull(statement, "Statement must not be null"); Assert.notNull(entityClass, "Entity type must not be null"); - ResultSet resultSet = getCqlOperations().queryForResultSet(statement); - - return StreamSupport.stream(resultSet.spliterator(), false) - .map(getMapper(entityClass, entityClass, EntityQueryUtils.getTableName(statement))); + Function mapper = getMapper(entityClass, entityClass, EntityQueryUtils.getTableName(statement)); + return getCqlOperations().queryForStream(statement, (row, rowNum) -> mapper.apply(row)); } // ------------------------------------------------------------------------- @@ -453,10 +450,8 @@ public class CassandraTemplate implements CassandraOperations, ApplicationEventP StatementBuilder