Examples with TransportService org.opensearch.transport.TransportService used on opensource projects

Example 96 with TransportService

use of org.opensearch.transport.TransportService in project OpenSearch by opensearch-project.

the class TransportAddVotingConfigExclusionsActionTests method setupForTest.

@Before
public void setupForTest() {
    final MockTransport transport = new MockTransport();
    transportService = transport.createTransportService(Settings.EMPTY, threadPool, TransportService.NOOP_TRANSPORT_INTERCEPTOR, boundTransportAddress -> localNode, null, emptySet());
    final Settings.Builder nodeSettingsBuilder = Settings.builder();
    if (randomBoolean()) {
        staticMaximum = between(5, 15);
        nodeSettingsBuilder.put(MAXIMUM_VOTING_CONFIG_EXCLUSIONS_SETTING.getKey(), staticMaximum);
    } else {
        staticMaximum = MAXIMUM_VOTING_CONFIG_EXCLUSIONS_SETTING.get(Settings.EMPTY);
    }
    final Settings nodeSettings = nodeSettingsBuilder.build();
    clusterSettings = new ClusterSettings(nodeSettings, ClusterSettings.BUILT_IN_CLUSTER_SETTINGS);
    new TransportAddVotingConfigExclusionsAction(nodeSettings, clusterSettings, transportService, clusterService, threadPool, new ActionFilters(emptySet()), new IndexNameExpressionResolver(new ThreadContext(Settings.EMPTY)));
    // registers action
    transportService.start();
    transportService.acceptIncomingRequests();
    final VotingConfiguration allNodesConfig = VotingConfiguration.of(localNode, otherNode1, otherNode2);
    setState(clusterService, builder(new ClusterName("cluster")).nodes(new Builder().add(localNode).add(otherNode1).add(otherNode2).add(otherDataNode).localNodeId(localNode.getId()).masterNodeId(localNode.getId())).metadata(Metadata.builder().coordinationMetadata(CoordinationMetadata.builder().lastAcceptedConfiguration(allNodesConfig).lastCommittedConfiguration(allNodesConfig).build())));
    clusterStateObserver = new ClusterStateObserver(clusterService, null, logger, threadPool.getThreadContext());
}

Example 97 with TransportService

use of org.opensearch.transport.TransportService in project OpenSearch by opensearch-project.

the class TransportBulkActionTests method setUp.

@Before
public void setUp() throws Exception {
    super.setUp();
    threadPool = new TestThreadPool(getClass().getName());
    DiscoveryNode discoveryNode = new DiscoveryNode("node", OpenSearchTestCase.buildNewFakeTransportAddress(), emptyMap(), DiscoveryNodeRole.BUILT_IN_ROLES, VersionUtils.randomCompatibleVersion(random(), Version.CURRENT));
    clusterService = createClusterService(threadPool, discoveryNode);
    CapturingTransport capturingTransport = new CapturingTransport();
    transportService = capturingTransport.createTransportService(clusterService.getSettings(), threadPool, TransportService.NOOP_TRANSPORT_INTERCEPTOR, boundAddress -> clusterService.localNode(), null, Collections.emptySet());
    transportService.start();
    transportService.acceptIncomingRequests();
    bulkAction = new TestTransportBulkAction();
}

Example 98 with TransportService

use of org.opensearch.transport.TransportService in project OpenSearch by opensearch-project.

the class TransportBulkActionIngestTests method validatePipelineWithBulkUpsert.

