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

DATAES-605 - Make batch size for streamQuery configurable.

Browse Source 
Original PR: #292
This commit is contained in:
rasmusfaber
2019-07-10 22:42:14 +02:00
committed by Peter-Josef Meisch
parent 7fd9986c6e
commit 216fc89fd2
2 changed files with 52 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/test/java/org/springframework/data/elasticsearch/repositories/custommethod/CustomMethodRepositoryBaseTests.java
View File

@@ -32,7 +32,6 @@ import java.util.List;
import java.util.stream.Stream;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.annotation.Id;
import org.springframework.data.annotation.Version;
@@ -65,6 +64,8 @@ public abstract class CustomMethodRepositoryBaseTests {
@Autowired private SampleCustomMethodRepository repository;
@Autowired private SampleStreamingCustomMethodRepository streamingRepository;
@Test
public void shouldExecuteCustomMethod() {
@@ -711,7 +712,7 @@ public abstract class CustomMethodRepositoryBaseTests {
repository.saveAll(entities);
// when
Stream<SampleEntity> stream = repository.findByType("abc");
Stream<SampleEntity> stream = streamingRepository.findByType("abc");
// then
assertThat(stream).isNotNull();
@@ -1236,13 +1237,41 @@ public abstract class CustomMethodRepositoryBaseTests {
assertThat(count).isEqualTo(1L);
}
@Test // DATAES-605
public void streamMethodsShouldWorkWithLargeResultSets() {
// given
List<SampleEntity> entities = createSampleEntities("abc", 10001);
repository.saveAll(entities);
// when
Stream<SampleEntity> stream = streamingRepository.findByType("abc");
// then
assertThat(stream).isNotNull();
assertThat(stream.count()).isEqualTo(10001L);
}
@Test // DATAES-605
public void streamMethodsCanHandlePageable() {
// given
List<SampleEntity> entities = createSampleEntities("abc", 10);
repository.saveAll(entities);
// when
Stream<SampleEntity> stream = streamingRepository.findByType("abc", PageRequest.of(0, 2));
// then
assertThat(stream).isNotNull();
assertThat(stream.count()).isEqualTo(10L);
}
private List<SampleEntity> createSampleEntities(String type, int numberOfEntities) {
List<SampleEntity> entities = new ArrayList<>();
for (int i = 0; i < numberOfEntities; i++) {
SampleEntity entity = new SampleEntity();
entity.setId(randomNumeric(numberOfEntities));
entity.setId(randomNumeric(32));
entity.setAvailable(true);
entity.setMessage("Message");
entity.setType(type);
@@ -1328,8 +1357,6 @@ public abstract class CustomMethodRepositoryBaseTests {
Page<SampleEntity> findByLocationNear(GeoPoint point, String distance, Pageable pageable);
Stream<SampleEntity> findByType(String type);
long countByType(String type);
long countByTypeNot(String type);
@@ -1371,4 +1398,12 @@ public abstract class CustomMethodRepositoryBaseTests {
long countByLocationNear(GeoPoint point, String distance);
}
/**
* @author Rasmus Faber-Espensen
*/
public interface SampleStreamingCustomMethodRepository extends ElasticsearchRepository<SampleEntity, String> {
Stream<SampleEntity> findByType(String type);
Stream<SampleEntity> findByType(String type, Pageable pageable);
}
}