001/* 002 * #%L 003 * Netarchivesuite - common 004 * %% 005 * Copyright (C) 2005 - 2014 The Royal Danish Library, the Danish State and University Library, 006 * the National Library of France and the Austrian National Library. 007 * %% 008 * This program is free software: you can redistribute it and/or modify 009 * it under the terms of the GNU Lesser General Public License as 010 * published by the Free Software Foundation, either version 2.1 of the 011 * License, or (at your option) any later version. 012 * 013 * This program is distributed in the hope that it will be useful, 014 * but WITHOUT ANY WARRANTY; without even the implied warranty of 015 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 016 * GNU General Lesser Public License for more details. 017 * 018 * You should have received a copy of the GNU General Lesser Public 019 * License along with this program. If not, see 020 * <http://www.gnu.org/licenses/lgpl-2.1.html>. 021 * #L% 022 */ 023package dk.netarkivet.common.utils; 024 025import java.io.IOException; 026import java.util.ArrayList; 027import java.util.List; 028 029import org.apache.lucene.index.AtomicReaderContext; 030import org.apache.lucene.search.Collector; 031import org.apache.lucene.search.ScoreDoc; 032import org.apache.lucene.search.Scorer; 033 034/** Simple Collector to collect all results from Lucene query. */ 035public class AllDocsCollector extends Collector { 036 037 private List<ScoreDoc> docs = new ArrayList<ScoreDoc>(); 038 private Scorer scorer; 039 private int docBase; 040 041 @Override 042 public boolean acceptsDocsOutOfOrder() { 043 return true; 044 } 045 046 @Override 047 public void setScorer(Scorer scorer) { 048 this.scorer = scorer; 049 } 050 051 @Override 052 public void collect(int doc) throws IOException { 053 docs.add(new ScoreDoc(doc + docBase, scorer.score())); 054 } 055 056 public List<ScoreDoc> getHits() { 057 return docs; 058 } 059 060 public void reset() { 061 docs.clear(); 062 } 063 064 @Override 065 public void setNextReader(AtomicReaderContext arg0) throws IOException { 066 this.docBase = arg0.docBase; 067 068 } 069 070}