private void validatePipelineWithBulkUpsert(@Nullable String indexRequestIndexName, String updateRequestIndexName) throws Exception {
    Exception exception = new Exception("fake exception");
    BulkRequest bulkRequest = new BulkRequest();
    IndexRequest indexRequest1 = new IndexRequest(indexRequestIndexName).id("id1").source(emptyMap());
    IndexRequest indexRequest2 = new IndexRequest(indexRequestIndexName).id("id2").source(emptyMap());
    IndexRequest indexRequest3 = new IndexRequest(indexRequestIndexName).id("id3").source(emptyMap());
    UpdateRequest upsertRequest = new UpdateRequest(updateRequestIndexName, "id1").upsert(indexRequest1).script(mockScript("1"));
    UpdateRequest docAsUpsertRequest = new UpdateRequest(updateRequestIndexName, "id2").doc(indexRequest2).docAsUpsert(true);
    // this test only covers the mechanics that scripted bulk upserts will execute a default pipeline. However, in practice scripted
    // bulk upserts with a default pipeline are a bit surprising since the script executes AFTER the pipeline.
    UpdateRequest scriptedUpsert = new UpdateRequest(updateRequestIndexName, "id2").upsert(indexRequest3).script(mockScript("1")).scriptedUpsert(true);
    bulkRequest.add(upsertRequest).add(docAsUpsertRequest).add(scriptedUpsert);
    AtomicBoolean responseCalled = new AtomicBoolean(false);
    AtomicBoolean failureCalled = new AtomicBoolean(false);
    assertNull(indexRequest1.getPipeline());
    assertNull(indexRequest2.getPipeline());
    assertNull(indexRequest3.getPipeline());
    action.execute(null, bulkRequest, ActionListener.wrap(response -> {
        BulkItemResponse itemResponse = response.iterator().next();
        assertThat(itemResponse.getFailure().getMessage(), containsString("fake exception"));
        responseCalled.set(true);
    }, e -> {
        assertThat(e, sameInstance(exception));
        failureCalled.set(true);
    }));
    // check failure works, and passes through to the listener
    // haven't executed yet
    assertFalse(action.isExecuted);
    assertFalse(responseCalled.get());
    assertFalse(failureCalled.get());
    verify(ingestService).executeBulkRequest(eq(bulkRequest.numberOfActions()), bulkDocsItr.capture(), failureHandler.capture(), completionHandler.capture(), any(), eq(Names.WRITE));
    assertEquals(indexRequest1.getPipeline(), "default_pipeline");
    assertEquals(indexRequest2.getPipeline(), "default_pipeline");
    assertEquals(indexRequest3.getPipeline(), "default_pipeline");
    completionHandler.getValue().accept(null, exception);
    assertTrue(failureCalled.get());
    // now check success of the transport bulk action
    // this is done by the real pipeline execution service when processing
    indexRequest1.setPipeline(IngestService.NOOP_PIPELINE_NAME);
    // this is done by the real pipeline execution service when processing
    indexRequest2.setPipeline(IngestService.NOOP_PIPELINE_NAME);
    // this is done by the real pipeline execution service when processing
    indexRequest3.setPipeline(IngestService.NOOP_PIPELINE_NAME);
    completionHandler.getValue().accept(DUMMY_WRITE_THREAD, null);
    assertTrue(action.isExecuted);
    // listener would only be called by real index action, not our mocked one
    assertFalse(responseCalled.get());
    verifyNoInteractions(transportService);
}

Example 99 with TransportService

use of org.opensearch.transport.TransportService in project OpenSearch by opensearch-project.

the class TransportBulkActionIngestTests method validateDefaultPipeline.

private void validateDefaultPipeline(IndexRequest indexRequest) {
    Exception exception = new Exception("fake exception");
    indexRequest.source(emptyMap());
    AtomicBoolean responseCalled = new AtomicBoolean(false);
    AtomicBoolean failureCalled = new AtomicBoolean(false);
    assertNull(indexRequest.getPipeline());
    singleItemBulkWriteAction.execute(null, indexRequest, ActionListener.wrap(response -> {
        responseCalled.set(true);
    }, e -> {
        assertThat(e, sameInstance(exception));
        failureCalled.set(true);
    }));
    // check failure works, and passes through to the listener
    // haven't executed yet
    assertFalse(action.isExecuted);
    assertFalse(responseCalled.get());
    assertFalse(failureCalled.get());
    verify(ingestService).executeBulkRequest(eq(1), bulkDocsItr.capture(), failureHandler.capture(), completionHandler.capture(), any(), eq(Names.WRITE));
    assertEquals(indexRequest.getPipeline(), "default_pipeline");
    completionHandler.getValue().accept(null, exception);
    assertTrue(failureCalled.get());
    // now check success
    // this is done by the real pipeline execution service when processing
    indexRequest.setPipeline(IngestService.NOOP_PIPELINE_NAME);
    completionHandler.getValue().accept(DUMMY_WRITE_THREAD, null);
    assertTrue(action.isExecuted);
    // listener would only be called by real index action, not our mocked one
    assertFalse(responseCalled.get());
    verifyNoInteractions(transportService);
}

