com.orientechnologies.lucene.tx.OLuceneTxChangesAbstract Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of orientdb-lucene Show documentation
Lucene plugin for OrientDB NoSQL document graph dbms
The newest version!
/*
 *
 *  * Copyright 2010-2016 OrientDB LTD (http://orientdb.com)
 *  *
 *  * Licensed under the Apache License, Version 2.0 (the "License");
 *  * you may not use this file except in compliance with the License.
 *  * You may obtain a copy of the License at
 *  *
 *  *      http://www.apache.org/licenses/LICENSE-2.0
 *  *
 *  * Unless required by applicable law or agreed to in writing, software
 *  * distributed under the License is distributed on an "AS IS" BASIS,
 *  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  * See the License for the specific language governing permissions and
 *  * limitations under the License.
 *
 */

package com.orientechnologies.lucene.tx;

import com.orientechnologies.common.exception.OException;
import com.orientechnologies.common.log.OLogManager;
import com.orientechnologies.lucene.engine.OLuceneIndexEngine;
import com.orientechnologies.lucene.exception.OLuceneIndexException;
import java.io.IOException;
import org.apache.lucene.index.DirectoryReader;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TopDocs;

/** Created by Enrico Risa on 28/09/15. */
public abstract class OLuceneTxChangesAbstract implements OLuceneTxChanges {
  public static final String TMP = "_tmp_rid";

  protected final OLuceneIndexEngine engine;
  protected final IndexWriter writer;
  protected final IndexWriter deletedIdx;

  public OLuceneTxChangesAbstract(
      final OLuceneIndexEngine engine, final IndexWriter writer, final IndexWriter deletedIdx) {
    this.engine = engine;
    this.writer = writer;
    this.deletedIdx = deletedIdx;
  }

  public IndexSearcher searcher() {
    // TODO optimize
    try {
      return new IndexSearcher(DirectoryReader.open(writer, true, true));
    } catch (IOException e) {
      //      OLogManager.instance().error(this, "Error during searcher index instantiation on new
      // documents", e);
      throw OException.wrapException(
          new OLuceneIndexException("Error during searcher index instantiation on new documents"),
          e);
    }
  }

  @Override
  public long deletedDocs(Query query) {
    try {
      final IndexSearcher indexSearcher =
          new IndexSearcher(DirectoryReader.open(deletedIdx, true, true));
      final TopDocs search = indexSearcher.search(query, Integer.MAX_VALUE);
      return search.totalHits;
    } catch (IOException e) {
      OLogManager.instance()
          .error(this, "Error during searcher index instantiation on deleted documents ", e);
    }
    return 0;
  }
}