博客
关于我
强烈建议你试试无所不能的chatGPT，快点击我
spring-data-elasticsearch api
阅读量：6575 次
发布时间：2019-06-24

本文共 69095 字，大约阅读时间需要 230 分钟。

/* * Copyright 2014-2016 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;import static org.apache.commons.lang.RandomStringUtils.*;import static org.elasticsearch.index.query.QueryBuilders.*;import static org.hamcrest.Matchers.*;import static org.junit.Assert.*;import static org.springframework.data.elasticsearch.utils.IndexBuilder.*;import java.util.*;import org.apache.commons.lang.StringUtils;import org.elasticsearch.action.get.MultiGetItemResponse;import org.elasticsearch.action.get.MultiGetResponse;import org.elasticsearch.action.index.IndexRequest;import org.elasticsearch.action.search.SearchResponse;import org.elasticsearch.index.engine.DocumentMissingException;import org.elasticsearch.script.Script;import org.elasticsearch.script.ScriptService;import org.elasticsearch.search.SearchHit;import org.elasticsearch.search.highlight.HighlightBuilder;import org.elasticsearch.search.sort.FieldSortBuilder;import org.elasticsearch.search.sort.SortOrder;import org.hamcrest.Matchers;import org.junit.Before;import org.junit.Ignore;import org.junit.Test;import org.junit.runner.RunWith;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.data.domain.Page;import org.springframework.data.domain.PageRequest;import org.springframework.data.domain.Pageable;import org.springframework.data.domain.Sort;import org.springframework.data.elasticsearch.ElasticsearchException;import org.springframework.data.elasticsearch.annotations.Document;import org.springframework.data.elasticsearch.core.aggregation.AggregatedPage;import org.springframework.data.elasticsearch.core.aggregation.impl.AggregatedPageImpl;import org.springframework.data.elasticsearch.core.query.*;import org.springframework.data.elasticsearch.entities.*;import org.springframework.data.util.CloseableIterator;import org.springframework.test.context.ContextConfiguration;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;/** * @author Rizwan Idrees * @author Mohsin Husen * @author Franck Marchand * @author Abdul Mohammed * @author Kevin Leturc * @author Mason Chan */@RunWith(SpringJUnit4ClassRunner.class)@ContextConfiguration("classpath:elasticsearch-template-test.xml")public class ElasticsearchTemplateTests {	private static final String INDEX_NAME = "test-index";	private static final String INDEX_1_NAME = "test-index-1";	private static final String INDEX_2_NAME = "test-index-2";	private static final String TYPE_NAME = "test-type";	@Autowired	private ElasticsearchTemplate elasticsearchTemplate;	@Before	public void before() {		elasticsearchTemplate.deleteIndex(SampleEntity.class);		elasticsearchTemplate.createIndex(SampleEntity.class);		elasticsearchTemplate.deleteIndex(INDEX_1_NAME);		elasticsearchTemplate.deleteIndex(INDEX_2_NAME);		elasticsearchTemplate.deleteIndex(UseServerConfigurationEntity.class);		elasticsearchTemplate.refresh(SampleEntity.class);	}	/*	DATAES-106	 */	@Test	public void shouldReturnCountForGivenCriteriaQuery() {		// given		String documentId = randomNumeric(5);		SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message")				.version(System.currentTimeMillis()).build();		IndexQuery indexQuery = getIndexQuery(sampleEntity);		elasticsearchTemplate.index(indexQuery);		elasticsearchTemplate.refresh(SampleEntity.class);		CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria());		// when		long count = elasticsearchTemplate.count(criteriaQuery, SampleEntity.class);		// then		assertThat(count, is(equalTo(1L)));	}	@Test	public void shouldReturnCountForGivenSearchQuery() {		// given		String documentId = randomNumeric(5);		SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message")				.version(System.currentTimeMillis()).build();		IndexQuery indexQuery = getIndexQuery(sampleEntity);		elasticsearchTemplate.index(indexQuery);		elasticsearchTemplate.refresh(SampleEntity.class);		SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).build();		// when		long count = elasticsearchTemplate.count(searchQuery, SampleEntity.class);		// then		assertThat(count, is(equalTo(1L)));	}	@Test	public void shouldReturnObjectForGivenId() {		// given		String documentId = randomNumeric(5);		SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message")				.version(System.currentTimeMillis()).build();		IndexQuery indexQuery = getIndexQuery(sampleEntity);		elasticsearchTemplate.index(indexQuery);		// when		GetQuery getQuery = new GetQuery();		getQuery.setId(documentId);		SampleEntity sampleEntity1 = elasticsearchTemplate.queryForObject(getQuery, SampleEntity.class);		// then		assertNotNull("entity can't be null....", sampleEntity1);		assertEquals(sampleEntity, sampleEntity1);	}	@Test	public void shouldReturnObjectsForGivenIdsUsingMultiGet() {		// given		List
    
      indexQueries = new ArrayList
     
      ();		// first document		String documentId = randomNumeric(5);		SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId).message("some message")				.version(System.currentTimeMillis()).build();		// second document		String documentId2 = randomNumeric(5);		SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some message")				.version(System.currentTimeMillis()).build();		indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2));		elasticsearchTemplate.bulkIndex(indexQueries);		elasticsearchTemplate.refresh(SampleEntity.class);		// when		SearchQuery query = new NativeSearchQueryBuilder().withIds(Arrays.asList(documentId, documentId2)).build();		LinkedList
      
        sampleEntities = elasticsearchTemplate.multiGet(query, SampleEntity.class);		// then		assertThat(sampleEntities.size(), is(equalTo(2)));		assertEquals(sampleEntities.get(0), sampleEntity1);		assertEquals(sampleEntities.get(1), sampleEntity2);	}	@Test	public void shouldReturnObjectsForGivenIdsUsingMultiGetWithFields() {		// given		List
       
         indexQueries = new ArrayList
        
         ();		// first document		String documentId = randomNumeric(5);		SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId)				.message("some message")				.type("type1")				.version(System.currentTimeMillis()).build();		// second document		String documentId2 = randomNumeric(5);		SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2)				.message("some message")				.type("type2")				.version(System.currentTimeMillis()).build();		indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2));		elasticsearchTemplate.bulkIndex(indexQueries);		elasticsearchTemplate.refresh(SampleEntity.class);		// when		SearchQuery query = new NativeSearchQueryBuilder()				.withIds(Arrays.asList(documentId, documentId2))				.withFields("message", "type")				.build();		LinkedList
         
           sampleEntities = elasticsearchTemplate.multiGet(query, SampleEntity.class, new MultiGetResultMapper() { @Override public 
          
            LinkedList
           
             mapResults(MultiGetResponse responses, Class
            
              clazz) { LinkedList
             
               list = new LinkedList
              
               (); for (MultiGetItemResponse response : responses.getResponses()) { SampleEntity entity = new SampleEntity(); entity.setId(response.getResponse().getId()); entity.setMessage((String) response.getResponse().getField("message").getValue()); entity.setType((String) response.getResponse().getField("type").getValue()); list.add((T) entity); } return list; } }); // then assertThat(sampleEntities.size(), is(equalTo(2))); } @Test public void shouldReturnPageForGivenSearchQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).build(); // when Page
               
                 sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(sampleEntities, is(notNullValue())); assertThat(sampleEntities.getTotalElements(), greaterThanOrEqualTo(1L)); } @Test public void shouldDoBulkIndex() { // given List
                
                  indexQueries = new ArrayList
                 
                  (); // first document String documentId = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); // second document String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some message") .version(System.currentTimeMillis()).build(); indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2)); // when elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).build(); Page
                  
                    sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); assertThat(sampleEntities.getTotalElements(), is(equalTo(2L))); } @Test public void shouldDoBulkUpdate() { //given String documentId = randomNumeric(5); String messageBeforeUpdate = "some test message"; String messageAfterUpdate = "test message"; SampleEntity sampleEntity = SampleEntity.builder().id(documentId) .message(messageBeforeUpdate) .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); IndexRequest indexRequest = new IndexRequest(); indexRequest.source("message", messageAfterUpdate); UpdateQuery updateQuery = new UpdateQueryBuilder().withId(documentId) .withClass(SampleEntity.class).withIndexRequest(indexRequest).build(); List
                   
                     queries = new ArrayList
                    
                     (); queries.add(updateQuery); // when elasticsearchTemplate.bulkUpdate(queries); //then GetQuery getQuery = new GetQuery(); getQuery.setId(documentId); SampleEntity indexedEntity = elasticsearchTemplate.queryForObject(getQuery, SampleEntity.class); assertThat(indexedEntity.getMessage(), is(messageAfterUpdate)); } @Test public void shouldDeleteDocumentForGivenId() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); // when elasticsearchTemplate.delete(INDEX_NAME, TYPE_NAME, documentId); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", documentId)).build(); Page
                     
                       sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); assertThat(sampleEntities.getTotalElements(), equalTo(0L)); } @Test public void shouldDeleteEntityForGivenId() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); // when elasticsearchTemplate.delete(SampleEntity.class, documentId); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", documentId)).build(); Page
                      
                        sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); assertThat(sampleEntities.getTotalElements(), equalTo(0L)); } @Test public void shouldDeleteDocumentForGivenQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); // when DeleteQuery deleteQuery = new DeleteQuery(); deleteQuery.setQuery(termQuery("id", documentId)); elasticsearchTemplate.delete(deleteQuery, SampleEntity.class); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", documentId)).build(); Page
                       
                         sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); assertThat(sampleEntities.getTotalElements(), equalTo(0L)); } @Test public void shouldFilterSearchResultsForGivenFilter() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()) .withFilter(boolQuery().filter(termQuery("id", documentId))).build(); // when Page
                        
                          sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), equalTo(1L)); } @Test public void shouldSortResultsGivenSortCriteria() { // given List
                         
                           indexQueries = new ArrayList
                          
                           (); // first document String documentId = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId) .message("abc") .rate(10) .version(System.currentTimeMillis()).build(); // second document String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2) .message("xyz") .rate(5) .version(System.currentTimeMillis()).build(); // third document String documentId3 = randomNumeric(5); SampleEntity sampleEntity3 = SampleEntity.builder().id(documentId3) .message("xyz") .rate(15) .version(System.currentTimeMillis()).build(); indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2, sampleEntity3)); elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()) .withSort(new FieldSortBuilder("rate").ignoreUnmapped(true).order(SortOrder.ASC)).build(); // when Page
                           
                             sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), equalTo(3L)); assertThat(sampleEntities.getContent().get(0).getRate(), is(sampleEntity2.getRate())); } @Test public void shouldSortResultsGivenMultipleSortCriteria() { // given List
                            
                              indexQueries = new ArrayList
                             
                              (); // first document String documentId = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId) .message("abc") .rate(10) .version(System.currentTimeMillis()).build(); // second document String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2) .message("xyz") .rate(5) .version(System.currentTimeMillis()).build(); // third document String documentId3 = randomNumeric(5); SampleEntity sampleEntity3 = SampleEntity.builder().id(documentId3) .message("xyz") .rate(15) .version(System.currentTimeMillis()).build(); indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2, sampleEntity3)); elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()) .withSort(new FieldSortBuilder("rate").ignoreUnmapped(true).order(SortOrder.ASC)) .withSort(new FieldSortBuilder("message").ignoreUnmapped(true).order(SortOrder.ASC)).build(); // when Page
                              
                                sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), equalTo(3L)); assertThat(sampleEntities.getContent().get(0).getRate(), is(sampleEntity2.getRate())); assertThat(sampleEntities.getContent().get(1).getMessage(), is(sampleEntity1.getMessage())); } @Test public void shouldExecuteStringQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); StringQuery stringQuery = new StringQuery(matchAllQuery().toString()); // when Page
                               
                                 sampleEntities = elasticsearchTemplate.queryForPage(stringQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), equalTo(1L)); } @Test public void shouldUseScriptedFields() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setId(documentId); sampleEntity.setRate(2); sampleEntity.setMessage("some message"); sampleEntity.setVersion(System.currentTimeMillis()); IndexQuery indexQuery = new IndexQuery(); indexQuery.setId(documentId); indexQuery.setObject(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); Map
                                
                                  params = new HashMap
                                 
                                  (); params.put("factor", 2); // when SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .withScriptField(new ScriptField("scriptedRate", new Script("doc['rate'].value * factor", ScriptService.ScriptType.INLINE, null, params))) .build(); Page
                                  
                                    sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), equalTo(1L)); assertThat(sampleEntities.getContent().get(0).getScriptedRate(), equalTo(4L)); } @Test public void shouldReturnPageableResultsGivenStringQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); StringQuery stringQuery = new StringQuery(matchAllQuery().toString(), new PageRequest(0, 10)); // when Page
                                   
                                     sampleEntities = elasticsearchTemplate.queryForPage(stringQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), is(greaterThanOrEqualTo(1L))); } @Test @Ignore("By default, the search request will fail if there is no mapping associated with a field. The ignore_unmapped option allows to ignore fields that have no mapping and not sort by them") public void shouldReturnSortedPageableResultsGivenStringQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setId(documentId); sampleEntity.setMessage("some message"); sampleEntity.setVersion(System.currentTimeMillis()); IndexQuery indexQuery = new IndexQuery(); indexQuery.setId(documentId); indexQuery.setObject(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); StringQuery stringQuery = new StringQuery(matchAllQuery().toString(), new PageRequest(0, 10), new Sort( new Sort.Order(Sort.Direction.ASC, "messsage"))); // when Page
                                    
                                      sampleEntities = elasticsearchTemplate.queryForPage(stringQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), is(greaterThanOrEqualTo(1L))); } @Test public void shouldReturnObjectMatchingGivenStringQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); StringQuery stringQuery = new StringQuery(termQuery("id", documentId).toString()); // when SampleEntity sampleEntity1 = elasticsearchTemplate.queryForObject(stringQuery, SampleEntity.class); // then assertThat(sampleEntity1, is(notNullValue())); assertThat(sampleEntity1.getId(), is(equalTo(documentId))); } @Test public void shouldCreateIndexGivenEntityClass() { // when boolean created = elasticsearchTemplate.createIndex(SampleEntity.class); final Map setting = elasticsearchTemplate.getSetting(SampleEntity.class); // then assertThat(created, is(true)); assertThat(setting.get("index.number_of_shards"), Matchers.
                                     is("1")); assertThat(setting.get("index.number_of_replicas"), Matchers.is("0")); } @Test public void shouldExecuteGivenCriteriaQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria("message").contains("test")); // when SampleEntity sampleEntity1 = elasticsearchTemplate.queryForObject(criteriaQuery, SampleEntity.class); // then assertThat(sampleEntity1, is(notNullValue())); } @Test public void shouldDeleteGivenCriteriaQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria("message").contains("test")); // when elasticsearchTemplate.delete(criteriaQuery, SampleEntity.class); elasticsearchTemplate.refresh(SampleEntity.class); // then StringQuery stringQuery = new StringQuery(matchAllQuery().toString()); List
                                       
                                         sampleEntities = elasticsearchTemplate.queryForList(stringQuery, SampleEntity.class); assertThat(sampleEntities.size(), is(0)); } @Test public void shouldReturnSpecifiedFields() { // given String documentId = randomNumeric(5); String message = "some test message"; SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message(message) .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices(INDEX_NAME) .withTypes(TYPE_NAME).withFields("message").build(); // when Page
                                        
                                          page = elasticsearchTemplate.queryForPage(searchQuery, String.class, new SearchResultMapper() { @Override public 
                                         
                                           AggregatedPage
                                          
                                            mapResults(SearchResponse response, Class
                                           
                                             clazz, Pageable pageable) { List
                                            
                                              values = new ArrayList
                                             
                                              (); for (SearchHit searchHit : response.getHits()) { values.add((String) searchHit.field("message").value()); } return new AggregatedPageImpl
                                              
                                               ((List
                                               
                                                ) values); } }); // then assertThat(page, is(notNullValue())); assertThat(page.getTotalElements(), is(equalTo(1L))); assertThat(page.getContent().get(0), is(message)); } @Test public void shouldReturnFieldsBasedOnSourceFilter() { // given String documentId = randomNumeric(5); String message = "some test message"; SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message(message) .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); FetchSourceFilterBuilder sourceFilter = new FetchSourceFilterBuilder(); sourceFilter.withIncludes("message"); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices(INDEX_NAME) .withTypes(TYPE_NAME).withSourceFilter(sourceFilter.build()).build(); // when Page
                                                
                                                  page = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(page, is(notNullValue())); assertThat(page.getTotalElements(), is(equalTo(1L))); assertThat(page.getContent().get(0).getMessage(), is(message)); } @Test public void shouldReturnSimilarResultsGivenMoreLikeThisQuery() { // given String sampleMessage = "So we build a web site or an application and want to add search to it, " + "and then it hits us: getting search working is hard. We want our search solution to be fast," + " we want a painless setup and a completely free search schema, we want to be able to index data simply using JSON over HTTP, " + "we want our search server to be always available, we want to be able to start with one machine and scale to hundreds, " + "we want real-time search, we want simple multi-tenancy, and we want a solution that is built for the cloud."; String documentId1 = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId1).message(sampleMessage) .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); String documentId2 = randomNumeric(5); elasticsearchTemplate.index(getIndexQuery(SampleEntity.builder().id(documentId2).message(sampleMessage) .version(System.currentTimeMillis()).build())); elasticsearchTemplate.refresh(SampleEntity.class); MoreLikeThisQuery moreLikeThisQuery = new MoreLikeThisQuery(); moreLikeThisQuery.setId(documentId2); moreLikeThisQuery.addFields("message"); moreLikeThisQuery.setMinDocFreq(1); // when Page
                                                 
                                                   sampleEntities = elasticsearchTemplate.moreLikeThis(moreLikeThisQuery, SampleEntity.class); // then assertThat(sampleEntities.getTotalElements(), is(equalTo(1L))); assertThat(sampleEntities.getContent(), hasItem(sampleEntity)); } /* DATAES-167 */ @Test public void shouldReturnResultsWithScanAndScrollForGivenCriteriaQuery() { //given List
                                                  
                                                    entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices(INDEX_NAME); criteriaQuery.addTypes(TYPE_NAME); criteriaQuery.setPageable(new PageRequest(0, 10)); String scrollId = elasticsearchTemplate.scan(criteriaQuery, 1000, false); List
                                                   
                                                     sampleEntities = new ArrayList
                                                    
                                                     (); boolean hasRecords = true; while (hasRecords) { Page
                                                     
                                                       page = elasticsearchTemplate.scroll(scrollId, 5000L, SampleEntity.class); if (page.hasContent()) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } @Test public void shouldReturnResultsWithScanAndScrollForGivenSearchQuery() { //given List
                                                      
                                                        entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices(INDEX_NAME) .withTypes(TYPE_NAME).withPageable(new PageRequest(0, 10)).build(); String scrollId = elasticsearchTemplate.scan(searchQuery, 1000, false); List
                                                       
                                                         sampleEntities = new ArrayList
                                                        
                                                         (); boolean hasRecords = true; while (hasRecords) { Page
                                                         
                                                           page = elasticsearchTemplate.scroll(scrollId, 5000L, SampleEntity.class); if (page.hasContent()) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } /* DATAES-167 */ @Test public void shouldReturnResultsWithScanAndScrollForSpecifiedFieldsForCriteriaCriteria() { //given List
                                                          
                                                            entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices(INDEX_NAME); criteriaQuery.addTypes(TYPE_NAME); criteriaQuery.addFields("message"); criteriaQuery.setPageable(new PageRequest(0, 10)); String scrollId = elasticsearchTemplate.scan(criteriaQuery, 5000, false); List
                                                           
                                                             sampleEntities = new ArrayList
                                                            
                                                             (); boolean hasRecords = true; while (hasRecords) { Page
                                                             
                                                               page = elasticsearchTemplate.scroll(scrollId, 5000L, new SearchResultMapper() { @Override public 
                                                              
                                                                AggregatedPage
                                                               
                                                                 mapResults(SearchResponse response, Class
                                                                
                                                                  clazz, Pageable pageable) { List
                                                                 
                                                                   result = new ArrayList
                                                                  
                                                                   (); for (SearchHit searchHit : response.getHits()) { String message = searchHit.getFields().get("message").getValue(); SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setId(searchHit.getId()); sampleEntity.setMessage(message); result.add(sampleEntity); } if (result.size() > 0) { return new AggregatedPageImpl
                                                                   
                                                                    ((List
                                                                    
                                                                     ) result); } return null; } }); if (page != null) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } /* DATAES-84 */ @Test public void shouldReturnResultsWithScanAndScrollForSpecifiedFieldsForSearchCriteria() { //given List
                                                                     
                                                                       entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()) .withIndices(INDEX_NAME) .withTypes(TYPE_NAME) .withFields("message") .withQuery(matchAllQuery()) .withPageable(new PageRequest(0, 10)) .build(); String scrollId = elasticsearchTemplate.scan(searchQuery, 10000, false); List
                                                                      
                                                                        sampleEntities = new ArrayList
                                                                       
                                                                        (); boolean hasRecords = true; while (hasRecords) { Page
                                                                        
                                                                          page = elasticsearchTemplate.scroll(scrollId, 10000L, new SearchResultMapper() { @Override public 
                                                                         
                                                                           AggregatedPage
                                                                          
                                                                            mapResults(SearchResponse response, Class
                                                                           
                                                                             clazz, Pageable pageable) { List
                                                                            
                                                                              result = new ArrayList
                                                                             
                                                                              (); for (SearchHit searchHit : response.getHits()) { String message = searchHit.getFields().get("message").getValue(); SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setId(searchHit.getId()); sampleEntity.setMessage(message); result.add(sampleEntity); } if (result.size() > 0) { return new AggregatedPageImpl
                                                                              
                                                                               ((List
                                                                               
                                                                                ) result); } return null; } }); if (page != null) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } /* DATAES-167 */ @Test public void shouldReturnResultsForScanAndScrollWithCustomResultMapperForGivenCriteriaQuery() { //given List
                                                                                
                                                                                  entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices(INDEX_NAME); criteriaQuery.addTypes(TYPE_NAME); criteriaQuery.setPageable(new PageRequest(0, 10)); String scrollId = elasticsearchTemplate.scan(criteriaQuery, 5000, false); List
                                                                                 
                                                                                   sampleEntities = new ArrayList
                                                                                  
                                                                                   (); boolean hasRecords = true; while (hasRecords) { Page
                                                                                   
                                                                                     page = elasticsearchTemplate.scroll(scrollId, 5000L, new SearchResultMapper() { @Override public 
                                                                                    
                                                                                      AggregatedPage
                                                                                     
                                                                                       mapResults(SearchResponse response, Class
                                                                                      
                                                                                        clazz, Pageable pageable) { List
                                                                                       
                                                                                         chunk = new ArrayList
                                                                                        
                                                                                         (); for (SearchHit searchHit : response.getHits()) { if (response.getHits().getHits().length <= 0) { return null; } SampleEntity user = new SampleEntity(); user.setId(searchHit.getId()); user.setMessage((String) searchHit.getSource().get("message")); chunk.add(user); } if (chunk.size() > 0) { return new AggregatedPageImpl
                                                                                         
                                                                                          ((List
                                                                                          
                                                                                           ) chunk); } return null; } }); if (page != null) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } @Test public void shouldReturnResultsForScanAndScrollWithCustomResultMapperForGivenSearchQuery() { //given List
                                                                                           
                                                                                             entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices(INDEX_NAME) .withTypes(TYPE_NAME).withPageable(new PageRequest(0, 10)).build(); String scrollId = elasticsearchTemplate.scan(searchQuery, 1000, false); List
                                                                                            
                                                                                              sampleEntities = new ArrayList
                                                                                             
                                                                                              (); boolean hasRecords = true; while (hasRecords) { Page
                                                                                              
                                                                                                page = elasticsearchTemplate.scroll(scrollId, 5000L, new SearchResultMapper() { @Override public 
                                                                                               
                                                                                                 AggregatedPage
                                                                                                
                                                                                                  mapResults(SearchResponse response, Class
                                                                                                 
                                                                                                   clazz, Pageable pageable) { List
                                                                                                  
                                                                                                    chunk = new ArrayList
                                                                                                   
                                                                                                    (); for (SearchHit searchHit : response.getHits()) { if (response.getHits().getHits().length <= 0) { return null; } SampleEntity user = new SampleEntity(); user.setId(searchHit.getId()); user.setMessage((String) searchHit.getSource().get("message")); chunk.add(user); } if (chunk.size() > 0) { return new AggregatedPageImpl
                                                                                                    
                                                                                                     ((List
                                                                                                     
                                                                                                      ) chunk); } return null; } }); if (page != null) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } /* DATAES-217 */ @Test public void shouldReturnResultsWithScanAndScrollForGivenCriteriaQueryAndClass() { //given List
                                                                                                      
                                                                                                        entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.setPageable(new PageRequest(0, 10)); String scrollId = elasticsearchTemplate.scan(criteriaQuery, 1000, false, SampleEntity.class); List
                                                                                                       
                                                                                                         sampleEntities = new ArrayList
                                                                                                        
                                                                                                         (); boolean hasRecords = true; while (hasRecords) { Page
                                                                                                         
                                                                                                           page = elasticsearchTemplate.scroll(scrollId, 5000L, SampleEntity.class); if (page.hasContent()) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } /* DATAES-217 */ @Test public void shouldReturnResultsWithScanAndScrollForGivenSearchQueryAndClass() { //given List
                                                                                                          
                                                                                                            entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()) .withPageable(new PageRequest(0, 10)).build(); String scrollId = elasticsearchTemplate.scan(searchQuery, 1000, false, SampleEntity.class); List
                                                                                                           
                                                                                                             sampleEntities = new ArrayList
                                                                                                            
                                                                                                             (); boolean hasRecords = true; while (hasRecords) { Page
                                                                                                             
                                                                                                               page = elasticsearchTemplate.scroll(scrollId, 5000L, SampleEntity.class); if (page.hasContent()) { sampleEntities.addAll(page.getContent()); } else { hasRecords = false; } } elasticsearchTemplate.clearScroll(scrollId); assertThat(sampleEntities.size(), is(equalTo(30))); } /* DATAES-167 */ @Test public void shouldReturnResultsWithStreamForGivenCriteriaQuery() { //given List
                                                                                                              
                                                                                                                entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); // then CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices(INDEX_NAME); criteriaQuery.addTypes(TYPE_NAME); criteriaQuery.setPageable(new PageRequest(0, 10)); CloseableIterator
                                                                                                               
                                                                                                                 stream = elasticsearchTemplate.stream(criteriaQuery, SampleEntity.class); List
                                                                                                                
                                                                                                                  sampleEntities = new ArrayList
                                                                                                                 
                                                                                                                  (); while (stream.hasNext()) { sampleEntities.add(stream.next()); } assertThat(sampleEntities.size(), is(equalTo(30))); } private static List
                                                                                                                  
                                                                                                                    createSampleEntitiesWithMessage(String message, int numberOfEntities) { List
                                                                                                                   
                                                                                                                     indexQueries = new ArrayList
                                                                                                                    
                                                                                                                     (); for (int i = 0; i < numberOfEntities; i++) { String documentId = UUID.randomUUID().toString(); SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setId(documentId); sampleEntity.setMessage(message); sampleEntity.setRate(2); sampleEntity.setVersion(System.currentTimeMillis()); IndexQuery indexQuery = new IndexQuery(); indexQuery.setId(documentId); indexQuery.setObject(sampleEntity); indexQueries.add(indexQuery); } return indexQueries; } @Test public void shouldReturnListForGivenCriteria() { // given List
                                                                                                                     
                                                                                                                       indexQueries = new ArrayList
                                                                                                                      
                                                                                                                       (); // first document String documentId = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId) .message("test message") .version(System.currentTimeMillis()).build(); // second document String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2) .message("test test") .rate(5) .version(System.currentTimeMillis()).build(); // third document String documentId3 = randomNumeric(5); SampleEntity sampleEntity3 = SampleEntity.builder().id(documentId3) .message("some message") .rate(15) .version(System.currentTimeMillis()).build(); indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2, sampleEntity3)); // when elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); // when CriteriaQuery singleCriteriaQuery = new CriteriaQuery(new Criteria("message").contains("test")); CriteriaQuery multipleCriteriaQuery = new CriteriaQuery(new Criteria("message").contains("some").and("message") .contains("message")); List
                                                                                                                       
                                                                                                                         sampleEntitiesForSingleCriteria = elasticsearchTemplate.queryForList(singleCriteriaQuery, SampleEntity.class); List
                                                                                                                        
                                                                                                                          sampleEntitiesForAndCriteria = elasticsearchTemplate.queryForList(multipleCriteriaQuery, SampleEntity.class); // then assertThat(sampleEntitiesForSingleCriteria.size(), is(2)); assertThat(sampleEntitiesForAndCriteria.size(), is(1)); } @Test public void shouldReturnListForGivenStringQuery() { // given // first document String documentId = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId) .message("test message") .version(System.currentTimeMillis()).build(); // second document String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2) .message("test test") .rate(5) .version(System.currentTimeMillis()).build(); // third document String documentId3 = randomNumeric(5); SampleEntity sampleEntity3 = SampleEntity.builder().id(documentId3) .message("some message") .rate(15) .version(System.currentTimeMillis()).build(); List
                                                                                                                         
                                                                                                                           indexQueries = getIndexQueries(Arrays.asList(sampleEntity1, sampleEntity2, sampleEntity3)); // when elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); // when StringQuery stringQuery = new StringQuery(matchAllQuery().toString()); List
                                                                                                                          
                                                                                                                            sampleEntities = elasticsearchTemplate.queryForList(stringQuery, SampleEntity.class); // then assertThat(sampleEntities.size(), is(3)); } @Test public void shouldPutMappingForGivenEntity() throws Exception { // given Class entity = SampleMappingEntity.class; elasticsearchTemplate.createIndex(entity); // when assertThat(elasticsearchTemplate.putMapping(entity), is(true)); } @Test public void shouldDeleteIndexForGivenEntity() { // given Class clazz = SampleEntity.class; // when elasticsearchTemplate.deleteIndex(clazz); // then assertThat(elasticsearchTemplate.indexExists(clazz), is(false)); } @Test public void shouldDoPartialUpdateForExistingDocument() { //given String documentId = randomNumeric(5); String messageBeforeUpdate = "some test message"; String messageAfterUpdate = "test message"; SampleEntity sampleEntity = SampleEntity.builder().id(documentId) .message(messageBeforeUpdate) .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); IndexRequest indexRequest = new IndexRequest(); indexRequest.source("message", messageAfterUpdate); UpdateQuery updateQuery = new UpdateQueryBuilder().withId(documentId) .withClass(SampleEntity.class).withIndexRequest(indexRequest).build(); // when elasticsearchTemplate.update(updateQuery); //then GetQuery getQuery = new GetQuery(); getQuery.setId(documentId); SampleEntity indexedEntity = elasticsearchTemplate.queryForObject(getQuery, SampleEntity.class); assertThat(indexedEntity.getMessage(), is(messageAfterUpdate)); } @Test(expected = DocumentMissingException.class) public void shouldThrowExceptionIfDocumentDoesNotExistWhileDoingPartialUpdate() { // when IndexRequest indexRequest = new IndexRequest(); UpdateQuery updateQuery = new UpdateQueryBuilder().withId(randomNumeric(5)) .withClass(SampleEntity.class).withIndexRequest(indexRequest).build(); elasticsearchTemplate.update(updateQuery); } @Test public void shouldDoUpsertIfDocumentDoesNotExist() { //given String documentId = randomNumeric(5); String message = "test message"; IndexRequest indexRequest = new IndexRequest(); indexRequest.source("message", message); UpdateQuery updateQuery = new UpdateQueryBuilder().withId(documentId) .withDoUpsert(true).withClass(SampleEntity.class) .withIndexRequest(indexRequest).build(); //when elasticsearchTemplate.update(updateQuery); //then GetQuery getQuery = new GetQuery(); getQuery.setId(documentId); SampleEntity indexedEntity = elasticsearchTemplate.queryForObject(getQuery, SampleEntity.class); assertThat(indexedEntity.getMessage(), is(message)); } @Test public void shouldReturnHighlightedFieldsForGivenQueryAndFields() { //given String documentId = randomNumeric(5); String actualMessage = "some test message"; String highlightedMessage = "some 
                                                                                                                           test message"; SampleEntity sampleEntity = SampleEntity.builder().id(documentId) .message(actualMessage) .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(termQuery("message", "test")) .withHighlightFields(new HighlightBuilder.Field("message")) .build(); Page
                                                                                                                           
                                                                                                                             sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class, new SearchResultMapper() { @Override public 
                                                                                                                            
                                                                                                                              AggregatedPage
                                                                                                                             
                                                                                                                               mapResults(SearchResponse response, Class
                                                                                                                              
                                                                                                                                clazz, Pageable pageable) { List
                                                                                                                               
                                                                                                                                 chunk = new ArrayList
                                                                                                                                
                                                                                                                                 (); for (SearchHit searchHit : response.getHits()) { if (response.getHits().getHits().length <= 0) { return null; } SampleEntity user = new SampleEntity(); user.setId(searchHit.getId()); user.setMessage((String) searchHit.getSource().get("message")); user.setHighlightedMessage(searchHit.getHighlightFields().get("message").fragments()[0].toString()); chunk.add(user); } if (chunk.size() > 0) { return new AggregatedPageImpl
                                                                                                                                 
                                                                                                                                  ((List
                                                                                                                                  
                                                                                                                                   ) chunk); } return null; } }); assertThat(sampleEntities.getContent().get(0).getHighlightedMessage(), is(highlightedMessage)); } @Test public void shouldDeleteDocumentBySpecifiedTypeUsingDeleteQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId) .message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); // when DeleteQuery deleteQuery = new DeleteQuery(); deleteQuery.setQuery(termQuery("id", documentId)); deleteQuery.setIndex(INDEX_NAME); deleteQuery.setType(TYPE_NAME); elasticsearchTemplate.delete(deleteQuery); elasticsearchTemplate.refresh(INDEX_NAME); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", documentId)).build(); Page
                                                                                                                                   
                                                                                                                                     sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); assertThat(sampleEntities.getTotalElements(), equalTo(0L)); } @Test public void shouldIndexDocumentForSpecifiedSource() { // given String documentSource = "{\"id\":\"2333343434\",\"type\":null,\"message\":\"some message\",\"rate\":0,\"available\":false,\"highlightedMessage\":null,\"version\":1385208779482}"; IndexQuery indexQuery = new IndexQuery(); indexQuery.setId("2333343434"); indexQuery.setSource(documentSource); indexQuery.setIndexName(INDEX_NAME); indexQuery.setType(TYPE_NAME); // when elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", indexQuery.getId())) .withIndices(INDEX_NAME) .withTypes(TYPE_NAME) .build(); // then Page
                                                                                                                                    
                                                                                                                                      page = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class, new SearchResultMapper() { @Override public 
                                                                                                                                     
                                                                                                                                       AggregatedPage
                                                                                                                                      
                                                                                                                                        mapResults(SearchResponse response, Class
                                                                                                                                       
                                                                                                                                         clazz, Pageable pageable) { List
                                                                                                                                        
                                                                                                                                          values = new ArrayList
                                                                                                                                         
                                                                                                                                          (); for (SearchHit searchHit : response.getHits()) { SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setId(searchHit.getId()); sampleEntity.setMessage((String) searchHit.getSource().get("message")); values.add(sampleEntity); } return new AggregatedPageImpl
                                                                                                                                          
                                                                                                                                           ((List
                                                                                                                                           
                                                                                                                                            ) values); } }); assertThat(page, is(notNullValue())); assertThat(page.getContent().size(), is(1)); assertThat(page.getContent().get(0).getId(), is(indexQuery.getId())); } @Test(expected = ElasticsearchException.class) public void shouldThrowElasticsearchExceptionWhenNoDocumentSpecified() { // given IndexQuery indexQuery = new IndexQuery(); indexQuery.setId("2333343434"); indexQuery.setIndexName(INDEX_NAME); indexQuery.setType(TYPE_NAME); //when elasticsearchTemplate.index(indexQuery); } @Test public void shouldReturnIds() { //given List
                                                                                                                                            
                                                                                                                                              entities = createSampleEntitiesWithMessage("Test message", 30); // when elasticsearchTemplate.bulkIndex(entities); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(termQuery("message", "message")) .withIndices(INDEX_NAME) .withTypes(TYPE_NAME) .withPageable(new PageRequest(0, 100)) .build(); // then List
                                                                                                                                             
                                                                                                                                               ids = elasticsearchTemplate.queryForIds(searchQuery); assertThat(ids, is(notNullValue())); assertThat(ids.size(), is(30)); } @Test public void shouldReturnDocumentAboveMinimalScoreGivenQuery() { // given List
                                                                                                                                              
                                                                                                                                                indexQueries = new ArrayList
                                                                                                                                               
                                                                                                                                                (); indexQueries.add(buildIndex(SampleEntity.builder().id("1").message("ab").build())); indexQueries.add(buildIndex(SampleEntity.builder().id("2").message("bc").build())); indexQueries.add(buildIndex(SampleEntity.builder().id("3").message("ac").build())); elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); // when SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(boolQuery().must(wildcardQuery("message", "*a*")).should(wildcardQuery("message", "*b*"))) .withIndices(INDEX_NAME) .withTypes(TYPE_NAME) .withMinScore(0.5F) .build(); Page
                                                                                                                                                
                                                                                                                                                  page = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(page.getTotalElements(), is(1L)); assertThat(page.getContent().get(0).getMessage(), is("ab")); } @Test public void shouldDoIndexWithoutId() { // given // document SampleEntity sampleEntity = new SampleEntity(); sampleEntity.setMessage("some message"); sampleEntity.setVersion(System.currentTimeMillis()); IndexQuery indexQuery = new IndexQuery(); indexQuery.setObject(sampleEntity); // when String documentId = elasticsearchTemplate.index(indexQuery); // then assertThat(sampleEntity.getId(), is(equalTo(documentId))); GetQuery getQuery = new GetQuery(); getQuery.setId(documentId); SampleEntity result = elasticsearchTemplate.queryForObject(getQuery, SampleEntity.class); assertThat(result.getId(), is(equalTo(documentId))); } @Test public void shouldDoBulkIndexWithoutId() { // given List
                                                                                                                                                 
                                                                                                                                                   indexQueries = new ArrayList
                                                                                                                                                  
                                                                                                                                                   (); // first document SampleEntity sampleEntity1 = new SampleEntity(); sampleEntity1.setMessage("some message"); sampleEntity1.setVersion(System.currentTimeMillis()); IndexQuery indexQuery1 = new IndexQuery(); indexQuery1.setObject(sampleEntity1); indexQueries.add(indexQuery1); // second document SampleEntity sampleEntity2 = new SampleEntity(); sampleEntity2.setMessage("some message"); sampleEntity2.setVersion(System.currentTimeMillis()); IndexQuery indexQuery2 = new IndexQuery(); indexQuery2.setObject(sampleEntity2); indexQueries.add(indexQuery2); // when elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(SampleEntity.class); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).build(); Page
                                                                                                                                                   
                                                                                                                                                     sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); assertThat(sampleEntities.getTotalElements(), is(equalTo(2L))); assertThat(sampleEntities.getContent().get(0).getId(), is(notNullValue())); assertThat(sampleEntities.getContent().get(1).getId(), is(notNullValue())); } @Test public void shouldIndexMapWithIndexNameAndTypeAtRuntime() { //given Map
                                                                                                                                                    
                                                                                                                                                      person1 = new HashMap
                                                                                                                                                     
                                                                                                                                                      (); person1.put("userId", "1"); person1.put("email", "smhdiu@gmail.com"); person1.put("title", "Mr"); person1.put("firstName", "Mohsin"); person1.put("lastName", "Husen"); Map
                                                                                                                                                      
                                                                                                                                                        person2 = new HashMap
                                                                                                                                                       
                                                                                                                                                        (); person2.put("userId", "2"); person2.put("email", "akonczak@gmail.com"); person2.put("title", "Mr"); person2.put("firstName", "Artur"); person2.put("lastName", "Konczak"); IndexQuery indexQuery1 = new IndexQuery(); indexQuery1.setId("1"); indexQuery1.setObject(person1); indexQuery1.setIndexName(INDEX_NAME); indexQuery1.setType(TYPE_NAME); IndexQuery indexQuery2 = new IndexQuery(); indexQuery2.setId("2"); indexQuery2.setObject(person2); indexQuery2.setIndexName(INDEX_NAME); indexQuery2.setType(TYPE_NAME); List
                                                                                                                                                        
                                                                                                                                                          indexQueries = new ArrayList
                                                                                                                                                         
                                                                                                                                                          (); indexQueries.add(indexQuery1); indexQueries.add(indexQuery2); //when elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.refresh(INDEX_NAME); // then SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices(INDEX_NAME) .withTypes(TYPE_NAME).withQuery(matchAllQuery()).build(); Page
                                                                                                                                                           sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, Map.class, new SearchResultMapper() { @Override public 
                                                                                                                                                           
                                                                                                                                                             AggregatedPage
                                                                                                                                                            
                                                                                                                                                              mapResults(SearchResponse response, Class
                                                                                                                                                             
                                                                                                                                                               clazz, Pageable pageable) { List
                                                                                                                                                               chunk = new ArrayList(); for (SearchHit searchHit : response.getHits()) { if (response.getHits().getHits().length <= 0) { return null; } Map
                                                                                                                                                                
                                                                                                                                                                  person = new HashMap
                                                                                                                                                                 
                                                                                                                                                                  (); person.put("userId", searchHit.getSource().get("userId")); person.put("email", searchHit.getSource().get("email")); person.put("title", searchHit.getSource().get("title")); person.put("firstName", searchHit.getSource().get("firstName")); person.put("lastName", searchHit.getSource().get("lastName")); chunk.add(person); } if (chunk.size() > 0) { return new AggregatedPageImpl
                                                                                                                                                                  
                                                                                                                                                                   ((List
                                                                                                                                                                   
                                                                                                                                                                    ) chunk); } return null; } }); assertThat(sampleEntities.getTotalElements(), is(equalTo(2L))); assertThat(sampleEntities.getContent().get(0).get("userId"), is(person1.get("userId"))); assertThat(sampleEntities.getContent().get(1).get("userId"), is(person2.get("userId"))); } @Test public void shouldIndexSampleEntityWithIndexAndTypeAtRuntime() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId) .message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = new IndexQueryBuilder().withId(documentId) .withIndexName(INDEX_NAME).withType(TYPE_NAME) .withObject(sampleEntity).build(); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(INDEX_NAME); SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices(INDEX_NAME) .withTypes(TYPE_NAME).withQuery(matchAllQuery()).build(); // when Page
                                                                                                                                                                    
                                                                                                                                                                      sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); // then assertThat(sampleEntities, is(notNullValue())); assertThat(sampleEntities.getTotalElements(), greaterThanOrEqualTo(1L)); } /* DATAES-106 */ @Test public void shouldReturnCountForGivenCriteriaQueryWithGivenIndexUsingCriteriaQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices("test-index"); // when long count = elasticsearchTemplate.count(criteriaQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-67 */ @Test public void shouldReturnCountForGivenSearchQueryWithGivenIndexUsingSearchQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .withIndices("test-index") .build(); // when long count = elasticsearchTemplate.count(searchQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-106 */ @Test public void shouldReturnCountForGivenCriteriaQueryWithGivenIndexAndTypeUsingCriteriaQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices("test-index"); criteriaQuery.addTypes("test-type"); // when long count = elasticsearchTemplate.count(criteriaQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-67 */ @Test public void shouldReturnCountForGivenSearchQueryWithGivenIndexAndTypeUsingSearchQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .withIndices("test-index") .withTypes("test-type") .build(); // when long count = elasticsearchTemplate.count(searchQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-106 */ @Test public void shouldReturnCountForGivenCriteriaQueryWithGivenMultiIndices() { // given cleanUpIndices(); String documentId1 = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId1).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery1 = new IndexQueryBuilder().withId(sampleEntity1.getId()) .withIndexName("test-index-1") .withObject(sampleEntity1) .build(); String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery2 = new IndexQueryBuilder().withId(sampleEntity2.getId()) .withIndexName("test-index-2") .withObject(sampleEntity2) .build(); elasticsearchTemplate.bulkIndex(Arrays.asList(indexQuery1, indexQuery2)); elasticsearchTemplate.refresh("test-index-1"); elasticsearchTemplate.refresh("test-index-2"); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices("test-index-1", "test-index-2"); // when long count = elasticsearchTemplate.count(criteriaQuery); // then assertThat(count, is(equalTo(2L))); } /* DATAES-67 */ @Test public void shouldReturnCountForGivenSearchQueryWithGivenMultiIndices() { // given cleanUpIndices(); String documentId1 = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId1).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery1 = new IndexQueryBuilder().withId(sampleEntity1.getId()) .withIndexName("test-index-1") .withObject(sampleEntity1) .build(); String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery2 = new IndexQueryBuilder().withId(sampleEntity2.getId()) .withIndexName("test-index-2") .withObject(sampleEntity2) .build(); elasticsearchTemplate.bulkIndex(Arrays.asList(indexQuery1, indexQuery2)); elasticsearchTemplate.refresh("test-index-1"); elasticsearchTemplate.refresh("test-index-2"); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .withIndices("test-index-1", "test-index-2") .build(); // when long count = elasticsearchTemplate.count(searchQuery); // then assertThat(count, is(equalTo(2L))); } private void cleanUpIndices() { elasticsearchTemplate.deleteIndex("test-index-1"); elasticsearchTemplate.deleteIndex("test-index-2"); elasticsearchTemplate.createIndex("test-index-1"); elasticsearchTemplate.createIndex("test-index-2"); elasticsearchTemplate.refresh("test-index-1"); elasticsearchTemplate.refresh("test-index-2"); } /* DATAES-71 */ @Test public void shouldCreatedIndexWithSpecifiedIndexName() { // given elasticsearchTemplate.deleteIndex("test-index"); // when elasticsearchTemplate.createIndex("test-index"); // then assertThat(elasticsearchTemplate.indexExists("test-index"), is(true)); } /* DATAES-72 */ @Test public void shouldDeleteIndexForSpecifiedIndexName() { // given elasticsearchTemplate.createIndex(SampleEntity.class); elasticsearchTemplate.refresh(SampleEntity.class); // when elasticsearchTemplate.deleteIndex("test-index"); // then assertThat(elasticsearchTemplate.indexExists("test-index"), is(false)); } /* DATAES-106 */ @Test public void shouldReturnCountForGivenCriteriaQueryWithGivenIndexNameForSpecificIndex() { // given cleanUpIndices(); String documentId1 = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId1).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery1 = new IndexQueryBuilder().withId(sampleEntity1.getId()) .withIndexName("test-index-1") .withObject(sampleEntity1) .build(); String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery2 = new IndexQueryBuilder().withId(sampleEntity2.getId()) .withIndexName("test-index-2") .withObject(sampleEntity2) .build(); elasticsearchTemplate.bulkIndex(Arrays.asList(indexQuery1, indexQuery2)); elasticsearchTemplate.refresh("test-index-1"); elasticsearchTemplate.refresh("test-index-2"); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); criteriaQuery.addIndices("test-index-1"); // when long count = elasticsearchTemplate.count(criteriaQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-67 */ @Test public void shouldReturnCountForGivenSearchQueryWithGivenIndexNameForSpecificIndex() { // given cleanUpIndices(); String documentId1 = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId1).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery1 = new IndexQueryBuilder().withId(sampleEntity1.getId()) .withIndexName("test-index-1") .withObject(sampleEntity1) .build(); String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery2 = new IndexQueryBuilder().withId(sampleEntity2.getId()) .withIndexName("test-index-2") .withObject(sampleEntity2) .build(); elasticsearchTemplate.bulkIndex(Arrays.asList(indexQuery1, indexQuery2)); elasticsearchTemplate.refresh("test-index-1"); elasticsearchTemplate.refresh("test-index-2"); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .withIndices("test-index-1") .build(); // when long count = elasticsearchTemplate.count(searchQuery); // then assertThat(count, is(equalTo(1L))); } @Test(expected = IllegalArgumentException.class) public void shouldThrowAnExceptionForGivenCriteriaQueryWhenNoIndexSpecifiedForCountQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); CriteriaQuery criteriaQuery = new CriteriaQuery(new Criteria()); // when long count = elasticsearchTemplate.count(criteriaQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-67 */ @Test(expected = IllegalArgumentException.class) public void shouldThrowAnExceptionForGivenSearchQueryWhenNoIndexSpecifiedForCountQuery() { // given String documentId = randomNumeric(5); SampleEntity sampleEntity = SampleEntity.builder().id(documentId).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery = getIndexQuery(sampleEntity); elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.refresh(SampleEntity.class); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .build(); // when long count = elasticsearchTemplate.count(searchQuery); // then assertThat(count, is(equalTo(1L))); } /* DATAES-71 */ @Test public void shouldCreateIndexWithGivenSettings() { // given String settings = "{\n" + " \"index\": {\n" + " \"number_of_shards\": \"1\",\n" + " \"number_of_replicas\": \"0\",\n" + " \"analysis\": {\n" + " \"analyzer\": {\n" + " \"emailAnalyzer\": {\n" + " \"type\": \"custom\",\n" + " \"tokenizer\": \"uax_url_email\"\n" + " }\n" + " }\n" + " }\n" + " }\n" + "}"; elasticsearchTemplate.deleteIndex("test-index"); // when elasticsearchTemplate.createIndex("test-index", settings); // then Map map = elasticsearchTemplate.getSetting("test-index"); boolean hasAnalyzer = map.containsKey("index.analysis.analyzer.emailAnalyzer.tokenizer"); String emailAnalyzer = (String) map.get("index.analysis.analyzer.emailAnalyzer.tokenizer"); assertThat(elasticsearchTemplate.indexExists("test-index"), is(true)); assertThat(hasAnalyzer, is(true)); assertThat(emailAnalyzer, is("uax_url_email")); } /* DATAES-71 */ @Test public void shouldCreateGivenSettingsForGivenIndex() { //given //delete , create and apply mapping in before method // then Map map = elasticsearchTemplate.getSetting(SampleEntity.class); assertThat(elasticsearchTemplate.indexExists("test-index"), is(true)); assertThat(map.containsKey("index.refresh_interval"), is(true)); assertThat(map.containsKey("index.number_of_replicas"), is(true)); assertThat(map.containsKey("index.number_of_shards"), is(true)); assertThat(map.containsKey("index.store.type"), is(true)); assertThat((String) map.get("index.refresh_interval"), is("-1")); assertThat((String) map.get("index.number_of_replicas"), is("0")); assertThat((String) map.get("index.number_of_shards"), is("1")); assertThat((String) map.get("index.store.type"), is("fs")); } /* DATAES-88 */ @Test public void shouldCreateIndexWithGivenClassAndSettings() { //given String settings = "{\n" + " \"index\": {\n" + " \"number_of_shards\": \"1\",\n" + " \"number_of_replicas\": \"0\",\n" + " \"analysis\": {\n" + " \"analyzer\": {\n" + " \"emailAnalyzer\": {\n" + " \"type\": \"custom\",\n" + " \"tokenizer\": \"uax_url_email\"\n" + " }\n" + " }\n" + " }\n" + " }\n" + "}"; elasticsearchTemplate.deleteIndex(SampleEntity.class); elasticsearchTemplate.createIndex(SampleEntity.class, settings); elasticsearchTemplate.refresh(SampleEntity.class); // then Map map = elasticsearchTemplate.getSetting(SampleEntity.class); assertThat(elasticsearchTemplate.indexExists("test-index"), is(true)); assertThat(map.containsKey("index.number_of_replicas"), is(true)); assertThat(map.containsKey("index.number_of_shards"), is(true)); assertThat((String) map.get("index.number_of_replicas"), is("0")); assertThat((String) map.get("index.number_of_shards"), is("1")); } @Test public void shouldTestResultsAcrossMultipleIndices() { // given String documentId1 = randomNumeric(5); SampleEntity sampleEntity1 = SampleEntity.builder().id(documentId1).message("some message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery1 = new IndexQueryBuilder().withId(sampleEntity1.getId()) .withIndexName("test-index-1") .withObject(sampleEntity1) .build(); String documentId2 = randomNumeric(5); SampleEntity sampleEntity2 = SampleEntity.builder().id(documentId2).message("some test message") .version(System.currentTimeMillis()).build(); IndexQuery indexQuery2 = new IndexQueryBuilder().withId(sampleEntity2.getId()) .withIndexName("test-index-2") .withObject(sampleEntity2) .build(); elasticsearchTemplate.bulkIndex(Arrays.asList(indexQuery1, indexQuery2)); elasticsearchTemplate.refresh("test-index-1"); elasticsearchTemplate.refresh("test-index-2"); SearchQuery searchQuery = new NativeSearchQueryBuilder() .withQuery(matchAllQuery()) .withIndices("test-index-1", "test-index-2") .build(); // when List
                                                                                                                                                                     
                                                                                                                                                                       sampleEntities = elasticsearchTemplate.queryForList(searchQuery, SampleEntity.class); // then assertThat(sampleEntities.size(), is(equalTo(2))); } @Test /** * This is basically a demonstration to show composing entities out of heterogeneous indexes. */ public void shouldComposeObjectsReturnedFromHeterogeneousIndexes() { // Given HetroEntity1 entity1 = new HetroEntity1(randomNumeric(3), "aFirstName"); HetroEntity2 entity2 = new HetroEntity2(randomNumeric(4), "aLastName"); IndexQuery idxQuery1 = new IndexQueryBuilder().withIndexName(INDEX_1_NAME).withId(entity1.getId()).withObject(entity1).build(); IndexQuery idxQuery2 = new IndexQueryBuilder().withIndexName(INDEX_2_NAME).withId(entity2.getId()).withObject(entity2).build(); elasticsearchTemplate.bulkIndex(Arrays.asList(idxQuery1, idxQuery2)); elasticsearchTemplate.refresh(INDEX_1_NAME); elasticsearchTemplate.refresh(INDEX_2_NAME); // When SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withTypes("hetro").withIndices(INDEX_1_NAME, INDEX_2_NAME).build(); Page
                                                                                                                                                                      
                                                                                                                                                                        page = elasticsearchTemplate.queryForPage(searchQuery, ResultAggregator.class, new SearchResultMapper() { @Override public 
                                                                                                                                                                       
                                                                                                                                                                         AggregatedPage
                                                                                                                                                                        
                                                                                                                                                                          mapResults(SearchResponse response, Class
                                                                                                                                                                         
                                                                                                                                                                           clazz, Pageable pageable) { List
                                                                                                                                                                          
                                                                                                                                                                            values = new ArrayList
                                                                                                                                                                           
                                                                                                                                                                            (); for (SearchHit searchHit : response.getHits()) { String id = String.valueOf(searchHit.getSource().get("id")); String firstName = StringUtils.isNotEmpty((String) searchHit.getSource().get("firstName")) ? (String) searchHit.getSource().get("firstName") : ""; String lastName = StringUtils.isNotEmpty((String) searchHit.getSource().get("lastName")) ? (String) searchHit.getSource().get("lastName") : ""; values.add(new ResultAggregator(id, firstName, lastName)); } return new AggregatedPageImpl
                                                                                                                                                                            
                                                                                                                                                                             ((List
                                                                                                                                                                             
                                                                                                                                                                              ) values); } }); assertThat(page.getTotalElements(), is(2l)); } @Test public void shouldCreateIndexUsingServerDefaultConfiguration() { //given //when boolean created = elasticsearchTemplate.createIndex(UseServerConfigurationEntity.class); //then assertThat(created, is(true)); final Map setting = elasticsearchTemplate.getSetting(UseServerConfigurationEntity.class); assertThat(setting.get("index.number_of_shards"), Matchers.
                                                                                                                                                                              is("5")); assertThat(setting.get("index.number_of_replicas"), Matchers.is("1")); } @Test public void shouldReadFileFromClasspathRetainingNewlines() { // given String settingsFile = "/settings/test-settings.yml"; // when String content = ElasticsearchTemplate.readFileFromClasspath(settingsFile); // then assertThat(content, is("index:\n" + " number_of_shards: 1\n" + " number_of_replicas: 0\n" + " analysis:\n" + " analyzer:\n" + " emailAnalyzer:\n" + " type: custom\n" + " tokenizer: uax_url_email\n")); } private IndexQuery getIndexQuery(SampleEntity sampleEntity) { return new IndexQueryBuilder().withId(sampleEntity.getId()).withObject(sampleEntity).build(); } private List
                                                                                                                                                                                
                                                                                                                                                                                  getIndexQueries(List
                                                                                                                                                                                 
                                                                                                                                                                                   sampleEntities) { List
                                                                                                                                                                                  
                                                                                                                                                                                    indexQueries = new ArrayList
                                                                                                                                                                                   
                                                                                                                                                                                    (); for (SampleEntity sampleEntity : sampleEntities) { indexQueries.add(new IndexQueryBuilder().withId(sampleEntity.getId()).withObject(sampleEntity).build()); } return indexQueries; } @Document(indexName = INDEX_2_NAME, replicas = 0, shards = 1) class ResultAggregator { private String id; private String firstName; private String lastName; ResultAggregator(String id, String firstName, String lastName) { this.id = id; this.firstName = firstName; this.lastName = lastName; } }}

