DATAES-26 - added support for histogram facets

src/test/java/org/springframework/data/elasticsearch/core/facet/ElasticsearchTemplateFacetTests.java

@@ -500,4 +500,32 @@ public class ElasticsearchTemplateFacetTests {
         assertThat(facet.getMax(), is(equalTo(2002.0)));
         assertThat(facet.getMin(), is(equalTo(2000.0)));
     }
+
+    @Test
+    public void shouldReturnHistogramFacetForGivenQuery() {
+        // given
+        String facetName = "numberPublicationPerYear";
+        SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery())
+                .withFacet(new HistogramFacetRequestBuilder(facetName).field(PUBLISHED_YEARS).interval(1).build()
+                ).build();
+        // when
+        FacetedPage<ArticleEntity> result = elasticsearchTemplate.queryForPage(searchQuery, ArticleEntity.class);
+        // then
+        assertThat(result.getNumberOfElements(), is(equalTo(4)));
+
+        HistogramResult facet = (HistogramResult) result.getFacet(facetName);
+        assertThat(facet.getIntervalUnit().size(), is(equalTo(3)));
+
+        IntervalUnit unit = facet.getIntervalUnit().get(0);
+        assertThat(unit.getKey(), is(Long.valueOf(YEAR_2000)));
+        assertThat(unit.getCount(), is(3L));
+
+        unit = facet.getIntervalUnit().get(1);
+        assertThat(unit.getKey(), is(Long.valueOf(YEAR_2001)));
+        assertThat(unit.getCount(), is(2L));
+
+        unit = facet.getIntervalUnit().get(2);
+        assertThat(unit.getKey(), is(Long.valueOf(YEAR_2002)));
+        assertThat(unit.getCount(), is(1L));
+    }
 }

src/test/java/org/springframework/data/elasticsearch/core/facet/ElasticsearchTemplateHistogramFacetTests.java

@@ -0,0 +1,139 @@
+/*
+ * Copyright 2013 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.elasticsearch.core.facet;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
+import org.springframework.data.elasticsearch.core.FacetedPage;
+import org.springframework.data.elasticsearch.core.facet.request.HistogramFacetRequestBuilder;
+import org.springframework.data.elasticsearch.core.facet.result.*;
+import org.springframework.data.elasticsearch.core.query.IndexQuery;
+import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
+import org.springframework.data.elasticsearch.core.query.SearchQuery;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.concurrent.TimeUnit;
+
+import static org.elasticsearch.index.query.QueryBuilders.matchAllQuery;
+import static org.hamcrest.Matchers.*;
+import static org.junit.Assert.assertThat;
+
+/**
+ * @author Rizwan Idrees
+ * @author Mohsin Husen
+ * @author Jonathan Yan
+ * @author Artur Konczak
+ */
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration("classpath:elasticsearch-template-test.xml")
+public class ElasticsearchTemplateHistogramFacetTests {
+
+    public static final long SEQUECE_CODE_INSERT = 1;
+    public static final long SEQUECE_CODE_UPDATE = 2;
+    public static final long SEQUECE_CODE_DELETE = 3;
+    public static final SimpleDateFormat dateFormatter = new SimpleDateFormat("yyyy-MM-dd HH:mm");
+    public static final String DATE_18 = "2013-10-18 18:01";
+    public static final String DATE_17 = "2013-10-18 17:01";
+    public static final String DATE_16 = "2013-10-18 16:01";
+
+
+    @Autowired
+    private ElasticsearchTemplate elasticsearchTemplate;
+
+    @Before
+    public void before() throws ParseException {
+        elasticsearchTemplate.deleteIndex(LogEntity.class);
+        elasticsearchTemplate.createIndex(LogEntity.class);
+        elasticsearchTemplate.putMapping(LogEntity.class);
+        elasticsearchTemplate.refresh(LogEntity.class, true);
+
+        IndexQuery entry1 = new LogEntityBuilder("1").action("update").date(dateFormatter.parse(DATE_18)).code(SEQUECE_CODE_UPDATE).buildIndex();
+        IndexQuery entry2 = new LogEntityBuilder("2").action("insert").date(dateFormatter.parse(DATE_17)).code(SEQUECE_CODE_INSERT).buildIndex();
+        IndexQuery entry3 = new LogEntityBuilder("3").action("update").date(dateFormatter.parse(DATE_17)).code(SEQUECE_CODE_UPDATE).buildIndex();
+        IndexQuery entry4 = new LogEntityBuilder("4").action("delete").date(dateFormatter.parse(DATE_16)).code(SEQUECE_CODE_DELETE).buildIndex();
+
+
+        elasticsearchTemplate.index(entry1);
+        elasticsearchTemplate.index(entry2);
+        elasticsearchTemplate.index(entry3);
+        elasticsearchTemplate.index(entry4);
+
+        elasticsearchTemplate.refresh(LogEntity.class, true);
+    }
+
+
+    @Test
+    public void shouldReturnSimpleHistogramFacetForGivenQuery() {
+        // given
+        String facetName = "sequenceCodeFacet";
+        SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery())
+                .withFacet(new HistogramFacetRequestBuilder(facetName).field("sequenceCode").interval(1).build()
+                ).build();
+        // when
+        FacetedPage<LogEntity> result = elasticsearchTemplate.queryForPage(searchQuery, LogEntity.class);
+        // then
+        assertThat(result.getNumberOfElements(), is(equalTo(4)));
+
+        HistogramResult facet = (HistogramResult) result.getFacet(facetName);
+        assertThat(facet.getIntervalUnit().size(), is(equalTo(3)));
+
+        IntervalUnit unit = facet.getIntervalUnit().get(0);
+        assertThat(unit.getKey(), is(SEQUECE_CODE_INSERT));
+        assertThat(unit.getCount(), is(1L));
+
+        unit = facet.getIntervalUnit().get(1);
+        assertThat(unit.getKey(), is(SEQUECE_CODE_UPDATE));
+        assertThat(unit.getCount(), is(2L));
+
+        unit = facet.getIntervalUnit().get(2);
+        assertThat(unit.getKey(), is(SEQUECE_CODE_DELETE));
+        assertThat(unit.getCount(), is(1L));
+    }
+
+    @Test
+    public void shouldReturnDateHistogramFacetForGivenQuery() throws ParseException {
+        // given
+        String facetName = "sequenceCodeFacet";
+        SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery())
+                .withFacet(new HistogramFacetRequestBuilder(facetName).field("date").interval(1).timeUnit(TimeUnit.HOURS).build()
+                ).build();
+        // when
+        FacetedPage<LogEntity> result = elasticsearchTemplate.queryForPage(searchQuery, LogEntity.class);
+        // then
+        assertThat(result.getNumberOfElements(), is(equalTo(4)));
+
+        HistogramResult facet = (HistogramResult) result.getFacet(facetName);
+        assertThat(facet.getIntervalUnit().size(), is(equalTo(3)));
+
+        IntervalUnit unit = facet.getIntervalUnit().get(0);
+        assertThat(unit.getKey(),is(dateFormatter.parse("2013-10-18 16:00").getTime()));
+        assertThat(unit.getCount(), is(1L));
+
+        unit = facet.getIntervalUnit().get(1);
+        assertThat(unit.getKey(),is(dateFormatter.parse("2013-10-18 17:00").getTime()));
+        assertThat(unit.getCount(), is(2L));
+
+        unit = facet.getIntervalUnit().get(2);
+        assertThat(unit.getKey(),is(dateFormatter.parse("2013-10-18 18:00").getTime()));
+        assertThat(unit.getCount(), is(1L));
+    }
+}

