使用聚合结果构造示例SearchResponse对象以进行测试

Question

我想创建一个带有聚合结果的searchResponse对象以进行测试。

我尝试了以下方法

SearchResponseSections section = new SearchResponseSections(null, null, .......);

这里的第一个参数用于我不想要的点击，但第二个参数接受聚合。

现在，我无法使用bucket创建自定义聚合对象。

测试代码如下：

public AutoCompleteResponseDto getAutoCompleteResponse(SearchResponse searchResponse, String field, int limit) {
  Terms aggregation = searchResponse.getAggregations().get("typeAhead");
  final Set<String> bucketKeys = aggregation.getBuckets().stream().map(Terms.Bucket::getKeyAsString).collect(Collectors.toSet());
  return AutoCompleteResponseDto.builder().totalCount(bucketKeys.size()).values(bucketKeys).field(field).build();
}

我还尝试使用Json创建SearchResponse，如下所示：

public static List<NamedXContentRegistry.Entry> getDefaultNamedXContents() {
  Map<String, ContextParser<Object, ? extends Aggregation>> map = new HashMap<>();
  map.put(StringTerms.NAME, (p, c) -> ParsedStringTerms.fromXContent(p, (String) c));
  return map.entrySet().stream().map(entry -> new NamedXContentRegistry.Entry(Aggregation.class, new ParseField(entry.getKey(), entry.getValue()))).collect(Collectors.toList());
}

public static SearchResponse getSearchResponse(String jsonString) {
  try {
    NamedXContentRegistry registry = new NamedXContentRegistry(getDefaultNamedXContents());
    XContentParser parser = new JsonXContentParser(registry, new JsonFactory().createParser(jsonString));
    return SearchResponse.fromXContent(parser);
  } catch (Exception ex) {
  }
  return new SearchResponse(); 
}

我使用以下json调用getSearchResponse（json）方法：

{
  "took": 65,
  "timed_out": false,
  "_shards": {
    "total": 5,
    "successful": 5,
    "skipped": 0,
    "failed": 0
  },
  "aggregations": {
    "typeAhead": {
      "buckets": [
        {
          "key": "1002312321",
          "docCount": 31521
        }
      ]
    }
  }
}

并获取ParseException：[无法解析键为[typeAhead]]的聚合。

ElasticSearch版本=5.6.15

注：如果有错别字，请忽略。

Answer 1

通过在json中将聚合键作为术语#typeAhead而不是仅使用typeAhead来解决问题。

聚合键应作为aggType#key给出，并且给定聚合类型的条目应在NamedXContent注册表中注册。