MtasSolrComponentDocument.java.html 23.5 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>MtasSolrComponentDocument.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.solr.handler.component.util</a> &gt; <span class="el_source">MtasSolrComponentDocument.java</span></div><h1>MtasSolrComponentDocument.java</h1><pre class="source lang-java linenums">package mtas.solr.handler.component.util;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Set;
import java.util.SortedSet;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.solr.common.util.NamedList;
import org.apache.solr.common.util.SimpleOrderedMap;
import org.apache.solr.handler.component.ResponseBuilder;
import org.apache.solr.handler.component.SearchComponent;
import org.apache.solr.handler.component.ShardRequest;

import mtas.codec.util.CodecComponent.ComponentDocument;
import mtas.codec.util.CodecComponent.ComponentField;
import mtas.codec.util.CodecComponent.ComponentFields;
import mtas.codec.util.collector.MtasDataCollector;
import mtas.solr.handler.component.MtasSolrSearchComponent;

/**
 * The Class MtasSolrComponentDocument.
 */
public class MtasSolrComponentDocument
    implements MtasSolrComponent&lt;ComponentDocument&gt; {

  /** The Constant log. */
<span class="fc" id="L29">  private static final Log log = LogFactory</span>
<span class="fc" id="L30">      .getLog(MtasSolrComponentDocument.class);</span>

  /** The Constant PARAM_MTAS_DOCUMENT. */
  public static final String PARAM_MTAS_DOCUMENT = MtasSolrSearchComponent.PARAM_MTAS
      + &quot;.document&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_FIELD. */
  public static final String NAME_MTAS_DOCUMENT_FIELD = &quot;field&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_KEY. */
  public static final String NAME_MTAS_DOCUMENT_KEY = &quot;key&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_PREFIX. */
  public static final String NAME_MTAS_DOCUMENT_PREFIX = &quot;prefix&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_TYPE. */
  public static final String NAME_MTAS_DOCUMENT_TYPE = &quot;type&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_REGEXP. */
  public static final String NAME_MTAS_DOCUMENT_REGEXP = &quot;regexp&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_LIST. */
  public static final String NAME_MTAS_DOCUMENT_LIST = &quot;list&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_LIST_REGEXP. */
  public static final String NAME_MTAS_DOCUMENT_LIST_REGEXP = &quot;listRegexp&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_LIST_EXPAND. */
  public static final String NAME_MTAS_DOCUMENT_LIST_EXPAND = &quot;listExpand&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_LIST_EXPAND_NUMBER. */
  public static final String NAME_MTAS_DOCUMENT_LIST_EXPAND_NUMBER = &quot;listExpandNumber&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_IGNORE_REGEXP. */
  public static final String NAME_MTAS_DOCUMENT_IGNORE_REGEXP = &quot;ignoreRegexp&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_IGNORE_LIST. */
  public static final String NAME_MTAS_DOCUMENT_IGNORE_LIST = &quot;ignoreList&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_IGNORE_LIST_REGEXP. */
  public static final String NAME_MTAS_DOCUMENT_IGNORE_LIST_REGEXP = &quot;ignoreListRegexp&quot;;

  /** The Constant NAME_MTAS_DOCUMENT_NUMBER. */
<span class="fc" id="L73">  public static final String NAME_MTAS_DOCUMENT_NUMBER = &quot;number&quot;;</span>

  /** The search component. */
  private MtasSolrSearchComponent searchComponent;

  /**
   * Instantiates a new mtas solr component document.
   *
   * @param searchComponent the search component
   */
<span class="fc" id="L83">  public MtasSolrComponentDocument(MtasSolrSearchComponent searchComponent) {</span>
<span class="fc" id="L84">    this.searchComponent = searchComponent;</span>
<span class="fc" id="L85">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see
   * mtas.solr.handler.component.util.MtasSolrComponent#prepare(org.apache.solr.
   * handler.component.ResponseBuilder,
   * mtas.codec.util.CodecComponent.ComponentFields)
   */
  public void prepare(ResponseBuilder rb, ComponentFields mtasFields)
      throws IOException {
<span class="nc" id="L97">    Set&lt;String&gt; ids = MtasSolrResultUtil</span>
<span class="nc" id="L98">        .getIdsFromParameters(rb.req.getParams(), PARAM_MTAS_DOCUMENT);</span>
<span class="nc bnc" id="L99" title="All 2 branches missed.">    if (!ids.isEmpty()) {</span>
<span class="nc" id="L100">      int tmpCounter = 0;</span>
<span class="nc" id="L101">      String[] fields = new String[ids.size()];</span>
<span class="nc" id="L102">      String[] keys = new String[ids.size()];</span>
<span class="nc" id="L103">      String[] prefixes = new String[ids.size()];</span>
<span class="nc" id="L104">      String[] types = new String[ids.size()];</span>
<span class="nc" id="L105">      String[] regexps = new String[ids.size()];</span>
<span class="nc" id="L106">      String[] lists = new String[ids.size()];</span>
<span class="nc" id="L107">      Boolean[] listRegexps = new Boolean[ids.size()];</span>
<span class="nc" id="L108">      Boolean[] listExpands = new Boolean[ids.size()];</span>
<span class="nc" id="L109">      int[] listExpandNumbers = new int[ids.size()];</span>
<span class="nc" id="L110">      String[] ignoreRegexps = new String[ids.size()];</span>
<span class="nc" id="L111">      String[] ignoreLists = new String[ids.size()];</span>
<span class="nc" id="L112">      Boolean[] ignoreListRegexps = new Boolean[ids.size()];</span>
<span class="nc" id="L113">      String[] listNumbers = new String[ids.size()];</span>
<span class="nc bnc" id="L114" title="All 2 branches missed.">      for (String id : ids) {</span>
<span class="nc" id="L115">        fields[tmpCounter] = rb.req.getParams().get(</span>
<span class="nc" id="L116">            PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_FIELD,</span>
<span class="nc" id="L117">            null);</span>
<span class="nc" id="L118">        keys[tmpCounter] = rb.req.getParams()</span>
<span class="nc" id="L119">            .get(PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_KEY,</span>
<span class="nc" id="L120">                String.valueOf(tmpCounter))</span>
<span class="nc" id="L121">            .trim();</span>
<span class="nc" id="L122">        prefixes[tmpCounter] = rb.req.getParams().get(</span>
<span class="nc" id="L123">            PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_PREFIX,</span>
<span class="nc" id="L124">            null);</span>
<span class="nc" id="L125">        types[tmpCounter] = rb.req.getParams().get(</span>
<span class="nc" id="L126">            PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_TYPE,</span>
<span class="nc" id="L127">            null);</span>
<span class="nc" id="L128">        regexps[tmpCounter] = rb.req.getParams().get(</span>
<span class="nc" id="L129">            PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_REGEXP,</span>
<span class="nc" id="L130">            null);</span>
<span class="nc" id="L131">        lists[tmpCounter] = rb.req.getParams().get(</span>
<span class="nc" id="L132">            PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_LIST,</span>
<span class="nc" id="L133">            null);</span>
<span class="nc" id="L134">        listRegexps[tmpCounter] = rb.req.getParams().getBool(PARAM_MTAS_DOCUMENT</span>
<span class="nc" id="L135">            + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_LIST_REGEXP, false);</span>
<span class="nc" id="L136">        listExpands[tmpCounter] = rb.req.getParams().getBool(PARAM_MTAS_DOCUMENT</span>
<span class="nc" id="L137">            + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_LIST_EXPAND, false);</span>
<span class="nc" id="L138">        listExpandNumbers[tmpCounter] = rb.req.getParams()</span>
<span class="nc" id="L139">            .getInt(PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot;</span>
<span class="nc" id="L140">                + NAME_MTAS_DOCUMENT_LIST_EXPAND_NUMBER, 10);</span>
<span class="nc" id="L141">        ignoreRegexps[tmpCounter] = rb.req.getParams().get(PARAM_MTAS_DOCUMENT</span>
<span class="nc" id="L142">            + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_IGNORE_REGEXP, null);</span>
<span class="nc" id="L143">        ignoreLists[tmpCounter] = rb.req.getParams().get(PARAM_MTAS_DOCUMENT</span>
<span class="nc" id="L144">            + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_IGNORE_LIST, null);</span>
<span class="nc" id="L145">        ignoreListRegexps[tmpCounter] = rb.req.getParams()</span>
<span class="nc" id="L146">            .getBool(PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot;</span>
<span class="nc" id="L147">                + NAME_MTAS_DOCUMENT_IGNORE_LIST_REGEXP, false);</span>
<span class="nc" id="L148">        listNumbers[tmpCounter] = rb.req.getParams().get(</span>
<span class="nc" id="L149">            PARAM_MTAS_DOCUMENT + &quot;.&quot; + id + &quot;.&quot; + NAME_MTAS_DOCUMENT_NUMBER,</span>
<span class="nc" id="L150">            null);</span>
<span class="nc" id="L151">        tmpCounter++;</span>
      }
<span class="nc" id="L153">      String uniqueKeyField = rb.req.getSchema().getUniqueKeyField().getName();</span>
<span class="nc" id="L154">      mtasFields.doDocument = true;</span>
<span class="nc" id="L155">      rb.setNeedDocList(true);</span>
<span class="nc bnc" id="L156" title="All 2 branches missed.">      for (String field : fields) {</span>
<span class="nc bnc" id="L157" title="All 4 branches missed.">        if (field == null || field.isEmpty()) {</span>
<span class="nc" id="L158">          throw new IOException(&quot;no (valid) field in mtas document&quot;);</span>
<span class="nc bnc" id="L159" title="All 2 branches missed.">        } else if (!mtasFields.list.containsKey(field)) {</span>
<span class="nc" id="L160">          mtasFields.list.put(field, new ComponentField(uniqueKeyField));</span>
        }
      }
<span class="nc" id="L163">      MtasSolrResultUtil.compareAndCheck(keys, fields, NAME_MTAS_DOCUMENT_KEY,</span>
<span class="nc" id="L164">          NAME_MTAS_DOCUMENT_FIELD, true);</span>
<span class="nc" id="L165">      MtasSolrResultUtil.compareAndCheck(prefixes, fields,</span>
<span class="nc" id="L166">          NAME_MTAS_DOCUMENT_PREFIX, NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc" id="L167">      MtasSolrResultUtil.compareAndCheck(types, fields, NAME_MTAS_DOCUMENT_TYPE,</span>
<span class="nc" id="L168">          NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc" id="L169">      MtasSolrResultUtil.compareAndCheck(regexps, fields,</span>
<span class="nc" id="L170">          NAME_MTAS_DOCUMENT_REGEXP, NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc" id="L171">      MtasSolrResultUtil.compareAndCheck(lists, fields, NAME_MTAS_DOCUMENT_LIST,</span>
<span class="nc" id="L172">          NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc" id="L173">      MtasSolrResultUtil.compareAndCheck(ignoreRegexps, fields,</span>
<span class="nc" id="L174">          NAME_MTAS_DOCUMENT_IGNORE_REGEXP, NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc" id="L175">      MtasSolrResultUtil.compareAndCheck(ignoreLists, fields,</span>
<span class="nc" id="L176">          NAME_MTAS_DOCUMENT_IGNORE_LIST, NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc" id="L177">      MtasSolrResultUtil.compareAndCheck(listNumbers, fields,</span>
<span class="nc" id="L178">          NAME_MTAS_DOCUMENT_NUMBER, NAME_MTAS_DOCUMENT_FIELD, false);</span>
<span class="nc bnc" id="L179" title="All 2 branches missed.">      for (int i = 0; i &lt; fields.length; i++) {</span>
<span class="nc bnc" id="L180" title="All 4 branches missed.">        String key = (keys[i] == null) || (keys[i].isEmpty())</span>
<span class="nc" id="L181">            ? String.valueOf(i) + &quot;:&quot; + fields[i] + &quot;:&quot; + prefixes[i]</span>
<span class="nc" id="L182">            : keys[i].trim();</span>
<span class="nc" id="L183">        String prefix = prefixes[i];</span>
<span class="nc bnc" id="L184" title="All 4 branches missed.">        if (prefix == null || prefix.isEmpty()) {</span>
<span class="nc" id="L185">          throw new IOException(&quot;no (valid) prefix in mtas document&quot;);</span>
        }
<span class="nc" id="L187">        String type = types[i];</span>
<span class="nc" id="L188">        String regexp = regexps[i];</span>
<span class="nc" id="L189">        String[] list = null;</span>
<span class="nc" id="L190">        Boolean listRegexp = listRegexps[i];</span>
<span class="nc" id="L191">        Boolean listExpand = listExpands[i];</span>
<span class="nc" id="L192">        int listExpandNumber = listExpandNumbers[i];</span>
<span class="nc bnc" id="L193" title="All 2 branches missed.">        if (lists[i] != null) {</span>
<span class="nc" id="L194">          ArrayList&lt;String&gt; tmpList = new ArrayList&lt;&gt;();</span>
<span class="nc" id="L195">          String[] subList = lists[i].split(&quot;(?&lt;!\\\\),&quot;);</span>
<span class="nc bnc" id="L196" title="All 2 branches missed.">          for (int j = 0; j &lt; subList.length; j++) {</span>
<span class="nc" id="L197">            tmpList.add(subList[j].replace(&quot;\\,&quot;, &quot;,&quot;).replace(&quot;\\\\&quot;, &quot;\\&quot;));</span>
          }
<span class="nc" id="L199">          list = tmpList.toArray(new String[tmpList.size()]);</span>
        }
<span class="nc" id="L201">        int listNumber = Math.max(0,</span>
<span class="nc bnc" id="L202" title="All 4 branches missed.">            (listNumbers[i] == null) || (listNumbers[i].isEmpty()) ? 0</span>
<span class="nc" id="L203">                : Integer.parseInt(listNumbers[i]));</span>
<span class="nc" id="L204">        String ignoreRegexp = ignoreRegexps[i];</span>
<span class="nc" id="L205">        String[] ignoreList = null;</span>
<span class="nc" id="L206">        Boolean ignoreListRegexp = ignoreListRegexps[i];</span>
<span class="nc bnc" id="L207" title="All 2 branches missed.">        if (ignoreLists[i] != null) {</span>
<span class="nc" id="L208">          ArrayList&lt;String&gt; tmpList = new ArrayList&lt;&gt;();</span>
<span class="nc" id="L209">          String[] subList = ignoreLists[i].split(&quot;(?&lt;!\\\\),&quot;);</span>
<span class="nc bnc" id="L210" title="All 2 branches missed.">          for (int j = 0; j &lt; subList.length; j++) {</span>
<span class="nc" id="L211">            tmpList.add(subList[j].replace(&quot;\\,&quot;, &quot;,&quot;).replace(&quot;\\\\&quot;, &quot;\\&quot;));</span>
          }
<span class="nc" id="L213">          ignoreList = tmpList.toArray(new String[tmpList.size()]);</span>
        }
<span class="nc" id="L215">        mtasFields.list.get(fields[i]).documentList.add(new ComponentDocument(</span>
<span class="nc" id="L216">            key, prefix, type, regexp, list, listNumber, listRegexp, listExpand,</span>
<span class="nc" id="L217">            listExpandNumber, ignoreRegexp, ignoreList, ignoreListRegexp));</span>
      }
    }
<span class="nc" id="L220">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see
   * mtas.solr.handler.component.util.MtasSolrComponent#create(mtas.codec.util.
   * CodecComponent.BasicComponent, java.lang.Boolean)
   */
  public SimpleOrderedMap&lt;Object&gt; create(ComponentDocument document,
      Boolean encode) throws IOException {
<span class="nc" id="L231">    SimpleOrderedMap&lt;Object&gt; mtasDocumentResponse = new SimpleOrderedMap&lt;&gt;();</span>
<span class="nc" id="L232">    mtasDocumentResponse.add(&quot;key&quot;, document.key);</span>
<span class="nc" id="L233">    ArrayList&lt;NamedList&lt;Object&gt;&gt; mtasDocumentItemResponses = new ArrayList&lt;&gt;();</span>
<span class="nc bnc" id="L234" title="All 2 branches missed.">    for (int docId : document.statsData.keySet()) {</span>
<span class="nc" id="L235">      NamedList&lt;Object&gt; mtasDocumentItemResponse = new SimpleOrderedMap&lt;&gt;();</span>
<span class="nc" id="L236">      MtasDataCollector&lt;?, ?&gt; stats = document.statsData.get(docId);</span>
<span class="nc" id="L237">      MtasDataCollector&lt;?, ?&gt; list = null;</span>
<span class="nc bnc" id="L238" title="All 2 branches missed.">      if (document.statsList != null) {</span>
<span class="nc" id="L239">        list = document.statsList.get(docId);</span>
      }
<span class="nc" id="L241">      mtasDocumentItemResponse.add(&quot;stats&quot;,</span>
<span class="nc" id="L242">          new MtasSolrMtasResult(stats, stats.getDataType(),</span>
<span class="nc" id="L243">              stats.getStatsType(), stats.getStatsItems(), null));</span>
<span class="nc" id="L244">      mtasDocumentItemResponse.add(&quot;documentKey&quot;,</span>
<span class="nc" id="L245">          document.uniqueKey.get(docId));</span>
<span class="nc bnc" id="L246" title="All 2 branches missed.">      if (list != null) {</span>
<span class="nc bnc" id="L247" title="All 2 branches missed.">        if (document.listExpand) {</span>
<span class="nc" id="L248">          mtasDocumentItemResponse.add(&quot;list&quot;, new MtasSolrMtasResult(list,</span>
<span class="nc" id="L249">              new String[] { list.getDataType(), list.getDataType() },</span>
<span class="nc" id="L250">              new String[] { list.getStatsType(), list.getStatsType() },</span>
<span class="nc" id="L251">              new SortedSet[] { list.getStatsItems(), list.getStatsItems() },</span>
<span class="nc" id="L252">              new String[] { null, null }, new String[] { null, null },</span>
<span class="nc" id="L253">              new Integer[] { 0, 0 }, new Integer[] { 1, 1 }, null));</span>
<span class="nc" id="L254">        } else {</span>
<span class="nc" id="L255">          mtasDocumentItemResponse.add(&quot;list&quot;,</span>
<span class="nc" id="L256">              new MtasSolrMtasResult(list, list.getDataType(),</span>
<span class="nc" id="L257">                  list.getStatsType(), list.getStatsItems(), null));</span>
        }

      }
      // add
<span class="nc" id="L262">      mtasDocumentItemResponses.add(mtasDocumentItemResponse);</span>
    }
<span class="nc" id="L264">    mtasDocumentResponse.add(&quot;list&quot;, mtasDocumentItemResponses);</span>
<span class="nc" id="L265">    MtasSolrResultUtil.rewrite(mtasDocumentResponse, searchComponent);</span>
<span class="nc" id="L266">    return mtasDocumentResponse;</span>
  }

  /*
   * (non-Javadoc)
   * 
   * @see
   * mtas.solr.handler.component.util.MtasSolrComponent#modifyRequest(org.apache
   * .solr.handler.component.ResponseBuilder,
   * org.apache.solr.handler.component.SearchComponent,
   * org.apache.solr.handler.component.ShardRequest)
   */
  public void modifyRequest(ResponseBuilder rb, SearchComponent who,
      ShardRequest sreq) {
<span class="nc bnc" id="L280" title="All 2 branches missed.">    if (sreq.params.getBool(MtasSolrSearchComponent.PARAM_MTAS, false)</span>
<span class="nc bnc" id="L281" title="All 2 branches missed.">        &amp;&amp; sreq.params.getBool(PARAM_MTAS_DOCUMENT, false)) {</span>
<span class="nc bnc" id="L282" title="All 2 branches missed.">      if ((sreq.purpose &amp; ShardRequest.PURPOSE_GET_FIELDS) != 0) {</span>
        // do nothing
      } else {
<span class="nc" id="L285">        Set&lt;String&gt; keys = MtasSolrResultUtil</span>
<span class="nc" id="L286">            .getIdsFromParameters(rb.req.getParams(), PARAM_MTAS_DOCUMENT);</span>
<span class="nc" id="L287">        sreq.params.remove(PARAM_MTAS_DOCUMENT);</span>
<span class="nc bnc" id="L288" title="All 2 branches missed.">        for (String key : keys) {</span>
<span class="nc" id="L289">          sreq.params.remove(</span>
<span class="nc" id="L290">              PARAM_MTAS_DOCUMENT + &quot;.&quot; + key + &quot;.&quot; + NAME_MTAS_DOCUMENT_FIELD);</span>
<span class="nc" id="L291">          sreq.params.remove(</span>
<span class="nc" id="L292">              PARAM_MTAS_DOCUMENT + &quot;.&quot; + key + &quot;.&quot; + NAME_MTAS_DOCUMENT_KEY);</span>
<span class="nc" id="L293">          sreq.params.remove(PARAM_MTAS_DOCUMENT + &quot;.&quot; + key + &quot;.&quot;</span>
<span class="nc" id="L294">              + NAME_MTAS_DOCUMENT_PREFIX);</span>
        }
      }
    }

<span class="nc" id="L299">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see
   * mtas.solr.handler.component.util.MtasSolrComponent#finishStage(org.apache.
   * solr.handler.component.ResponseBuilder)
   */
  public void finishStage(ResponseBuilder rb) {
<span class="nc bnc" id="L309" title="All 2 branches missed.">    if (rb.req.getParams().getBool(MtasSolrSearchComponent.PARAM_MTAS, false)</span>
<span class="nc bnc" id="L310" title="All 2 branches missed.">        &amp;&amp; rb.stage &gt;= ResponseBuilder.STAGE_EXECUTE_QUERY</span>
<span class="nc bnc" id="L311" title="All 2 branches missed.">        &amp;&amp; rb.stage &lt; ResponseBuilder.STAGE_GET_FIELDS) {</span>
<span class="nc bnc" id="L312" title="All 2 branches missed.">      for (ShardRequest sreq : rb.finished) {</span>
<span class="nc bnc" id="L313" title="All 2 branches missed.">        if (sreq.params.getBool(MtasSolrSearchComponent.PARAM_MTAS, false)</span>
<span class="nc" id="L314">            &amp;&amp; sreq.params.getBool(PARAM_MTAS_DOCUMENT, false)) {</span>
          // nothing to do
        }
      }
    }
<span class="nc" id="L319">  }</span>

  /*
   * (non-Javadoc)
   * 
   * @see
   * mtas.solr.handler.component.util.MtasSolrComponent#distributedProcess(org.
   * apache.solr.handler.component.ResponseBuilder,
   * mtas.codec.util.CodecComponent.ComponentFields)
   */
  public void distributedProcess(ResponseBuilder rb, ComponentFields mtasFields)
      throws IOException {
    // rewrite
<span class="nc" id="L332">    NamedList&lt;Object&gt; mtasResponse = null;</span>
    try {
<span class="nc" id="L334">      mtasResponse = (NamedList&lt;Object&gt;) rb.rsp.getValues().get(&quot;mtas&quot;);</span>
<span class="nc" id="L335">    } catch (ClassCastException e) {</span>
<span class="nc" id="L336">      log.debug(e);</span>
<span class="nc" id="L337">      mtasResponse = null;</span>
    }

<span class="nc bnc" id="L340" title="All 2 branches missed.">    if (mtasResponse != null) {</span>
      ArrayList&lt;Object&gt; mtasResponseDocument;
      try {
<span class="nc" id="L343">        mtasResponseDocument = (ArrayList&lt;Object&gt;) mtasResponse.get(&quot;document&quot;);</span>
<span class="nc bnc" id="L344" title="All 2 branches missed.">        if (mtasResponseDocument != null) {</span>
<span class="nc" id="L345">          MtasSolrResultUtil.rewrite(mtasResponseDocument, searchComponent);</span>
        }
<span class="nc" id="L347">      } catch (ClassCastException e) {</span>
<span class="nc" id="L348">        log.debug(e);</span>
      }
    }

<span class="nc" id="L352">  }</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>