src/test/java/org/springframework/data/elasticsearch/core/facet/LogEntity.java

@@ -0,0 +1,74 @@
+package org.springframework.data.elasticsearch.core.facet;
+
+import org.springframework.data.annotation.Id;
+import org.springframework.data.elasticsearch.annotations.DateFormat;
+import org.springframework.data.elasticsearch.annotations.Document;
+import org.springframework.data.elasticsearch.annotations.Field;
+import org.springframework.data.elasticsearch.annotations.FieldType;
+
+import java.text.SimpleDateFormat;
+import java.util.Date;
+
+/**
+ * User: Artur Konczak
+ * Date: 18/10/13
+ * Time: 17:33
+ */
+@Document(indexName = "logs", type = "log", shards = 1, replicas = 0, refreshInterval = "-1", indexStoreType = "memory")
+public class LogEntity {
+
+    private static final SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm");
+
+    @Id
+    private String id;
+
+    private String action;
+
+    private long sequenceCode;
+
+    @Field(type = FieldType.Date, format = DateFormat.basic_date_time)
+    private Date date;
+
+    private LogEntity() {
+    }
+
+    public LogEntity(String id) {
+        this.id = id;
+    }
+
+    public String getId() {
+        return id;
+    }
+
+    public void setId(String id) {
+        this.id = id;
+    }
+
+    public String getAction() {
+        return action;
+    }
+
+    public String toString(){
+        return new StringBuffer().append("{id:").append(id).append(",action:").append(action).append(",code:").append(sequenceCode).append(",date:").append(format.format(date)).append("}").toString();
+    }
+
+    public void setAction(String action) {
+        this.action = action;
+    }
+
+    public long getSequenceCode() {
+        return sequenceCode;
+    }
+
+    public void setSequenceCode(long sequenceCode) {
+        this.sequenceCode = sequenceCode;
+    }
+
+    public Date getDate() {
+        return date;
+    }
+
+    public void setDate(Date date) {
+        this.date = date;
+    }
+}

src/test/java/org/springframework/data/elasticsearch/core/facet/LogEntityBuilder.java

@@ -0,0 +1,46 @@
+package org.springframework.data.elasticsearch.core.facet;
+
+import org.springframework.data.elasticsearch.core.query.IndexQuery;
+
+import java.util.Date;
+
+/**
+ * User: Artur Konczak
+ * Date: 18/10/13
+ * Time: 17:33
+ */
+public class LogEntityBuilder {
+
+    private LogEntity result;
+
+    public LogEntityBuilder(String id) {
+        result = new LogEntity(id);
+    }
+
+    public LogEntityBuilder action(String action) {
+        result.setAction(action);
+        return this;
+    }
+
+    public LogEntityBuilder code(long sequenceCode) {
+        result.setSequenceCode(sequenceCode);
+        return this;
+    }
+
+    public LogEntityBuilder date(Date date) {
+        result.setDate(date);
+        return this;
+    }
+
+    public LogEntity build() {
+        return result;
+    }
+
+    public IndexQuery buildIndex() {
+        IndexQuery indexQuery = new IndexQuery();
+        indexQuery.setId(result.getId());
+        indexQuery.setObject(result);
+        return indexQuery;
+    }
+
+}