/** * @return The result size of the query. */ @Override public int getResultSize() { return hSearchQuery.queryResultSize(); }
/** * @return The result size of the query. */ @Override public int getResultSize() { partitionHandlingSupport.checkCacheAvailable(); return queryDefinition.getHsQuery().queryResultSize(); }
public int doGetResultSize() { if ( getLoader().isSizeSafe() ) { return hSearchQuery.queryResultSize(); } else { throw log.cannotGetResultSizeWithCriteriaAndRestriction( criteria.toString() ); } }
@Override public QueryResponse perform() { query.afterDeserialise(getSearchFactory()); DocumentExtractor extractor = query.queryDocumentExtractor(); try { int resultSize = query.queryResultSize(); QueryResponse queryResponse = new QueryResponse(resultSize); return queryResponse; } finally { extractor.close(); } }
private void assertDocumentIndexed(SearchIntegrator integrator, int expectedIndexSize) { int size = integrator .createHSQuery( new MatchAllDocsQuery(), Quote.class ) .queryResultSize(); String logMessage = "Index size=" + size + ", expected=" + expectedIndexSize; System.out.println( logMessage ); Assert.assertTrue( logMessage, size >= expectedIndexSize ); }
@Override QueryResponse perform() { HSQuery query = queryDefinition.getHsQuery(); query.afterDeserialise(getSearchFactory()); try (DocumentExtractor ignored = query.queryDocumentExtractor()) { int resultSize = query.queryResultSize(); return new QueryResponse(resultSize); } } }
public AbstractIntegerAssert<?> asResultSize() { HSQuery hsQuery = getHSQuery(); int actualSize = hsQuery.queryResultSize(); return Assertions.assertThat( actualSize ) .as( "Number of results of query " + toString( hsQuery ) ); }
/** * Run the actual query. Assert sizes to verify everything else is working. * @param expected number of elements found in the index */ private void assertElementsInIndex(int expected) { HSQuery hsQuery = searchIntegrator.createHSQuery( new MatchAllDocsQuery(), Book.class, Dvd.class ); int resultSize = hsQuery.queryResultSize(); Assert.assertEquals( expected, resultSize ); }
@Override QueryResponse perform() { HSQuery query = queryDefinition.getHsQuery(); query.afterDeserialise(getSearchFactory()); try (DocumentExtractor extractor = query.queryDocumentExtractor()) { int resultSize = query.queryResultSize(); return resultSize == 0 ? new QueryResponse(0) : new QueryResponse(collectKeys(extractor, query)); } }
private void assertIndexed(Class<?> entity, int count) { HSQuery hsQuery = sfHolder.getSearchFactory().createHSQuery( new MatchAllDocsQuery(), entity ); Assert.assertEquals( count, hsQuery.queryResultSize() ); }
private void assertIndexingFinished(SearchFactoryHolder searchFactoryHolder, int expectedDocsCount) { HSQuery query = searchFactoryHolder.getSearchFactory().createHSQuery( new MatchAllDocsQuery(), Quote.class ); Assert.assertEquals( expectedDocsCount, query.queryResultSize() ); }
@Benchmark @Threads(20) public void queryBooksByBestRating(QueryEngineHolder eh, Blackhole bh) { SearchIntegrator searchIntegrator = eh.si; Query luceneQuery = searchIntegrator.buildQueryBuilder() .forEntity( BookEntity.class ) .get() .all() .createQuery(); long expectedIndexSize = eh.getExpectedIndexSize(); int maxResults = eh.getMaxResults(); HSQuery hsQuery = searchIntegrator.createHSQuery( luceneQuery, BookEntity.class ); hsQuery.sort( new Sort( new SortField( "rating", SortField.Type.FLOAT, true ) ) ); hsQuery.maxResults( maxResults ); int queryResultSize = hsQuery.queryResultSize(); List<EntityInfo> queryEntityInfos = hsQuery.queryEntityInfos(); if ( eh.isQuerySync() && queryResultSize != expectedIndexSize ) { throw new RuntimeException( "Unexpected index size" ); } if ( maxResults != queryEntityInfos.size() ) { throw new RuntimeException( "Unexpected resultset size" ); } bh.consume( queryEntityInfos ); }
protected int queryIndex(Cache<?, ?> cache, String query) throws ParseException { QueryParser qp = createQueryParser("blurb"); Query q = qp.parse(query); HSQuery hsQuery = Search.getSearchManager(cache).unwrap(SearchIntegrator.class).createHSQuery(q, Person.class); return hsQuery.queryResultSize(); }
@Test @TestForIssue(jiraKey = "HSEARCH-2662") public void testQueryWhenEmpty() throws Exception { HSQuery query = getExtendedSearchIntegrator().createHSQuery( new MatchAllDocsQuery(), Animal.class ); assertEquals( 0, query.queryResultSize() ); SomeOtherEntity someOtherIndexedObject = new SomeOtherEntity(); insert( someOtherIndexedObject ); HSQuery queryAgain = getExtendedSearchIntegrator().createHSQuery( new MatchAllDocsQuery(), Animal.class ); assertEquals( 0, queryAgain.queryResultSize() ); }
@Override public QueryResponse perform() { query.afterDeserialise(getSearchFactory()); DocumentExtractor extractor = query.queryDocumentExtractor(); try { int resultSize = query.queryResultSize(); NodeTopDocs eagerTopDocs = resultSize == 0 ? null : collectKeys(extractor); QueryResponse queryResponse = new QueryResponse(eagerTopDocs, getQueryBox().getMyId(), resultSize); queryResponse.setAddress(cache.getAdvancedCache().getRpcManager().getAddress()); return queryResponse; } finally { extractor.close(); } }
@Override public QueryResponse perform() { query.afterDeserialise(getSearchFactory()); DocumentExtractor extractor = query.queryDocumentExtractor(); int resultSize = query.queryResultSize(); QueryBox box = getQueryBox(); // registering... box.put(lazyQueryId, extractor); // returning the QueryResponse TopDocs topDocs = extractor.getTopDocs(); QueryResponse queryResponse = new QueryResponse(new NodeTopDocs(topDocs), box.getMyId(), resultSize); queryResponse.setAddress(cache.getAdvancedCache().getRpcManager().getAddress()); return queryResponse; }
@Benchmark @GroupThreads(5) @Group("concurrentReadWriteTest") public void readWriteTestReader(QueryEngineHolder eh, Blackhole bh) { SearchIntegrator searchIntegrator = eh.si; Query luceneQuery = searchIntegrator.buildQueryBuilder() .forEntity( BookEntity.class ) .get() .all() .createQuery(); int maxResults = eh.getMaxResults(); HSQuery hsQuery = searchIntegrator.createHSQuery( luceneQuery, BookEntity.class ); hsQuery.maxResults( maxResults ); int queryResultSize = hsQuery.queryResultSize(); List<EntityInfo> queryEntityInfos = hsQuery.queryEntityInfos(); bh.consume( queryEntityInfos ); bh.consume( queryResultSize ); }
@Benchmark @GroupThreads(2 * AbstractBookEntity.TYPE_COUNT) @Group("concurrentReadWriteTest") public void readWriteTestReader(NonStreamWriteEngineHolder eh, NonStreamQueryParams qp, Blackhole bh) { SearchIntegrator searchIntegrator = eh.getSearchIntegrator(); Class<?> entityType = qp.getEntityType(); Query luceneQuery = searchIntegrator.buildQueryBuilder() .forEntity( entityType ) .get() .all() .createQuery(); int maxResults = qp.getQueryMaxResults(); HSQuery hsQuery = searchIntegrator.createHSQuery( luceneQuery, entityType ); hsQuery.maxResults( maxResults ); int queryResultSize = hsQuery.queryResultSize(); List<EntityInfo> queryEntityInfos = hsQuery.queryEntityInfos(); bh.consume( queryEntityInfos ); bh.consume( queryResultSize ); }
@Benchmark @Threads(6 * AbstractBookEntity.TYPE_COUNT) public void queryBooksByBestRating(NonStreamWriteEngineHolder eh, NonStreamQueryParams qp, Blackhole bh) { SearchIntegrator searchIntegrator = eh.getSearchIntegrator(); Class<?> entityType = qp.getEntityType(); Query luceneQuery = searchIntegrator.buildQueryBuilder() .forEntity( entityType ) .get() .all() .createQuery(); int maxResults = qp.getQueryMaxResults(); HSQuery hsQuery = searchIntegrator.createHSQuery( luceneQuery, entityType ); hsQuery.sort( new Sort( new SortField( "rating", SortField.Type.FLOAT, true ) ) ); hsQuery.maxResults( maxResults ); int queryResultSize = hsQuery.queryResultSize(); List<EntityInfo> queryEntityInfos = hsQuery.queryEntityInfos(); bh.consume( queryResultSize ); bh.consume( queryEntityInfos ); }
TShirt.class ); assertEquals( 0, hsQuery.queryResultSize() ); } );