Search in sources :

Example 26 with Search

use of io.searchbox.core.Search in project herd by FINRAOS.

the class JestClientHelperTest method testSearchExecute.

@Test
public void testSearchExecute() throws Exception {
    // Mock
    Search search = mock(Search.class);
    SearchResult searchResult = mock(SearchResult.class);
    JestClient jestClient = mock(JestClient.class);
    when(jestClientFactory.getJestClient()).thenReturn(jestClient);
    when(jestClient.execute(search)).thenReturn(searchResult);
    // Test
    SearchResult result = jestClientHelper.searchExecute(search);
    // Validate
    assertThat(result, is(not(nullValue())));
    // Verify
    verify(jestClientFactory).getJestClient();
    verify(jestClient).execute(search);
    verifyNoMoreInteractions(createdMocks.toArray());
}
Also used : Search(io.searchbox.core.Search) SearchResult(io.searchbox.core.SearchResult) JestClient(io.searchbox.client.JestClient) Test(org.junit.Test)

Example 27 with Search

use of io.searchbox.core.Search in project herd by FINRAOS.

the class BusinessObjectDefinitionIndexSearchDaoImpl method searchResponseIntoFacetInformation.

private List<TagTypeIndexSearchResponseDto> searchResponseIntoFacetInformation(final SearchRequestBuilder searchRequestBuilder, String indexName) {
    // Retrieve the search response
    final Search.Builder searchBuilder = new Search.Builder(searchRequestBuilder.toString()).addIndex(indexName);
    SearchResult searchResult = jestClientHelper.searchExecute(searchBuilder.build());
    return elasticsearchHelper.getNestedTagTagIndexSearchResponseDto(searchResult);
}
Also used : Search(io.searchbox.core.Search) SearchResult(io.searchbox.core.SearchResult)

Example 28 with Search

use of io.searchbox.core.Search in project ff4j by ff4j.

the class ElasticQueryBuilder method queryGetEventById.

public Search queryGetEventById(String uuid) {
    SearchSourceBuilder source = new SearchSourceBuilder();
    source.query(QueryBuilders.matchQuery("uuid", uuid));
    return new Search.Builder(source.toString()).addIndex(connection.getIndexName()).addType(ElasticConstants.TYPE_EVENT).build();
}
Also used : Search(io.searchbox.core.Search) SearchSourceBuilder(org.elasticsearch.search.builder.SearchSourceBuilder)

Example 29 with Search

use of io.searchbox.core.Search in project ff4j by ff4j.

the class ElasticQueryBuilder method getFeatureTechIdByGroup.

@SuppressWarnings({ "rawtypes" })
public Set<String> getFeatureTechIdByGroup(String groupName) {
    SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
    searchSourceBuilder.query(QueryBuilders.matchQuery("group", groupName));
    Search search = // 
    new Search.Builder(searchSourceBuilder.toString()).addIndex(// 
    connection.getIndexName()).addType(// 
    ElasticConstants.TYPE_FEATURE).build();
    SearchResult result = connection.search(search, true);
    Set<String> metadatas = new HashSet<String>();
    if (null != result && result.isSucceeded()) {
        List<Hit<Map, Void>> features = result.getHits(Map.class);
        for (Hit<Map, Void> hit : features) {
            metadatas.add(hit.source.get(JestResult.ES_METADATA_ID).toString());
        }
    }
    return metadatas;
}
Also used : Hit(io.searchbox.core.SearchResult.Hit) Search(io.searchbox.core.Search) SearchResult(io.searchbox.core.SearchResult) Map(java.util.Map) SearchSourceBuilder(org.elasticsearch.search.builder.SearchSourceBuilder) HashSet(java.util.HashSet)

Example 30 with Search

use of io.searchbox.core.Search in project ff4j by ff4j.

the class ElasticQueryBuilder method queryGetFeatureById.

/**
 * Syntaxic sugar to have query on feature.
 *
 * @param uid
 *            target feature uid
 * @return query for JEST
 */
public Search queryGetFeatureById(String uid) {
    SearchSourceBuilder source = new SearchSourceBuilder();
    source.query(QueryBuilders.matchQuery("uid", uid));
    return new Search.Builder(source.toString()).addIndex(connection.getIndexName()).addType(ElasticConstants.TYPE_FEATURE).build();
}
Also used : Search(io.searchbox.core.Search) SearchSourceBuilder(org.elasticsearch.search.builder.SearchSourceBuilder)

Aggregations

Search (io.searchbox.core.Search)45 SearchResult (io.searchbox.core.SearchResult)24 SearchSourceBuilder (org.elasticsearch.search.builder.SearchSourceBuilder)23 Test (org.junit.Test)15 Hit (io.searchbox.core.SearchResult.Hit)12 JestClient (io.searchbox.client.JestClient)11 SearchSourceBuilder (org.graylog.shaded.elasticsearch6.org.elasticsearch.search.builder.SearchSourceBuilder)10 QueryBuilder (org.elasticsearch.index.query.QueryBuilder)9 JsonObject (com.google.gson.JsonObject)8 Map (java.util.Map)8 IOException (java.io.IOException)7 List (java.util.List)7 Set (java.util.Set)6 Collectors (java.util.stream.Collectors)6 Inject (javax.inject.Inject)6 IndexMapping (org.graylog2.indexer.IndexMapping)6 Optional (java.util.Optional)5 BoolQueryBuilder (org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.BoolQueryBuilder)5 QueryBuilder (org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryBuilder)5 QueryBuilders (org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryBuilders)5