DATAES-11 - another example how to use @FieldType for Collections

Author: akonczak

src/main/java/org/springframework/data/elasticsearch/entities/Article.java

@@ -7,6 +7,7 @@
 import org.springframework.data.elasticsearch.annotations.NestedField;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.List;
 
 import static org.springframework.data.elasticsearch.annotations.FieldIndex.analyzed;
@@ -32,7 +33,10 @@ public class Article {
     private List<String> authors = new ArrayList<String>();
 
     @Field(type = Integer, store = true)
-    private List<Integer> publishedYears = new ArrayList<Integer>();
+    private List<Integer> publishedYears;
+
+    @Field(type = String, store = true)
+    private Collection<String> tags;
 
     private int score;
 
@@ -83,4 +87,12 @@ public int getScore() {
     public void setScore(int score) {
         this.score = score;
     }
+
+    public Collection<String> getTags() {
+        return tags;
+    }
+
+    public void setTags(Collection<String> tags) {
+        this.tags = tags;
+    }
 }

src/main/java/org/springframework/data/elasticsearch/entities/ArticleBuilder.java

@@ -2,6 +2,9 @@
 
 import org.springframework.data.elasticsearch.core.query.IndexQuery;
 
+import java.util.ArrayList;
+import java.util.List;
+
 public class ArticleBuilder {
 
     private Article resutl;
@@ -34,6 +37,17 @@ public Article build() {
         return resutl;
     }
 
+    public ArticleBuilder addTag(String tag) {
+        List<String> tagsTmp = new ArrayList<String>();
+        if(resutl.getTags()==null){
+            resutl.setTags(tagsTmp);
+        }else {
+            tagsTmp = (List<String>) resutl.getTags();
+        }
+        tagsTmp.add(tag);
+        return this;
+    }
+
     public IndexQuery buildIndex() {
         IndexQuery indexQuery = new IndexQuery();
         indexQuery.setId(resutl.getId());

src/test/java/org/springframework/data/elasticsearch/repositories/SampleArticleRepositoryTest.java

@@ -9,6 +9,9 @@
 
 import javax.annotation.Resource;
 
+import java.util.ArrayList;
+import java.util.List;
+
 import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.junit.Assert.assertThat;
@@ -31,11 +34,22 @@ public void shouldIndexSingleBookEntity(){
         Article article = new Article();
         article.setId("123455");
         article.setTitle("Spring Data Elasticsearch");
+        List<String> authors = new ArrayList<String>();
+        authors.add("Author1");
+        authors.add("Author2");
+        article.setAuthors(authors);
+        List<String> tags = new ArrayList<String>();
+        tags.add("tag1");
+        tags.add("tag2");
+        tags.add("tag3");
+        article.setTags(tags);
         //Indexing using sampleArticleRepository
         sampleArticleRepository.save(article);
         //lets try to search same record in elasticsearch
         Article indexedArticle = sampleArticleRepository.findOne(article.getId());
         assertThat(indexedArticle,is(notNullValue()));
         assertThat(indexedArticle.getId(),is(article.getId()));
+        assertThat(indexedArticle.getAuthors().size(),is(authors.size()));
+        assertThat(indexedArticle.getTags().size(),is(tags.size()));
     }
 }