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

Example 1 with BulkItemRequest

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

the class IndexingPressureServiceTests method testCoordinatingOperationForShardIndexingPressure.

public void testCoordinatingOperationForShardIndexingPressure() {
    IndexingPressureService service = new IndexingPressureService(settings, clusterService);
    Index index = new Index("IndexName", "UUID");
    ShardId shardId = new ShardId(index, 0);
    BulkItemRequest[] items = new BulkItemRequest[1];
    DocWriteRequest<IndexRequest> writeRequest = new IndexRequest("index").id("id").source(Requests.INDEX_CONTENT_TYPE, "foo", "bar");
    items[0] = new BulkItemRequest(0, writeRequest);
    BulkShardRequest bulkShardRequest = new BulkShardRequest(shardId, WriteRequest.RefreshPolicy.NONE, items);
    Releasable releasable = service.markCoordinatingOperationStarted(shardId, bulkShardRequest::ramBytesUsed, false);
    IndexingPressurePerShardStats shardStats = service.shardStats(CommonStatsFlags.ALL).getIndexingPressureShardStats(shardId);
    assertEquals(bulkShardRequest.ramBytesUsed(), shardStats.getCurrentCoordinatingBytes());
    releasable.close();
}

Example 2 with BulkItemRequest

use of org.opensearch.action.bulk.BulkItemRequest 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);
    }
}

Example 3 with BulkItemRequest

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

the class OpenSearchIndexLevelReplicationTestCase method executeReplicationRequestOnPrimary.

private <Request extends ReplicatedWriteRequest & DocWriteRequest> BulkShardRequest executeReplicationRequestOnPrimary(IndexShard primary, Request request) throws Exception {
    final BulkShardRequest bulkShardRequest = new BulkShardRequest(shardId, request.getRefreshPolicy(), new BulkItemRequest[] { new BulkItemRequest(0, request) });
    final PlainActionFuture<BulkShardRequest> res = new PlainActionFuture<>();
    executeShardBulkOnPrimary(primary, bulkShardRequest, ActionListener.map(res, TransportReplicationAction.PrimaryResult::replicaRequest));
    return res.get();
}