org.elasticsearch.search.aggregations.AbstractAggregationBuilder Java Examples

The following examples show how to use org.elasticsearch.search.aggregations.AbstractAggregationBuilder. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example #1

Source File: GroupConditionBuilder.java From search-spring-boot-starter with Apache License 2.0

7 votes

public AbstractAggregationBuilder getAggregationFunction(FunctionCondition functionCondition) {
    final String field = functionCondition.getField();
    final SqlFunctionEnum sqlFunction = functionCondition.getFunction();
    final String functionName = functionCondition.getFunctionName();
    AbstractAggregationBuilder aggregation;
    switch (sqlFunction) {
        case AVG:
            aggregation = AggregationBuilders.avg(functionName).field(field);
            break;
        case MAX:
            aggregation = AggregationBuilders.max(functionName).field(field);
            break;
        case MIN:
            aggregation = AggregationBuilders.min(functionName).field(field);
            break;
        case SUM:
            aggregation = AggregationBuilders.sum(functionName).field(field);
            break;
        case COUNT:
            aggregation = AggregationBuilders.count(functionName).field(field);
            break;
        default:
            throw new RuntimeException("不支持的聚合类型");
    }
    return aggregation;
}

Example #2

Source File: GroupConditionBuilder.java From search-spring-boot-starter with Apache License 2.0

6 votes

/**
 * 封装查询请求的聚合条件
 *
 * @param rootTermsBuilder     聚合根
 * @param termsBuilder         当前聚合
 * @param functions            聚合函数集
 * @param searchRequestBuilder 聚合功能查询请求
 */
private void aggregationFunction(AggregationBuilder rootTermsBuilder, AggregationBuilder termsBuilder,
                                 final List<FunctionCondition> functions,
                                 final SearchRequestBuilder searchRequestBuilder) {

    for (FunctionCondition functionCondition : functions) {

        AbstractAggregationBuilder aggregation = getAggregationFunction(functionCondition);

        if (termsBuilder != null) {
            termsBuilder.subAggregation(aggregation);
        } else {
            searchRequestBuilder.addAggregation(aggregation);
        }
    }
    if (termsBuilder != null) {
        searchRequestBuilder.addAggregation(rootTermsBuilder);
    }
}

Example #3

Source File: ProfileThreadSnapshotQueryEsDAO.java From skywalking with Apache License 2.0

6 votes

protected int querySequenceWithAgg(AbstractAggregationBuilder aggregationBuilder, String segmentId, long start, long end) throws IOException {
    SearchSourceBuilder sourceBuilder = SearchSourceBuilder.searchSource();

    BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
    sourceBuilder.query(boolQueryBuilder);
    List<QueryBuilder> mustQueryList = boolQueryBuilder.must();

    mustQueryList.add(QueryBuilders.termQuery(ProfileThreadSnapshotRecord.SEGMENT_ID, segmentId));
    mustQueryList.add(QueryBuilders.rangeQuery(ProfileThreadSnapshotRecord.DUMP_TIME).gte(start).lte(end));
    sourceBuilder.size(0);

    sourceBuilder.aggregation(aggregationBuilder);
    SearchResponse response = getClient().search(ProfileThreadSnapshotRecord.INDEX_NAME, sourceBuilder);
    NumericMetricsAggregation.SingleValue agg = response.getAggregations().get(ProfileThreadSnapshotRecord.SEQUENCE);

    return (int) agg.value();
}

Example #4

Source File: StatsTrendAction.java From foxtrot with Apache License 2.0

6 votes

@Override
public SearchRequestBuilder getRequestBuilder(StatsTrendRequest parameter) {
    SearchRequestBuilder searchRequestBuilder;
    try {
        final String table = parameter.getTable();
        AbstractAggregationBuilder aggregation = buildAggregation(parameter, table);
        searchRequestBuilder = getConnection().getClient()
                .prepareSearch(ElasticsearchUtils.getIndices(table, parameter))
                .setTypes(ElasticsearchUtils.DOCUMENT_TYPE_NAME)
                .setIndicesOptions(Utils.indicesOptions())
                .setQuery(new ElasticSearchQueryGenerator().genFilter(parameter.getFilters()))
                .setSize(0)
                .addAggregation(aggregation);
    }
    catch (Exception e) {
        throw FoxtrotExceptions.queryCreationException(parameter, e);
    }
    return searchRequestBuilder;
}

Example #5

Source File: GroupAction.java From foxtrot with Apache License 2.0

6 votes

