protected int searchFlat( AbstractSearchRequest req, Collection<ArtifactInfo> result, IndexingContext context, Query query, int from, int aiCount) throws IOException { Hits hits = context .getIndexSearcher() .search(query, new Sort(new SortField(ArtifactInfo.UINFO, SortField.STRING))); if (hits == null || hits.length() == 0) { return 0; } if (req.isHitLimited() && hits.length() > req.getResultHitLimit()) { return AbstractSearchResponse.LIMIT_EXCEEDED; } int hitCount = hits.length(); int start = 0; // from == FlatSearchRequest.UNDEFINED ? 0 : from; // we have to pack the results as long: a) we have found aiCount ones b) we depleted hits for (int i = start; i < hits.length(); i++) { Document doc = hits.doc(i); ArtifactInfo artifactInfo = IndexUtils.constructArtifactInfo(doc, context); if (artifactInfo != null) { artifactInfo.repository = context.getRepositoryId(); artifactInfo.context = context.getId(); result.add(artifactInfo); if (req.isHitLimited() && result.size() > req.getResultHitLimit()) { // we hit limit, back out now !! return AbstractSearchResponse.LIMIT_EXCEEDED; } } } return hitCount; }
protected int searchGrouped( AbstractSearchRequest req, Map<String, ArtifactInfoGroup> result, Grouping grouping, IndexingContext context, Query query) throws IOException { Hits hits = context .getIndexSearcher() .search(query, new Sort(new SortField(ArtifactInfo.UINFO, SortField.STRING))); if (hits != null && hits.length() != 0) { int hitCount = hits.length(); for (int i = 0; i < hits.length(); i++) { ArtifactInfo artifactInfo = IndexUtils.constructArtifactInfo(hits.doc(i), context); if (artifactInfo != null) { artifactInfo.repository = context.getRepositoryId(); artifactInfo.context = context.getId(); if (!grouping.addArtifactInfo(result, artifactInfo)) { // fix the hitCount accordingly hitCount--; } } } if (req.isHitLimited() && hits.length() > req.getResultHitLimit()) { return AbstractSearchResponse.LIMIT_EXCEEDED; } return hitCount; } else { return 0; } }