Package org.elasticsearch.action.get

Examples of org.elasticsearch.action.get.MultiGetRequestBuilder

165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
    List<P> list = new ArrayList<P>();
    if (ids == null || ids.isEmpty()) {
      return list;
    }
    try {
      MultiGetRequestBuilder mgr = client().prepareMultiGet();
      for (String id : ids) {
        MultiGetRequest.Item i = new MultiGetRequest.Item(stripRouting(appid), null, id);
        i.routing(getRouting(appid, id));
        mgr.add(i);
      }

      MultiGetResponse response = mgr.execute().actionGet();
      for (MultiGetItemResponse multiGetItemResponse : response.getResponses()) {
        GetResponse res = multiGetItemResponse.getResponse();
        if (res.isExists() && !res.isSourceEmpty()) {
          list.add((P) Utils.setAnnotatedFields(res.getSource()));
        }
View Full Code Here
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
        if (resolveConflicts) {
            String index = getElasticSearchIndexNameFromDatabase(database);
            // the following debug code is verbose in the hopes of better understanding CBES-13
            MultiGetResponse response = null;
            if(client != null) {
                MultiGetRequestBuilder builder = client.prepareMultiGet();
                if(builder != null) {
                    if(index == null) {
                        logger.debug("index is null");
                    }
                    int added = 0;
                    for (String id : responseMap.keySet()) {
                        String type = typeSelector.getType(index, id);
                        if(documentTypeRoutingFields != null && documentTypeRoutingFields.containsKey(type)) {
                            // if this type requires special routing, we can't find it without the doc body
                            // so we skip this id in the lookup to avoid errors
                            continue;
                        }
                        builder = builder.add(index, type, id);
                        added++;
                    }
                    if(builder != null) {
                        if(added > 0) {
                            ListenableActionFuture<MultiGetResponse> laf = builder.execute();
                            if(laf != null) {
                                response = laf.actionGet();
                            } else {
                                logger.debug("laf was null");
                            }
View Full Code Here
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
    List<P> list = new ArrayList<P>();
    if (ids == null || ids.isEmpty()) {
      return list;
    }
    try {
      MultiGetRequestBuilder mgr = client().prepareMultiGet();
      for (String id : ids) {
        MultiGetRequest.Item i = new MultiGetRequest.Item(stripRouting(appid), null, id);
        i.routing(getRouting(appid, id));
        mgr.add(i);
      }

      MultiGetResponse response = mgr.execute().actionGet();
      for (MultiGetItemResponse multiGetItemResponse : response.getResponses()) {
        GetResponse res = multiGetItemResponse.getResponse();
        if (res.isExists() && !res.isSourceEmpty()) {
          list.add(Utils.setAnnotatedFields(res.getSource()));
        }
View Full Code Here
235
236
237
238
239
240
241
242
243
244
245
246
247
    @Override
    public ListenableActionFuture<MultiGetResponse> execute(Client client, String indexName, OperationContext helper) {
      final TypeMapping typeMapping = helper.findTypeMapping(_type);
      final String type = typeMapping.getTypeAlias();

      MultiGetRequestBuilder builder = client.prepareMultiGet();
      for (Object id : _ids) {
        // ignore indexName as it may be a multi-index
        builder.add(helper.findIndexName(_type), type, typeMapping.toIdString(id));
      }

      return builder.execute();
    }
View Full Code Here
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
    Assert.notNull(indexName, "No index defined for Query");
    Assert.notNull(type, "No type define for Query");
    Assert.notEmpty(searchQuery.getIds(), "No Id define for Query");

    MultiGetRequestBuilder builder = client.prepareMultiGet();

    for (String id : searchQuery.getIds()) {

      MultiGetRequest.Item item = new MultiGetRequest.Item(indexName, type, id);

      if (searchQuery.getRoute() != null) {
        item = item.routing(searchQuery.getRoute());
      }

      if (searchQuery.getFields() != null && !searchQuery.getFields().isEmpty()) {
        item = item.fields(toArray(searchQuery.getFields()));
      }
      builder.add(item);
    }
    return builder.execute().actionGet();
  }
View Full Code Here
39
40
41
42
43
44
45
46
47
48
49
50
51
52
  SearchClient searchClient = new SearchClient(new Settings(), profiling);

  @Test
  public void multi_get() {
    try {
      MultiGetRequestBuilder request = searchClient.prepareMultiGet();
      request.add(new MultiGetRequest.Item(IndexDefinition.RULE.getIndexName(), IndexDefinition.RULE.getIndexType(), "ruleKey")
        .fetchSourceContext(FetchSourceContext.FETCH_SOURCE));
      request.get();

      // expected to fail because elasticsearch is not correctly configured, but that does not matter
      fail();
    } catch (Exception e) {
      assertThat(e).isInstanceOf(IllegalStateException.class);
View Full Code Here
67
68
69
70
71
72
73
74
75
76
77
78
79
80
  public void with_profiling_basic() {
    Profiling profiling = new Profiling(new Settings().setProperty(Profiling.CONFIG_PROFILING_LEVEL, Profiling.Level.BASIC.name()));
    SearchClient searchClient = new SearchClient(new Settings(), profiling);

    try {
      MultiGetRequestBuilder request = searchClient.prepareMultiGet();
      request.add(new MultiGetRequest.Item(IndexDefinition.RULE.getIndexName(), IndexDefinition.RULE.getIndexType(), "ruleKey")
        .fetchSourceContext(FetchSourceContext.FETCH_SOURCE));
      request.get();

      // expected to fail because elasticsearch is not correctly configured, but that does not matter
      fail();
    } catch (Exception e) {
      assertThat(e).isInstanceOf(IllegalStateException.class);
View Full Code Here
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
  @Test
  public void getNodeItems() {
    // Setup
    Way way = OsmDataBuilder.buildSampleWay();

    MultiGetRequestBuilder multiGetRequestBuilderMocked = mock(MultiGetRequestBuilder.class);
    when(clientMocked.prepareMultiGet()).thenReturn(multiGetRequestBuilderMocked);

    ListenableActionFuture<MultiGetResponse> listenableActionFutureMocked = mock(ListenableActionFuture.class);
    when(multiGetRequestBuilderMocked.execute()).thenReturn(listenableActionFutureMocked);
    MultiGetResponse multiGetResponseMocked = mock(MultiGetResponse.class);
    when(listenableActionFutureMocked.actionGet()).thenReturn(multiGetResponseMocked);
    Iterator<MultiGetItemResponse> iteratorMocked = mock(Iterator.class);
    when(multiGetResponseMocked.iterator()).thenReturn(iteratorMocked);
View Full Code Here
349
350
351
352
353
354
355
356
357
358
359
  public void findAll() {
    // Setup
    Node node1 = OsmDataBuilder.buildSampleNode(1);
    Node node2 = OsmDataBuilder.buildSampleNode(2);

    MultiGetRequestBuilder multiGetRequestBuilderMocked = mock(MultiGetRequestBuilder.class);
    doReturn(multiGetRequestBuilderMocked).when(entityDao).buildMultiGetRequest(ESNode.class, 1, 2);
    doReturn(Arrays.asList(node1, node2)).when(entityDao).executeMultiGetRequest(ESNode.class, multiGetRequestBuilderMocked);

    // Action
    List<ESNode> nodes = entityDao.findAll(ESNode.class, 1, 2);
View Full Code Here
385
386
387
388
389
390
391
392
393
394
395
  }

  @Test(expected = DaoException.class)
  public void findAll_withException() {
    // Setup
    MultiGetRequestBuilder multiGetRequestBuilderMocked = mock(MultiGetRequestBuilder.class);
    doReturn(multiGetRequestBuilderMocked).when(entityDao).buildMultiGetRequest(ESNode.class, 1, 2);
    doThrow(new RuntimeException("Simulated Exception")).when(entityDao).executeMultiGetRequest(ESNode.class, multiGetRequestBuilderMocked);

    // Action
    entityDao.findAll(ESNode.class, 1, 2);
View Full Code Here
TOP

Related Classes of org.elasticsearch.action.get.MultiGetRequestBuilder

  • at.molindo.esi4j.core.impl.AbstractIndex$MultiGet
  • com.erudika.para.search.ElasticSearch
  • org.elasticsearch.action.bulk.BulkProcessorTests
  • org.elasticsearch.transport.couchbase.capi.ElasticSearchCAPIBehavior
  • org.openstreetmap.osmosis.plugin.elasticsearch.dao.EntityDao
  • org.openstreetmap.osmosis.plugin.elasticsearch.dao.EntityDaoUTest
  • org.sonar.server.search.request.ProxyMultiGetRequestBuilderTest
  • org.springframework.data.elasticsearch.core.ElasticsearchTemplate
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.