Example usage for org.springframework.data.util CloseableIterator next

List of usage examples for org.springframework.data.util CloseableIterator next

Introduction

In this page you can find the example usage for org.springframework.data.util CloseableIterator next.

Prototype

E next();

Source Link

Document

Returns the next element in the iteration.

Usage

From source file:com.github.vanroy.springdata.jest.JestElasticsearchTemplateTests.java

@Test
public void shouldReturnResultsWithStreamForGivenCriteriaQuery() {
    //given// w  ww .j  a v  a  2 s  . co  m
    List<IndexQuery> entities = createSampleEntitiesWithMessage("Test message", 30);
    // when
    elasticsearchTemplate.bulkIndex(entities);
    elasticsearchTemplate.refresh(SampleEntity.class);
    // then

    CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria());
    criteriaQuery.addIndices(INDEX_NAME);
    criteriaQuery.addTypes(TYPE_NAME);
    criteriaQuery.setPageable(new PageRequest(0, 10));

    CloseableIterator<SampleEntity> stream = elasticsearchTemplate.stream(criteriaQuery, SampleEntity.class);
    List<SampleEntity> sampleEntities = new ArrayList<SampleEntity>();
    while (stream.hasNext()) {
        sampleEntities.add(stream.next());
    }
    assertThat(sampleEntities.size(), is(equalTo(30)));
}