Example usage for org.apache.solr.common.params StatsParams STATS_CALC_DISTINCT

List of usage examples for org.apache.solr.common.params StatsParams STATS_CALC_DISTINCT

Introduction

In this page you can find the example usage for org.apache.solr.common.params StatsParams STATS_CALC_DISTINCT.

Prototype

String STATS_CALC_DISTINCT

To view the source code for org.apache.solr.common.params StatsParams STATS_CALC_DISTINCT.

Click Source Link

Usage

From source file:com.frank.search.solr.core.DefaultQueryParser.java

License:Apache License

private void processStatsOptions(SolrQuery solrQuery, Query query) {
    StatsOptions statsOptions = query.getStatsOptions();

    if (statsOptions == null || (CollectionUtils.isEmpty(statsOptions.getFields())
            && CollectionUtils.isEmpty(statsOptions.getFacets())
            && CollectionUtils.isEmpty(statsOptions.getSelectiveFacets()))) {
        return;//from   ww  w  .ja  v a2s.  c o m
    }

    solrQuery.set(StatsParams.STATS, true);

    for (Field field : statsOptions.getFields()) {
        solrQuery.add(StatsParams.STATS_FIELD, field.getName());

        String selectiveCalcDistinctParam = CommonParams.FIELD + "." + field.getName() + "."
                + StatsParams.STATS_CALC_DISTINCT;
        Boolean selectiveCountDistincts = statsOptions.isSelectiveCalcDistincts(field);

        if (selectiveCountDistincts != null) {
            solrQuery.add(selectiveCalcDistinctParam, String.valueOf(selectiveCountDistincts.booleanValue()));
        }

    }

    for (Field field : statsOptions.getFacets()) {
        solrQuery.add(StatsParams.STATS_FACET, field.getName());
    }

    for (Entry<Field, Collection<Field>> entry : statsOptions.getSelectiveFacets().entrySet()) {

        Field field = entry.getKey();
        String prefix = CommonParams.FIELD + "." + field.getName() + ".";

        String paramName = prefix + StatsParams.STATS_FACET;
        for (Field facetField : entry.getValue()) {
            solrQuery.add(paramName, facetField.getName());
        }

    }

}

From source file:org.springframework.data.solr.core.DefaultQueryParserTests.java

License:Apache License

/**
 * @see DATASOLR-160//from   www.ja  v  a2s.  c  o  m
 */
@Test
public void testConstructSolrQueryWithSelectiveStatsCountDistinct() {
    StatsOptions statsOptions = new StatsOptions()//
            .addField(new SimpleField("field_1")).setSelectiveCalcDistinct(true) //
            .addField(new SimpleField("field_2")).setSelectiveCalcDistinct(false) //
            .addField(new SimpleField("field_3"));

    SimpleQuery query = new SimpleQuery("*:*");
    query.setStatsOptions(statsOptions);

    SolrQuery solrQuery = queryParser.constructSolrQuery(query);

    String[] fields = solrQuery.getParams(StatsParams.STATS_FIELD);
    String[] calc1 = solrQuery.getParams(CommonParams.FIELD + ".field_1." + StatsParams.STATS_CALC_DISTINCT);
    String[] calc2 = solrQuery.getParams(CommonParams.FIELD + ".field_2." + StatsParams.STATS_CALC_DISTINCT);
    String[] calc3 = solrQuery.getParams(CommonParams.FIELD + ".field_3." + StatsParams.STATS_CALC_DISTINCT);

    Arrays.sort(fields);

    Assert.assertEquals(3, fields.length);
    Assert.assertArrayEquals(new String[] { "field_1", "field_2", "field_3" }, fields);
    Assert.assertEquals("true", calc1[0]);
    Assert.assertEquals("false", calc2[0]);
    Assert.assertNull(calc3);
}