MtasDisabledTwoPhaseIteratorSpanQuery.java.html 7.7 KB
<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"><html xmlns="http://www.w3.org/1999/xhtml" lang="en"><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8"/><link rel="stylesheet" href="../jacoco-resources/report.css" type="text/css"/><link rel="shortcut icon" href="../jacoco-resources/report.gif" type="image/gif"/><title>MtasDisabledTwoPhaseIteratorSpanQuery.java</title><link rel="stylesheet" href="../jacoco-resources/prettify.css" type="text/css"/><script type="text/javascript" src="../jacoco-resources/prettify.js"></script></head><body onload="window['PR_TAB_WIDTH']=4;prettyPrint()"><div class="breadcrumb" id="breadcrumb"><span class="info"><a href="../jacoco-sessions.html" class="el_session">Sessions</a></span><a href="../index.html" class="el_report">MTAS</a> &gt; <a href="index.source.html" class="el_package">mtas.search.spans.util</a> &gt; <span class="el_source">MtasDisabledTwoPhaseIteratorSpanQuery.java</span></div><h1>MtasDisabledTwoPhaseIteratorSpanQuery.java</h1><pre class="source lang-java linenums">package mtas.search.spans.util;

import java.io.IOException;
import java.util.Map;
import java.util.Set;

import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.LeafReaderContext;
import org.apache.lucene.index.Term;
import org.apache.lucene.index.TermContext;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.spans.SpanWeight;
import mtas.search.spans.MtasSpanMatchNoneQuery;

/**
 * The Class MtasDisabledTwoPhaseIteratorSpanQuery.
 */
public class MtasDisabledTwoPhaseIteratorSpanQuery extends MtasSpanQuery {

  /** The sub query. */
  private MtasSpanQuery subQuery;

  /**
   * Instantiates a new mtas disabled two phase iterator span query.
   *
   * @param q the q
   */
  public MtasDisabledTwoPhaseIteratorSpanQuery(MtasSpanQuery q) {
<span class="fc" id="L29">    super(q.getMinimumWidth(), q.getMaximumWidth());</span>
<span class="fc" id="L30">    this.subQuery = q;</span>
<span class="fc" id="L31">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see
   * mtas.search.spans.util.MtasSpanQuery#createWeight(org.apache.lucene.search.
   * IndexSearcher, boolean)
   */
  @Override
  public MtasSpanWeight createWeight(IndexSearcher searcher,
      boolean needsScores, float boost) throws IOException {
<span class="fc" id="L43">    SpanWeight subWeight = subQuery.createWeight(searcher, needsScores, boost);</span>
<span class="fc" id="L44">    return new MtasDisabledTwoPhaseIteratorWeight(subWeight, searcher,</span>
        needsScores, boost);
  }

