OrDocIdSet.java

package com.browseengine.bobo.docidset;

import java.io.IOException;
import java.io.Serializable;
import java.util.Comparator;
import java.util.List;

import org.apache.lucene.search.DocIdSet;
import org.apache.lucene.search.DocIdSetIterator;

public class OrDocIdSet extends ImmutableDocSet implements Serializable {
  private static final long serialVersionUID = 1L;

  private static final int INVALID = -1;

  List<DocIdSet> sets = null;

  private int _size = INVALID;

  public OrDocIdSet(List<DocIdSet> docSets) {
    this.sets = docSets;
  }

  @Override
  public DocIdSetIterator iterator() throws IOException {
    return new OrDocIdSetIterator(sets);
  }

  /**
   * Find existence in the set with index
   *
   * NOTE :  Expensive call. Avoid.
   * @param val value to find the index for
   * @return index where the value is
   */
  @Override
  public int findWithIndex(int val) throws IOException {
    DocIdSetIterator finder = new OrDocIdSetIterator(sets);
    int cursor = -1;
    try {
      int docid;
      while ((docid = finder.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS) {
        if (docid > val) return -1;
        else if (docid == val) return ++cursor;
        else ++cursor;

      }
    } catch (IOException e) {
      return -1;
    }
    return -1;
  }

  @Override
  public int size() throws IOException {

    if (_size == INVALID) {
      _size = 0;
      DocIdSetIterator it = this.iterator();

      try {
        while (it.nextDoc() != DocIdSetIterator.NO_MORE_DOCS)
          _size++;
      } catch (IOException e) {
        e.printStackTrace();
        _size = INVALID;
      }

    }
    return _size;
  }
}