Example 100 with TransportService

use of org.opensearch.transport.TransportService in project OpenSearch by opensearch-project.

the class TransportMultiGetActionTests method beforeClass.

@BeforeClass
public static void beforeClass() throws Exception {
    threadPool = new TestThreadPool(TransportMultiGetActionTests.class.getSimpleName());
    transportService = new TransportService(Settings.EMPTY, mock(Transport.class), threadPool, TransportService.NOOP_TRANSPORT_INTERCEPTOR, boundAddress -> DiscoveryNode.createLocal(Settings.builder().put("node.name", "node1").build(), boundAddress.publishAddress(), randomBase64UUID()), null, emptySet()) {

        @Override
        public TaskManager getTaskManager() {
            return taskManager;
        }
    };
    final Index index1 = new Index("index1", randomBase64UUID());
    final Index index2 = new Index("index2", randomBase64UUID());
    final ClusterState clusterState = ClusterState.builder(new ClusterName(TransportMultiGetActionTests.class.getSimpleName())).metadata(new Metadata.Builder().put(new IndexMetadata.Builder(index1.getName()).settings(Settings.builder().put("index.version.created", Version.CURRENT).put("index.number_of_shards", 1).put("index.number_of_replicas", 1).put(IndexMetadata.SETTING_INDEX_UUID, index1.getUUID())).putMapping("_doc", XContentHelper.convertToJson(BytesReference.bytes(XContentFactory.jsonBuilder().startObject().startObject("_doc").startObject("_routing").field("required", false).endObject().endObject().endObject()), true, XContentType.JSON))).put(new IndexMetadata.Builder(index2.getName()).settings(Settings.builder().put("index.version.created", Version.CURRENT).put("index.number_of_shards", 1).put("index.number_of_replicas", 1).put(IndexMetadata.SETTING_INDEX_UUID, index1.getUUID())).putMapping("_doc", XContentHelper.convertToJson(BytesReference.bytes(XContentFactory.jsonBuilder().startObject().startObject("_doc").startObject("_routing").field("required", true).endObject().endObject().endObject()), true, XContentType.JSON)))).build();
    final ShardIterator index1ShardIterator = mock(ShardIterator.class);
    when(index1ShardIterator.shardId()).thenReturn(new ShardId(index1, randomInt()));
    final ShardIterator index2ShardIterator = mock(ShardIterator.class);
    when(index2ShardIterator.shardId()).thenReturn(new ShardId(index2, randomInt()));
    final OperationRouting operationRouting = mock(OperationRouting.class);
    when(operationRouting.getShards(eq(clusterState), eq(index1.getName()), anyString(), nullable(String.class), nullable(String.class))).thenReturn(index1ShardIterator);
    when(operationRouting.shardId(eq(clusterState), eq(index1.getName()), nullable(String.class), nullable(String.class))).thenReturn(new ShardId(index1, randomInt()));
    when(operationRouting.getShards(eq(clusterState), eq(index2.getName()), anyString(), nullable(String.class), nullable(String.class))).thenReturn(index2ShardIterator);
    when(operationRouting.shardId(eq(clusterState), eq(index2.getName()), nullable(String.class), nullable(String.class))).thenReturn(new ShardId(index2, randomInt()));
    clusterService = mock(ClusterService.class);
    when(clusterService.localNode()).thenReturn(transportService.getLocalNode());
    when(clusterService.state()).thenReturn(clusterState);
    when(clusterService.operationRouting()).thenReturn(operationRouting);
    shardAction = new TransportShardMultiGetAction(clusterService, transportService, mock(IndicesService.class), threadPool, new ActionFilters(emptySet()), new Resolver()) {

        @Override
        protected void doExecute(Task task, MultiGetShardRequest request, ActionListener<MultiGetShardResponse> listener) {
        }
    };
}