@Override
public SearchRequestBuilder getRequestBuilder(GroupRequest parameter) {
    SearchRequestBuilder query;
    try {
        query = getConnection().getClient()
                .prepareSearch(ElasticsearchUtils.getIndices(parameter.getTable(), parameter))
                .setIndicesOptions(Utils.indicesOptions());
        AbstractAggregationBuilder aggregation = buildAggregation(parameter);
        query.setQuery(new ElasticSearchQueryGenerator().genFilter(parameter.getFilters()))
                .setSize(QUERY_SIZE)
                .addAggregation(aggregation);
    }
    catch (Exception e) {
        throw FoxtrotExceptions.queryCreationException(parameter, e);
    }
    return query;
}

Example #6

Source File: HistogramAction.java From foxtrot with Apache License 2.0

6 votes

@Override
public SearchRequestBuilder getRequestBuilder(HistogramRequest parameter) {
    SearchRequestBuilder searchRequestBuilder;
    AbstractAggregationBuilder aggregationBuilder = buildAggregation(parameter);
    try {
        searchRequestBuilder = getConnection().getClient()
                .prepareSearch(ElasticsearchUtils.getIndices(parameter.getTable(), parameter))
                .setTypes(ElasticsearchUtils.DOCUMENT_TYPE_NAME)
                .setIndicesOptions(Utils.indicesOptions())
                .setQuery(new ElasticSearchQueryGenerator().genFilter(parameter.getFilters()))
                .setSize(QUERY_SIZE)
                .addAggregation(aggregationBuilder);
    }
    catch (Exception e) {
        throw FoxtrotExceptions.queryCreationException(parameter, e);
    }
    return searchRequestBuilder;
}

Example #7

Source File: TrendAction.java From foxtrot with Apache License 2.0

6 votes

@Override
public SearchRequestBuilder getRequestBuilder(TrendRequest parameter) {
    SearchRequestBuilder searchRequestBuilder;
    try {
        AbstractAggregationBuilder aggregationBuilder = buildAggregation(parameter);
        searchRequestBuilder = getConnection().getClient()
                .prepareSearch(ElasticsearchUtils.getIndices(parameter.getTable(), parameter))
                .setIndicesOptions(Utils.indicesOptions())
                .setQuery(new ElasticSearchQueryGenerator().genFilter(parameter.getFilters()))
                .setSize(QUERY_SIZE)
                .addAggregation(aggregationBuilder);
    } catch (Exception e) {
        throw FoxtrotExceptions.queryCreationException(parameter, e);
    }
    return searchRequestBuilder;
}

Example #8

Source File: TrendAction.java From foxtrot with Apache License 2.0

5 votes

private AbstractAggregationBuilder buildAggregation(TrendRequest request) {
    DateHistogramInterval interval = Utils.getHistogramInterval(request.getPeriod());
    String field = request.getField();

    DateHistogramAggregationBuilder histogramBuilder = Utils.buildDateHistogramAggregation(request.getTimestamp(), interval);
    if(!CollectionUtils.isNullOrEmpty(getParameter().getUniqueCountOn())) {
        histogramBuilder.subAggregation(Utils.buildCardinalityAggregation(
                getParameter().getUniqueCountOn(), request.accept(new CountPrecisionThresholdVisitorAdapter(
                        elasticsearchTuningConfig.getPrecisionThreshold()))));
    }
    return Utils.buildTermsAggregation(Lists.newArrayList(new ResultSort(field, ResultSort.Order.asc)),
            Sets.newHashSet(histogramBuilder), elasticsearchTuningConfig.getAggregationSize());
}

Example #9

Source File: HistogramAction.java From foxtrot with Apache License 2.0

5 votes

private AbstractAggregationBuilder buildAggregation(HistogramRequest parameter) {
    DateHistogramInterval interval = Utils.getHistogramInterval(getParameter().getPeriod());
    DateHistogramAggregationBuilder histogramBuilder = Utils.buildDateHistogramAggregation(getParameter().getField(),
                                                                                           interval);
    if (!CollectionUtils.isNullOrEmpty(getParameter().getUniqueCountOn())) {
        histogramBuilder.subAggregation(Utils.buildCardinalityAggregation(
                getParameter().getUniqueCountOn(), parameter.accept(new CountPrecisionThresholdVisitorAdapter(
                        elasticsearchTuningConfig.getPrecisionThreshold()))));
    }
    return histogramBuilder;
}

Example #10

Source File: EsAbstractConditionAggregation.java From fess with Apache License 2.0

5 votes

protected void regA(AbstractAggregationBuilder<?> builder) {
    assertObjectNotNull("builder", builder);
    if (aggregationAggregationBuilderList == null) {
        aggregationAggregationBuilderList = new ArrayList<>();
    }
    aggregationAggregationBuilderList.add(builder);
}

Example #11

