List of usage examples for org.springframework.data.solr.core.query Query setPageRequest
<T extends Query> T setPageRequest(Pageable pageable);
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()); }