Java SolrIndexSearcher.QueryCommand方法代码示例

import org.apache.solr.search.SolrIndexSearcher; //导入方法依赖的package包/类
private CommandHandler(SolrIndexSearcher.QueryCommand queryCommand,
                       List<Command> commands,
                       SolrIndexSearcher searcher,
                       boolean needDocset,
                       boolean truncateGroups,
                       boolean includeHitCount) {
  this.queryCommand = queryCommand;
  this.commands = commands;
  this.searcher = searcher;
  this.needDocset = needDocset;
  this.truncateGroups = truncateGroups;
  this.includeHitCount = includeHitCount;
}
 

import org.apache.solr.search.SolrIndexSearcher; //导入方法依赖的package包/类
/**
 * Creates a SolrIndexSearcher.QueryCommand from this
 * ResponseBuilder.  TimeAllowed is left unset.
 */
public SolrIndexSearcher.QueryCommand getQueryCommand() {
  SolrIndexSearcher.QueryCommand cmd = new SolrIndexSearcher.QueryCommand();
  cmd.setQuery(wrap(getQuery()))
          .setFilterList(getFilters())
          .setSort(getSortSpec().getSort())
          .setOffset(getSortSpec().getOffset())
          .setLen(getSortSpec().getCount())
          .setFlags(getFieldFlags())
          .setNeedDocSet(isNeedDocSet())
          .setCursorMark(getCursorMark());
  return cmd;
}
 

import org.apache.solr.search.SolrIndexSearcher; //导入方法依赖的package包/类
public Builder setQueryCommand(SolrIndexSearcher.QueryCommand queryCommand) {
  this.queryCommand = queryCommand;
  this.needDocSet = (queryCommand.getFlags() & SolrIndexSearcher.GET_DOCSET) != 0;
  return this;
}
 

import org.apache.solr.search.SolrIndexSearcher; //导入方法依赖的package包/类
private void prepareGrouping(ResponseBuilder rb) throws IOException {

    SolrQueryRequest req = rb.req;
    SolrParams params = req.getParams();

    if (null != rb.getCursorMark()) {
      // It's hard to imagine, conceptually, what it would mean to combine
      // grouping with a cursor - so for now we just don't allow the combination at all
      throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "Can not use Grouping with " +
                              CursorMarkParams.CURSOR_MARK_PARAM);
    }

    SolrIndexSearcher.QueryCommand cmd = rb.getQueryCommand();
    SolrIndexSearcher searcher = rb.req.getSearcher();
    GroupingSpecification groupingSpec = new GroupingSpecification();
    rb.setGroupingSpec(groupingSpec);

    //TODO: move weighting of sort
    Sort groupSort = searcher.weightSort(cmd.getSort());
    if (groupSort == null) {
      groupSort = Sort.RELEVANCE;
    }

    // groupSort defaults to sort
    String groupSortStr = params.get(GroupParams.GROUP_SORT);
    //TODO: move weighting of sort
    Sort sortWithinGroup = groupSortStr == null ?  groupSort : searcher.weightSort(QueryParsing.parseSortSpec(groupSortStr, req).getSort());
    if (sortWithinGroup == null) {
      sortWithinGroup = Sort.RELEVANCE;
    }

    groupingSpec.setSortWithinGroup(sortWithinGroup);
    groupingSpec.setGroupSort(groupSort);

    String formatStr = params.get(GroupParams.GROUP_FORMAT, Grouping.Format.grouped.name());
    Grouping.Format responseFormat;
    try {
       responseFormat = Grouping.Format.valueOf(formatStr);
    } catch (IllegalArgumentException e) {
      throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, String.format(Locale.ROOT, "Illegal %s parameter", GroupParams.GROUP_FORMAT));
    }
    groupingSpec.setResponseFormat(responseFormat);

    groupingSpec.setFields(params.getParams(GroupParams.GROUP_FIELD));
    groupingSpec.setQueries(params.getParams(GroupParams.GROUP_QUERY));
    groupingSpec.setFunctions(params.getParams(GroupParams.GROUP_FUNC));
    groupingSpec.setGroupOffset(params.getInt(GroupParams.GROUP_OFFSET, 0));
    groupingSpec.setGroupLimit(params.getInt(GroupParams.GROUP_LIMIT, 1));
    groupingSpec.setOffset(rb.getSortSpec().getOffset());
    groupingSpec.setLimit(rb.getSortSpec().getCount());
    groupingSpec.setIncludeGroupCount(params.getBool(GroupParams.GROUP_TOTAL_COUNT, false));
    groupingSpec.setMain(params.getBool(GroupParams.GROUP_MAIN, false));
    groupingSpec.setNeedScore((cmd.getFlags() & SolrIndexSearcher.GET_SCORES) != 0);
    groupingSpec.setTruncateGroups(params.getBool(GroupParams.GROUP_TRUNCATE, false));
  }
 

import org.apache.solr.search.SolrIndexSearcher; //导入方法依赖的package包/类
@Test
public void testDocListConversion() throws Exception {
  assertU("", adoc("id", "3234", "val_i", "1", 
                   "val_dynamic", "quick red fox"));
  assertU("", adoc("id", "3235", "val_i", "1", 
                   "val_dynamic", "quick green fox"));
  assertU("", adoc("id", "3236", "val_i", "1", 
                   "val_dynamic", "quick brown fox"));
  assertU("", commit());

  RefCounted<SolrIndexSearcher> holder = h.getCore().getSearcher();
  try {
    SolrIndexSearcher srchr = holder.get();
    SolrIndexSearcher.QueryResult qr = new SolrIndexSearcher.QueryResult();
    SolrIndexSearcher.QueryCommand cmd = new SolrIndexSearcher.QueryCommand();
    cmd.setQuery(new MatchAllDocsQuery());
    cmd.setLen(10);
    qr = srchr.search(qr, cmd);
    
    DocList docs = qr.getDocList();
    assertEquals("wrong docs size", 3, docs.size());
    Set<String> fields = new HashSet<>();
    fields.add("val_dynamic");
    fields.add("dynamic_val");
    fields.add("range_facet_l"); // copied from id
    
    SolrDocumentList list = SolrPluginUtils.docListToSolrDocumentList(docs, srchr, fields, null);
    assertEquals("wrong list Size", docs.size(), list.size());
    for (SolrDocument document : list) {
      
      assertTrue("unexpected field", ! document.containsKey("val_i"));
      assertTrue("unexpected id field", ! document.containsKey("id"));

      assertTrue("original field", document.containsKey("val_dynamic"));
      assertTrue("dyn copy field", document.containsKey("dynamic_val"));
      assertTrue("copy field", document.containsKey("range_facet_l"));
      
      assertNotNull("original field null", document.get("val_dynamic"));
      assertNotNull("dyn copy field null", document.get("dynamic_val"));
      assertNotNull("copy field null", document.get("range_facet_l"));
    }
  } finally {
    if (null != holder) holder.decref();
  }
}