Source File: GroupAction.java From foxtrot with Apache License 2.0

5 votes

private AbstractAggregationBuilder buildAggregation(GroupRequest parameter) {
    return Utils.buildTermsAggregation(getParameter().getNesting()
                                               .stream()
                                               .map(x -> new ResultSort(x, ResultSort.Order.asc))
                                               .collect(Collectors.toList()),
                                       !CollectionUtils.isNullOrEmpty(getParameter().getUniqueCountOn())
                                       ? Sets.newHashSet(
                                               Utils.buildCardinalityAggregation(getParameter().getUniqueCountOn(),
                                                                                 parameter.accept(new CountPrecisionThresholdVisitorAdapter(
                                                                                         elasticsearchTuningConfig.getPrecisionThreshold()))))
                                       : Sets.newHashSet(), elasticsearchTuningConfig.getAggregationSize());

}

Example #12

Source File: EsAbstractConditionAggregation.java From fess with Apache License 2.0

5 votes

protected void regA(AbstractAggregationBuilder<?> builder) {
    assertObjectNotNull("builder", builder);
    if (aggregationAggregationBuilderList == null) {
        aggregationAggregationBuilderList = new ArrayList<>();
    }
    aggregationAggregationBuilderList.add(builder);
}

Example #13

Source File: StatsTrendAction.java From foxtrot with Apache License 2.0

5 votes

private AbstractAggregationBuilder buildAggregation(StatsTrendRequest request, String table) {
    final String field = request.getField();
    DateHistogramInterval interval = Utils.getHistogramInterval(request.getPeriod());
    AbstractAggregationBuilder dateHistogramBuilder = Utils.buildDateHistogramAggregation(request.getTimestamp(),
                                                                                          interval);
    boolean isNumericField = Utils.isNumericField(getTableMetadataManager(), table, field);
    if (isNumericField) {
        dateHistogramBuilder
                .subAggregation(Utils.buildStatsAggregation(field, getParameter().getStats()));
        if (!AnalyticsRequestFlags.hasFlag(request.getFlags(), AnalyticsRequestFlags.STATS_SKIP_PERCENTILES)) {
            dateHistogramBuilder.subAggregation(Utils.buildPercentileAggregation(
                    field, request.getPercentiles(), request.getCompression()));
        }
    }
    else {
        dateHistogramBuilder
                .subAggregation(Utils.buildStatsAggregation(field, Collections.singleton(Stat.COUNT)));
    }

    if (CollectionUtils.isNullOrEmpty(getParameter().getNesting())) {
        return dateHistogramBuilder;
    }
    return Utils.buildTermsAggregation(getParameter().getNesting()
                                               .stream()
                                               .map(x -> new ResultSort(x, ResultSort.Order.asc))
                                               .collect(Collectors.toList()), Sets.newHashSet(dateHistogramBuilder),
                                       elasticsearchTuningConfig.getAggregationSize());
}

Example #14

Source File: EsProductServiceImpl.java From BigDataPlatform with GNU General Public License v3.0

5 votes

@Override
public EsProductRelatedInfo searchRelatedInfo(String keyword) {
    NativeSearchQueryBuilder builder = new NativeSearchQueryBuilder();
    //搜索条件
    if(StringUtils.isEmpty(keyword)){
        builder.withQuery(QueryBuilders.matchAllQuery());
    }else{
        builder.withQuery(QueryBuilders.multiMatchQuery(keyword,"name","subTitle","keywords"));
    }
    //聚合搜索品牌名称
    builder.addAggregation(AggregationBuilders.terms("brandNames").field("brandName"));
    //集合搜索分类名称
    builder.addAggregation(AggregationBuilders.terms("productCategoryNames").field("productCategoryName"));
    //聚合搜索商品属性，去除type=1的属性
    AbstractAggregationBuilder aggregationBuilder = AggregationBuilders.nested("allAttrValues","attrValueList")
            .subAggregation(AggregationBuilders.filter("productAttrs",QueryBuilders.termQuery("attrValueList.type",1))
                    .subAggregation(AggregationBuilders.terms("attrIds")
                            .field("attrValueList.productAttributeId")
                            .subAggregation(AggregationBuilders.terms("attrValues")
                                    .field("attrValueList.value"))
                            .subAggregation(AggregationBuilders.terms("attrNames")
                                    .field("attrValueList.name"))));
    builder.addAggregation(aggregationBuilder);
    NativeSearchQuery searchQuery = builder.build();
    return elasticsearchTemplate.query(searchQuery, response -> {
        LOGGER.info("DSL:{}",searchQuery.getQuery().toString());
        return convertProductRelatedInfo(response);
    });
}

Example #15