  /*
   * (non-Javadoc)
   * 
   * @see mtas.search.spans.util.MtasSpanQuery#rewrite(org.apache.lucene.index.
   * IndexReader)
   */
  @Override
  public MtasSpanQuery rewrite(IndexReader reader) throws IOException {
<span class="fc" id="L56">    MtasSpanQuery newQ = subQuery.rewrite(reader);</span>
<span class="pc bpc" id="L57" title="1 of 2 branches missed.">    if (newQ == null) {</span>
<span class="nc" id="L58">      newQ = new MtasSpanMatchNoneQuery(subQuery.getField());</span>
<span class="nc" id="L59">      return new MtasDisabledTwoPhaseIteratorSpanQuery(newQ);</span>
    } else {
<span class="fc" id="L61">      newQ.disableTwoPhaseIterator();</span>
<span class="fc bfc" id="L62" title="All 2 branches covered.">      if (!newQ.equals(subQuery)) {</span>
<span class="fc" id="L63">        return new MtasDisabledTwoPhaseIteratorSpanQuery(newQ).rewrite(reader);</span>
      } else {
<span class="fc" id="L65">        return super.rewrite(reader);</span>
      }
    }
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.spans.SpanQuery#getField()
   */
  public String getField() {
<span class="nc" id="L76">    return subQuery.getField();</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#toString(java.lang.String)
   */
  @Override
  public String toString(String field) {
<span class="nc" id="L86">    return subQuery.toString(field);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#equals(java.lang.Object)
   */
  @Override
  public boolean equals(Object obj) {
<span class="nc bnc" id="L96" title="All 2 branches missed.">    if (this == obj)</span>
<span class="nc" id="L97">      return true;</span>
<span class="nc bnc" id="L98" title="All 2 branches missed.">    if (obj == null)</span>
<span class="nc" id="L99">      return false;</span>
<span class="nc bnc" id="L100" title="All 2 branches missed.">    if (getClass() != obj.getClass())</span>
<span class="nc" id="L101">      return false;</span>
<span class="nc" id="L102">    final MtasDisabledTwoPhaseIteratorSpanQuery that = (MtasDisabledTwoPhaseIteratorSpanQuery) obj;</span>
<span class="nc" id="L103">    return that.subQuery.equals(subQuery);</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see org.apache.lucene.search.Query#hashCode()
   */
  @Override
  public int hashCode() {
<span class="nc" id="L113">    int h = Integer.rotateLeft(classHash(), 1);</span>
<span class="nc" id="L114">    h ^= subQuery.hashCode();</span>
<span class="nc" id="L115">    return h;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see mtas.search.spans.util.MtasSpanQuery#disableTwoPhaseIterator()
   */
  @Override
  public void disableTwoPhaseIterator() {
<span class="nc" id="L125">    super.disableTwoPhaseIterator();</span>
<span class="nc" id="L126">    subQuery.disableTwoPhaseIterator();</span>
<span class="nc" id="L127">  }</span>

  /**
   * The Class MtasDisabledTwoPhaseIteratorWeight.
   */
  private class MtasDisabledTwoPhaseIteratorWeight extends MtasSpanWeight {

    /** The sub weight. */
    SpanWeight subWeight;

    /**
     * Instantiates a new mtas disabled two phase iterator weight.
     *
     * @param subWeight the sub weight
     * @param searcher the searcher
     * @param needsScores the needs scores
     * @throws IOException Signals that an I/O exception has occurred.
     */
    public MtasDisabledTwoPhaseIteratorWeight(SpanWeight subWeight,
<span class="fc" id="L146">        IndexSearcher searcher, boolean needsScores, float boost) throws IOException {</span>
<span class="pc bpc" id="L147" title="1 of 2 branches missed.">      super(subQuery, searcher,</span>
<span class="pc" id="L148">          needsScores ? getTermContexts(subWeight) : null, boost);</span>
<span class="fc" id="L149">      this.subWeight = subWeight;</span>
<span class="fc" id="L150">    }</span>

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.apache.lucene.search.spans.SpanWeight#extractTermContexts(java.util.
     * Map)
     */
    @Override
    public void extractTermContexts(Map&lt;Term, TermContext&gt; contexts) {
<span class="nc" id="L161">      subWeight.extractTermContexts(contexts);</span>
<span class="nc" id="L162">    }</span>

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.apache.lucene.search.spans.SpanWeight#getSpans(org.apache.lucene.
     * index.LeafReaderContext,
     * org.apache.lucene.search.spans.SpanWeight.Postings)
     */
    @Override
    public MtasSpans getSpans(LeafReaderContext ctx, Postings requiredPostings)
        throws IOException {
<span class="fc" id="L175">      return new MtasDisabledTwoPhaseIteratorSpans(</span>
<span class="fc" id="L176">          subWeight.getSpans(ctx, requiredPostings));</span>
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.apache.lucene.search.Weight#extractTerms(java.util.Set)
     */
    @Override
    public void extractTerms(Set&lt;Term&gt; terms) {
<span class="nc" id="L186">      subWeight.extractTerms(terms);</span>
<span class="nc" id="L187">    }</span>

  }

}
</pre><div class="footer"><span class="right">Created with <a href="http://www.jacoco.org/jacoco">JaCoCo</a> 0.7.9.201702052155</span></div></body></html>