SuggestionCollector.java

package com.tyndalehouse.step.core.service.helpers;

import org.apache.lucene.index.IndexReader;
import org.apache.lucene.search.Collector;
import org.apache.lucene.search.Scorer;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author chrisburrell
 */
public class SuggestionCollector extends Collector {
    private final List<Integer> docIds = new ArrayList<Integer>(32);

    @Override
    public void setScorer(final Scorer scorer) throws IOException {
        
    }

    @Override
    public void collect(final int doc) throws IOException {
        this.docIds.add(doc);
    }

    @Override
    public void setNextReader(final IndexReader reader, final int docBase) throws IOException {
        //not implemented
    }

    @Override
    public boolean acceptsDocsOutOfOrder() {
        return false;
    }
}