转载地址：http://ehmjo.baihongyu.com/

你可能感兴趣的文章
gulp常用命令
查看>>
TCP(Socket基础编程)
查看>>
RowSet的使用
查看>>
表单提交中的input、button、submit的区别
查看>>
每日一记--cookie
查看>>
约瑟夫环
查看>>
S5:桥接模式 Bridge
查看>>
线程池-Executors
查看>>
WPF and Silverlight 学习笔记（十二）：WPF Panel内容模型、Decorator内容模型及其他...
查看>>
FLUSH TABLES WITH READ LOCK 和 LOCK TABLES比较
查看>>
MySQL：创建、修改和删除表
查看>>
Java多线程程序设计详细解析
查看>>
IOS 7 Study - UISegmentedControl
查看>>
八、通用类型系统
查看>>
JQuery的ajaxFileUpload的使用
查看>>
Java分享笔记：使用keySet方法获取Map集合中的元素
查看>>
Java面向对象练习题之人员信息
查看>>
关于Integer类中parseInt()和valueOf()方法的区别以及int和String类性的转换.以及String类valueOf()方法...
查看>>
python之sys模块详解
查看>>
ios 控制器的生命周期
查看>>
喝酒易醉，品茶养心，人生如梦，品茶悟道，何以解忧？唯有杜康！-- 愿君每日到此一游！
当前时间: 2025-01-22 19:47:30   当前IP: 3.137.182.201   联系邮箱:javaeecc@qq.com     Copyright © 2020 - 2022 baihongyu.com 京ICP备2021015314号-2
强烈建议你试试无所不能的CHAT-GPT，快点击我
强烈建议你试试无所不能的CHAT-GPT，快点击我