Example usage for org.springframework.data.solr.core.query Query setPageRequest

List of usage examples for org.springframework.data.solr.core.query Query setPageRequest

Introduction

In this page you can find the example usage for org.springframework.data.solr.core.query Query setPageRequest.

Prototype

<T extends Query> T setPageRequest(Pageable pageable);

Source Link

Document

restrict result to entries on given page.

Usage

From source file:com.att.archive.restful.repositories.solr.DocumentRepositoryImpl.java

private Page<T> performQuery(Query qCriteria, SearchQuery query) {
    Pageable page = (query.getPage() == null) ? new SolrPageRequest(0, 20) : query.getPage();
    qCriteria.setPageRequest(page);
    if (!query.getGroup().isEmpty())
        return this.getSolrOperations().queryForGroupPage(qCriteria, getEntityClass());
    else//from  w w  w  .j a v a2  s.c  o  m
        return this.getSolrOperations().queryForPage(qCriteria, getEntityClass());
}

From source file:org.springframework.data.solr.core.SolrTemplateTest.java

@Test
public void testCountWhenPagingSet() throws SolrServerException {
    ArgumentCaptor<SolrQuery> captor = ArgumentCaptor.forClass(SolrQuery.class);
    QueryResponse responseMock = Mockito.mock(QueryResponse.class);
    SolrDocumentList resultList = new SolrDocumentList();
    resultList.setNumFound(10);//  w w w.  jav a2 s.  com
    Mockito.when(responseMock.getResults()).thenReturn(resultList);
    Mockito.when(solrServerMock.query(Mockito.any(SolrQuery.class))).thenReturn(responseMock);

    Query query = new SimpleQuery(new Criteria("field_1").is("value1"));
    query.setPageRequest(new PageRequest(0, 5));
    long result = solrTemplate.count(query);
    Assert.assertEquals(resultList.getNumFound(), result);

    Mockito.verify(solrServerMock, Mockito.times(1)).query(captor.capture());

    Assert.assertEquals(Integer.valueOf(0), captor.getValue().getStart());
    Assert.assertEquals(Integer.valueOf(0), captor.getValue().getRows());
}