Examples with BulkShardResponse org.opensearch.action.bulk.BulkShardResponse used on opensource projects

Example 1 with BulkShardResponse

use of org.opensearch.action.bulk.BulkShardResponse in project OpenSearch by opensearch-project.

the class OpenSearchIndexLevelReplicationTestCase method executeShardBulkOnPrimary.

private void executeShardBulkOnPrimary(IndexShard primary, BulkShardRequest request, ActionListener<TransportWriteAction.WritePrimaryResult<BulkShardRequest, BulkShardResponse>> listener) {
    for (BulkItemRequest itemRequest : request.items()) {
        if (itemRequest.request() instanceof IndexRequest) {
            ((IndexRequest) itemRequest.request()).process(Version.CURRENT, null, index.getName());
        }
    }
    final PlainActionFuture<Releasable> permitAcquiredFuture = new PlainActionFuture<>();
    primary.acquirePrimaryOperationPermit(permitAcquiredFuture, ThreadPool.Names.SAME, request);
    try (Releasable ignored = permitAcquiredFuture.actionGet()) {
        MappingUpdatePerformer noopMappingUpdater = (update, shardId, listener1) -> {
        };
        TransportShardBulkAction.performOnPrimary(request, primary, null, System::currentTimeMillis, noopMappingUpdater, null, ActionTestUtils.assertNoFailureListener(result -> {
            TransportWriteActionTestHelper.performPostWriteActions(primary, request, ((TransportWriteAction.WritePrimaryResult<BulkShardRequest, BulkShardResponse>) result).location, logger);
            listener.onResponse((TransportWriteAction.WritePrimaryResult<BulkShardRequest, BulkShardResponse>) result);
        }), threadPool, Names.WRITE);
    } catch (Exception e) {
        listener.onFailure(e);
    }
}