com.orientechnologies.lucene.query.QueryContext 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
There is a newer version: 3.2.41
/*
 *
 *  * Copyright 2014 Orient Technologies.
 *  *
 *  * 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.query;

import com.orientechnologies.lucene.tx.OLuceneTxChanges;
import com.orientechnologies.orient.core.command.OCommandContext;
import org.apache.lucene.facet.FacetsConfig;
import org.apache.lucene.facet.taxonomy.TaxonomyReader;
import org.apache.lucene.index.MultiReader;
import org.apache.lucene.search.Filter;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.Sort;

import java.io.IOException;

/**
 * Created by Enrico Risa on 08/01/15.
 */
public class QueryContext {

  public final OCommandContext  context;
  protected final IndexSearcher searcher;
  public final Query            query;
  public final Filter           filter;
  public final Sort             sort;
  public QueryContextCFG        cfg;
  public boolean                facet     = false;
  public boolean                drillDown = false;
  public TaxonomyReader         reader;
  private FacetsConfig          facetConfig;
  private String                facetField;
  private String                drillDownQuery;
  protected OLuceneTxChanges    changes;

  public QueryContext(OCommandContext context, IndexSearcher searcher, Query query) {
    this(context, searcher, query, null, null);
  }

  public QueryContext(OCommandContext context, IndexSearcher searcher, Query query, Filter filter) {
    this(context, searcher, query, filter, null);
  }

  public QueryContext(OCommandContext context, IndexSearcher searcher, Query query, Filter filter, Sort sort) {
    this.context = context;
    this.searcher = searcher;
    this.query = query;
    this.filter = filter;
    this.sort = sort;
    initCFG();
  }

  private void initCFG() {
    if (filter != null && sort != null) {
      cfg = QueryContextCFG.FILTER_SORT;
    } else if (filter == null && sort == null) {
      cfg = QueryContextCFG.NO_FILTER_NO_SORT;
    } else if (filter != null) {
      cfg = QueryContextCFG.FILTER;
    } else {
      cfg = QueryContextCFG.SORT;
    }
  }

  public QueryContext setFacet(boolean facet) {
    this.facet = facet;
    return this;
  }

  public QueryContext setReader(TaxonomyReader reader) {
    this.reader = reader;
    return this;
  }

  public void setFacetConfig(FacetsConfig facetConfig) {
    this.facetConfig = facetConfig;
  }

  public FacetsConfig getFacetConfig() {
    return facetConfig;
  }

  public void setFacetField(String facetField) {
    this.facetField = facetField;
  }

  public String getFacetField() {
    return facetField;
  }

  public void setDrillDownQuery(String drillDownQuery) {
    this.drillDownQuery = drillDownQuery;
    drillDown = drillDownQuery != null;
  }

  public boolean isDrillDown() {
    return drillDown;
  }

  public String getDrillDownQuery() {
    return drillDownQuery;
  }

  public boolean isInTx() {
    return changes != null;
  }

  public enum QueryContextCFG {
    NO_FILTER_NO_SORT, FILTER_SORT, FILTER, SORT
  }

  public QueryContext setChanges(OLuceneTxChanges changes) {
    this.changes = changes;
    return this;
  }

  public OLuceneTxChanges changes() {
    return changes;
  }

  public IndexSearcher getSearcher() throws IOException {

    return changes == null ? searcher : new IndexSearcher(new MultiReader(searcher.getIndexReader(), changes.searcher()
        .getIndexReader()));
  }
}