org.elasticsearch.action.ListenableActionFuture Java Examples

The following examples show how to use org.elasticsearch.action.ListenableActionFuture. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example #1

Source File: TermsByQueryBenchmark.java From siren-join with GNU Affero General Public License v3.0

6 votes

public void warmFieldData(String parentField, String childField) {
    ListenableActionFuture<SearchResponse> parentSearch = null;
    ListenableActionFuture<SearchResponse> childSearch = null;

    if (parentField != null) {
        parentSearch = client
                .prepareSearch(PARENT_INDEX)
                .setQuery(matchAllQuery()).addAggregation(terms("parentfield").field(parentField)).execute();
    }

    if (childField != null) {
        childSearch = client
                .prepareSearch(CHILD_INDEX)
                .setQuery(matchAllQuery()).addAggregation(terms("childfield").field(childField)).execute();
    }

    if (parentSearch != null) parentSearch.actionGet();
    if (childSearch != null) childSearch.actionGet();
}

Example #2

Source File: FilterJoinBenchmark.java From siren-join with GNU Affero General Public License v3.0

6 votes

public void warmFieldData(String parentField, String childField) {
    ListenableActionFuture<SearchResponse> parentSearch = null;
    ListenableActionFuture<SearchResponse> childSearch = null;

    if (parentField != null) {
        parentSearch = client
                .prepareSearch(PARENT_INDEX)
                .setQuery(matchAllQuery()).addAggregation(terms("parentfield").field(parentField)).execute();
    }

    if (childField != null) {
        childSearch = client
                .prepareSearch(CHILD_INDEX)
                .setQuery(matchAllQuery()).addAggregation(terms("childfield").field(childField)).execute();
    }

    if (parentSearch != null) parentSearch.actionGet();
    if (childSearch != null) childSearch.actionGet();
}

Example #3

Source File: ElasticsearchEmitterTest.java From amazon-kinesis-connectors with Apache License 2.0

6 votes

@Before
public void setUp() {
    // object under test
    Properties props = new Properties();
    AWSCredentialsProvider creds = createMock(AWSCredentialsProvider.class);
    configuration = new KinesisConnectorConfiguration(props, creds);

    emitter = new ElasticsearchEmitter(configuration);

    buffer = createMock(UnmodifiableBuffer.class);
    mockBulkBuilder = createMock(BulkRequestBuilder.class);
    mockIndexBuilder = createMock(IndexRequestBuilder.class);
    mockFuture = createMock(ListenableActionFuture.class);
    mockBulkResponse = createMock(BulkResponse.class);

    // overwrite the elasticseach client with a mock
    elasticsearchClientMock = createMock(TransportClient.class);

    setField(emitter, "elasticsearchClient", elasticsearchClientMock);
    // overwrite the default backoff time with 0 seconds to speed up tests
    setField(emitter, "BACKOFF_PERIOD", 0);
}

Example #4

Source File: SQLTransportExecutor.java From crate with Apache License 2.0

5 votes

public static ListenableActionFuture<SQLResponse> execute(String stmt,
                                                          @Nullable Object[] args,
                                                          Session session) {
    PlainListenableActionFuture<SQLResponse> actionFuture = PlainListenableActionFuture.newListenableFuture();
    execute(stmt, args, actionFuture, session);
    return actionFuture;
}

Example #5

Source File: TestElasticSearchTransportClient.java From ingestion with Apache License 2.0

5 votes

@Test(expected = EventDeliveryException.class)
public void shouldThrowExceptionOnExecuteFailed() throws Exception {
  ListenableActionFuture<BulkResponse> action =
      (ListenableActionFuture<BulkResponse>) mock(ListenableActionFuture.class);
  BulkResponse response = mock(BulkResponse.class);
  when(bulkRequestBuilder.execute()).thenReturn(action);
  when(action.actionGet()).thenReturn(response);
  when(response.hasFailures()).thenReturn(true);

  fixture.addEvent(event, nameBuilder, "bar_type", 10);
  fixture.execute();
}

Example #6

Source File: TestElasticSearchTransportClient.java From ingestion with Apache License 2.0

5 votes

@Test
public void shouldExecuteBulkRequestBuilder() throws Exception {
  ListenableActionFuture<BulkResponse> action =
      (ListenableActionFuture<BulkResponse>) mock(ListenableActionFuture.class);
  BulkResponse response = mock(BulkResponse.class);
  when(bulkRequestBuilder.execute()).thenReturn(action);
  when(action.actionGet()).thenReturn(response);
  when(response.hasFailures()).thenReturn(false);

  fixture.addEvent(event, nameBuilder, "bar_type", 10);
  fixture.execute();
  verify(bulkRequestBuilder).execute();
}

Example #7

Source File: InternalEsClient.java From io with Apache License 2.0

5 votes

/**
 * Mapping定義を更新する.
 * @param index インデックス名
 * @param type タイプ名
 * @param mappings マッピング情報
 * @return 非同期応答
 */
public ListenableActionFuture<PutMappingResponse> putMapping(String index,
        String type,
        Map<String, Object> mappings) {
    PutMappingRequestBuilder builder = new PutMappingRequestBuilder(esTransportClient.admin().indices())
            .setIndices(index)
            .setType(type)
            .setSource(mappings);
    return builder.execute();
}

Example #8

Source File: InternalEsClient.java From io with Apache License 2.0

5 votes

/**
 * Mapping定義を更新する.
 * @param index インデックス名
 * @param type タイプ名
 * @param mappings マッピング情報
 * @return 非同期応答
 */
public ListenableActionFuture<PutMappingResponse> putMapping(String index,
        String type,
        Map<String, Object> mappings) {
    PutMappingRequestBuilder builder = new PutMappingRequestBuilder(esTransportClient.admin().indices(), PutMappingAction.INSTANCE)
            .setIndices(index)
            .setType(type)
            .setSource(mappings);
    return builder.execute();
}

Example #9