Examples with ChildrenAggregationBuilder org.opensearch.join.aggregations.ChildrenAggregationBuilder used on opensource projects

Example 1 with ChildrenAggregationBuilder

use of org.opensearch.join.aggregations.ChildrenAggregationBuilder in project OpenSearch by opensearch-project.

the class SearchIT method testSearchWithParentJoin.

public void testSearchWithParentJoin() throws IOException {
    final String indexName = "child_example";
    Request createIndex = new Request(HttpPut.METHOD_NAME, "/" + indexName);
    createIndex.setJsonEntity("{\n" + "    \"mappings\": {\n" + "        \"properties\" : {\n" + "            \"qa_join_field\" : {\n" + "                \"type\" : \"join\",\n" + "                \"relations\" : { \"question\" : \"answer\" }\n" + "            }\n" + "        }\n" + "    }" + "}");
    client().performRequest(createIndex);
    Request questionDoc = new Request(HttpPut.METHOD_NAME, "/" + indexName + "/_doc/1");
    questionDoc.setJsonEntity("{\n" + "    \"body\": \"<p>I have Windows 2003 server and i bought a new Windows 2008 server...\",\n" + "    \"title\": \"Whats the best way to file transfer my site from server to a newer one?\",\n" + "    \"tags\": [\n" + "        \"windows-server-2003\",\n" + "        \"windows-server-2008\",\n" + "        \"file-transfer\"\n" + "    ],\n" + "    \"qa_join_field\" : \"question\"\n" + "}");
    client().performRequest(questionDoc);
    Request answerDoc1 = new Request(HttpPut.METHOD_NAME, "/" + indexName + "/_doc/2");
    answerDoc1.addParameter("routing", "1");
    answerDoc1.setJsonEntity("{\n" + "    \"owner\": {\n" + "        \"location\": \"Norfolk, United Kingdom\",\n" + "        \"display_name\": \"Sam\",\n" + "        \"id\": 48\n" + "    },\n" + "    \"body\": \"<p>Unfortunately you're pretty much limited to FTP...\",\n" + "    \"qa_join_field\" : {\n" + "        \"name\" : \"answer\",\n" + "        \"parent\" : \"1\"\n" + "    },\n" + "    \"creation_date\": \"2009-05-04T13:45:37.030\"\n" + "}");
    client().performRequest(answerDoc1);
    Request answerDoc2 = new Request(HttpPut.METHOD_NAME, "/" + indexName + "/_doc/3");
    answerDoc2.addParameter("routing", "1");
    answerDoc2.setJsonEntity("{\n" + "    \"owner\": {\n" + "        \"location\": \"Norfolk, United Kingdom\",\n" + "        \"display_name\": \"Troll\",\n" + "        \"id\": 49\n" + "    },\n" + "    \"body\": \"<p>Use Linux...\",\n" + "    \"qa_join_field\" : {\n" + "        \"name\" : \"answer\",\n" + "        \"parent\" : \"1\"\n" + "    },\n" + "    \"creation_date\": \"2009-05-05T13:45:37.030\"\n" + "}");
    client().performRequest(answerDoc2);
    client().performRequest(new Request(HttpPost.METHOD_NAME, "/_refresh"));
    TermsAggregationBuilder leafTermAgg = new TermsAggregationBuilder("top-names").userValueTypeHint(ValueType.STRING).field("owner.display_name.keyword").size(10);
    ChildrenAggregationBuilder childrenAgg = new ChildrenAggregationBuilder("to-answers", "answer").subAggregation(leafTermAgg);
    TermsAggregationBuilder termsAgg = new TermsAggregationBuilder("top-tags").userValueTypeHint(ValueType.STRING).field("tags.keyword").size(10).subAggregation(childrenAgg);
    SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
    searchSourceBuilder.size(0).aggregation(termsAgg);
    SearchRequest searchRequest = new SearchRequest(indexName);
    searchRequest.source(searchSourceBuilder);
    SearchResponse searchResponse = execute(searchRequest, highLevelClient()::search, highLevelClient()::searchAsync);
    assertSearchHeader(searchResponse);
    assertNull(searchResponse.getSuggest());
    assertEquals(Collections.emptyMap(), searchResponse.getProfileResults());
    assertEquals(3, searchResponse.getHits().getTotalHits().value);
    assertEquals(0, searchResponse.getHits().getHits().length);
    assertEquals(Float.NaN, searchResponse.getHits().getMaxScore(), 0f);
    assertEquals(1, searchResponse.getAggregations().asList().size());
    Terms terms = searchResponse.getAggregations().get("top-tags");
    assertEquals(0, terms.getDocCountError());
    assertEquals(0, terms.getSumOfOtherDocCounts());
    assertEquals(3, terms.getBuckets().size());
    for (Terms.Bucket bucket : terms.getBuckets()) {
        assertThat(bucket.getKeyAsString(), either(equalTo("file-transfer")).or(equalTo("windows-server-2003")).or(equalTo("windows-server-2008")));
        assertEquals(1, bucket.getDocCount());
        assertEquals(1, bucket.getAggregations().asList().size());
        Children children = bucket.getAggregations().get("to-answers");
        assertEquals(2, children.getDocCount());
        assertEquals(1, children.getAggregations().asList().size());
        Terms leafTerms = children.getAggregations().get("top-names");
        assertEquals(0, leafTerms.getDocCountError());
        assertEquals(0, leafTerms.getSumOfOtherDocCounts());
        assertEquals(2, leafTerms.getBuckets().size());
        assertEquals(2, leafTerms.getBuckets().size());
        Terms.Bucket sam = leafTerms.getBucketByKey("Sam");
        assertEquals(1, sam.getDocCount());
        Terms.Bucket troll = leafTerms.getBucketByKey("Troll");
        assertEquals(1, troll.getDocCount());
    }
}