Search in sources :

Example 1 with RangeFacet

use of org.apache.solr.client.solrj.response.RangeFacet in project lucene-solr by apache.

the class DistributedFacetPivotSmallTest method createExpectedRange.

private List<RangeFacet> createExpectedRange(String key, int start, int end, int gap, int... values) {
    List<RangeFacet> expectedRanges = new ArrayList<>();
    RangeFacet expectedPrices = new RangeFacet.Numeric(key, start, end, gap, null, null, null);
    expectedRanges.add(expectedPrices);
    int idx = 0;
    for (int range = start; range < end; range += gap) {
        expectedPrices.addCount(String.valueOf(range), values[idx]);
        if (idx < values.length) {
            idx++;
        }
    }
    return expectedRanges;
}
Also used : RangeFacet(org.apache.solr.client.solrj.response.RangeFacet) ArrayList(java.util.ArrayList)

Example 2 with RangeFacet

use of org.apache.solr.client.solrj.response.RangeFacet in project lucene-solr by apache.

the class SolrExampleTests method testPivotFacetsRanges.

@Test
public void testPivotFacetsRanges() throws Exception {
    SolrClient client = getSolrClient();
    // Empty the database...
    // delete everything!
    client.deleteByQuery("*:*");
    client.commit();
    // make sure it got in
    assertNumFound("*:*", 0);
    int id = 1;
    ArrayList<SolrInputDocument> docs = new ArrayList<>();
    docs.add(makeTestDoc("id", id++, "features", "aaa", "cat", "a", "inStock", true, "popularity", 12, "price", .017));
    docs.add(makeTestDoc("id", id++, "features", "aaa", "cat", "a", "inStock", false, "popularity", 13, "price", 16.04));
    docs.add(makeTestDoc("id", id++, "features", "aaa", "cat", "a", "inStock", true, "popularity", 14, "price", 12.34));
    docs.add(makeTestDoc("id", id++, "features", "aaa", "cat", "b", "inStock", false, "popularity", 24, "price", 51.39));
    docs.add(makeTestDoc("id", id++, "features", "aaa", "cat", "b", "inStock", true, "popularity", 28, "price", 131.39));
    docs.add(makeTestDoc("id", id++, "features", "bbb", "cat", "a", "inStock", false, "popularity", 32));
    docs.add(makeTestDoc("id", id++, "features", "bbb", "cat", "a", "inStock", true, "popularity", 31, "price", 131.39));
    docs.add(makeTestDoc("id", id++, "features", "bbb", "cat", "b", "inStock", false, "popularity", 36));
    docs.add(makeTestDoc("id", id++, "features", "bbb", "cat", "b", "inStock", true, "popularity", 37, "price", 1.39));
    docs.add(makeTestDoc("id", id++, "features", "bbb", "cat", "b", "inStock", false, "popularity", 38, "price", 47.98));
    docs.add(makeTestDoc("id", id++, "features", "bbb", "cat", "b", "inStock", true, "popularity", -38));
    // something not matching all fields
    docs.add(makeTestDoc("id", id++, "cat", "b"));
    client.add(docs);
    client.commit();
    SolrQuery query = new SolrQuery("*:*");
    query.addFacetPivotField("{!range=s1}features,manu");
    query.add(FacetParams.FACET_RANGE, "{!key=price1 tag=s1}price");
    query.add(String.format(Locale.ROOT, "f.%s.%s", "price", FacetParams.FACET_RANGE_START), "0");
    query.add(String.format(Locale.ROOT, "f.%s.%s", "price", FacetParams.FACET_RANGE_END), "200");
    query.add(String.format(Locale.ROOT, "f.%s.%s", "price", FacetParams.FACET_RANGE_GAP), "50");
    query.set(FacetParams.FACET, true);
    query.add(FacetParams.FACET_RANGE, "{!key=price2 tag=s1}price");
    query.setFacetMinCount(0);
    query.setRows(0);
    QueryResponse rsp = client.query(query);
    List<RangeFacet> list = rsp.getFacetRanges();
    assertEquals(2, list.size());
    @SuppressWarnings("unchecked") RangeFacet<Float, Float> range1 = list.get(0);
    assertEquals("price1", range1.getName());
    assertEquals(0, range1.getStart().intValue());
    assertEquals(200, range1.getEnd().intValue());
    assertEquals(50, range1.getGap().intValue());
    List<Count> counts1 = range1.getCounts();
    assertEquals(4, counts1.size());
    assertEquals(5, counts1.get(0).getCount());
    assertEquals("0.0", counts1.get(0).getValue());
    assertEquals(1, counts1.get(1).getCount());
    assertEquals("50.0", counts1.get(1).getValue());
    assertEquals(2, counts1.get(2).getCount());
    assertEquals("100.0", counts1.get(2).getValue());
    assertEquals(0, counts1.get(3).getCount());
    assertEquals("150.0", counts1.get(3).getValue());
    @SuppressWarnings("unchecked") RangeFacet<Float, Float> range2 = list.get(1);
    assertEquals("price2", range2.getName());
    assertEquals(0, range2.getStart().intValue());
    assertEquals(200, range2.getEnd().intValue());
    assertEquals(50, range2.getGap().intValue());
    List<Count> counts2 = range2.getCounts();
    assertEquals(4, counts2.size());
    assertEquals(5, counts2.get(0).getCount());
    assertEquals("0.0", counts2.get(0).getValue());
    assertEquals(1, counts2.get(1).getCount());
    assertEquals("50.0", counts2.get(1).getValue());
    assertEquals(2, counts2.get(2).getCount());
    assertEquals("100.0", counts2.get(2).getValue());
    assertEquals(0, counts2.get(3).getCount());
    assertEquals("150.0", counts2.get(3).getValue());
    NamedList<List<PivotField>> pivots = rsp.getFacetPivot();
    List<PivotField> pivotValues = pivots.get("features,manu");
    PivotField featuresBBBPivot = pivotValues.get(0);
    assertEquals("features", featuresBBBPivot.getField());
    assertEquals("bbb", featuresBBBPivot.getValue());
    List<RangeFacet> featuresBBBRanges = featuresBBBPivot.getFacetRanges();
    for (RangeFacet range : featuresBBBRanges) {
        if (range.getName().equals("price1")) {
            assertNotNull(range);
            assertEquals(0, ((Float) range.getStart()).intValue());
            assertEquals(200, ((Float) range.getEnd()).intValue());
            assertEquals(50, ((Float) range.getGap()).intValue());
            List<Count> counts = range.getCounts();
            assertEquals(4, counts.size());
            for (Count count : counts) {
                switch(count.getValue()) {
                    case "0.0":
                        assertEquals(2, count.getCount());
                        break;
                    case "50.0":
                        assertEquals(0, count.getCount());
                        break;
                    case "100.0":
                        assertEquals(1, count.getCount());
                        break;
                    case "150.0":
                        assertEquals(0, count.getCount());
                        break;
                }
            }
        } else if (range.getName().equals("price2")) {
            assertNotNull(range);
            assertEquals(0, ((Float) range.getStart()).intValue());
            assertEquals(200, ((Float) range.getEnd()).intValue());
            assertEquals(50, ((Float) range.getGap()).intValue());
            List<Count> counts = range.getCounts();
            assertEquals(4, counts.size());
            for (Count count : counts) {
                switch(count.getValue()) {
                    case "0.0":
                        assertEquals(2, count.getCount());
                        break;
                    case "50.0":
                        assertEquals(0, count.getCount());
                        break;
                    case "100.0":
                        assertEquals(1, count.getCount());
                        break;
                    case "150.0":
                        assertEquals(0, count.getCount());
                        break;
                }
            }
        }
    }
    PivotField featuresAAAPivot = pivotValues.get(1);
    assertEquals("features", featuresAAAPivot.getField());
    assertEquals("aaa", featuresAAAPivot.getValue());
    List<RangeFacet> facetRanges = featuresAAAPivot.getFacetRanges();
    for (RangeFacet range : facetRanges) {
        if (range.getName().equals("price1")) {
            assertNotNull(range);
            assertEquals(0, ((Float) range.getStart()).intValue());
            assertEquals(200, ((Float) range.getEnd()).intValue());
            assertEquals(50, ((Float) range.getGap()).intValue());
            List<Count> counts = range.getCounts();
            assertEquals(4, counts.size());
            for (Count count : counts) {
                switch(count.getValue()) {
                    case "0.0":
                        assertEquals(3, count.getCount());
                        break;
                    case "50.0":
                        assertEquals(1, count.getCount());
                        break;
                    case "100.0":
                        assertEquals(1, count.getCount());
                        break;
                    case "150.0":
                        assertEquals(0, count.getCount());
                        break;
                }
            }
        } else if (range.getName().equals("price2")) {
            assertNotNull(range);
            assertEquals(0, ((Float) range.getStart()).intValue());
            assertEquals(200, ((Float) range.getEnd()).intValue());
            assertEquals(50, ((Float) range.getGap()).intValue());
            List<Count> counts = range.getCounts();
            assertEquals(4, counts.size());
            for (Count count : counts) {
                switch(count.getValue()) {
                    case "0.0":
                        assertEquals(3, count.getCount());
                        break;
                    case "50.0":
                        assertEquals(1, count.getCount());
                        break;
                    case "100.0":
                        assertEquals(1, count.getCount());
                        break;
                    case "150.0":
                        assertEquals(0, count.getCount());
                        break;
                }
            }
        }
    }
}
Also used : RangeFacet(org.apache.solr.client.solrj.response.RangeFacet) ArrayList(java.util.ArrayList) Count(org.apache.solr.client.solrj.response.RangeFacet.Count) SolrInputDocument(org.apache.solr.common.SolrInputDocument) ErrorTrackingConcurrentUpdateSolrClient(org.apache.solr.client.solrj.embedded.SolrExampleStreamingTest.ErrorTrackingConcurrentUpdateSolrClient) HttpSolrClient(org.apache.solr.client.solrj.impl.HttpSolrClient) QueryResponse(org.apache.solr.client.solrj.response.QueryResponse) PivotField(org.apache.solr.client.solrj.response.PivotField) SolrDocumentList(org.apache.solr.common.SolrDocumentList) List(java.util.List) ArrayList(java.util.ArrayList) NamedList(org.apache.solr.common.util.NamedList) Test(org.junit.Test)

Aggregations

ArrayList (java.util.ArrayList)2 RangeFacet (org.apache.solr.client.solrj.response.RangeFacet)2 List (java.util.List)1 ErrorTrackingConcurrentUpdateSolrClient (org.apache.solr.client.solrj.embedded.SolrExampleStreamingTest.ErrorTrackingConcurrentUpdateSolrClient)1 HttpSolrClient (org.apache.solr.client.solrj.impl.HttpSolrClient)1 PivotField (org.apache.solr.client.solrj.response.PivotField)1 QueryResponse (org.apache.solr.client.solrj.response.QueryResponse)1 Count (org.apache.solr.client.solrj.response.RangeFacet.Count)1 SolrDocumentList (org.apache.solr.common.SolrDocumentList)1 SolrInputDocument (org.apache.solr.common.SolrInputDocument)1 NamedList (org.apache.solr.common.util.NamedList)1 Test (org.junit.Test)1