Examples with RestoreSnapshotRequest org.opensearch.action.admin.cluster.snapshots.restore.RestoreSnapshotRequest used on opensource projects

Example 1 with RestoreSnapshotRequest

use of org.opensearch.action.admin.cluster.snapshots.restore.RestoreSnapshotRequest in project OpenSearch by opensearch-project.

the class SnapshotRequestConverters method restoreSnapshot.

static Request restoreSnapshot(RestoreSnapshotRequest restoreSnapshotRequest) throws IOException {
    String endpoint = new RequestConverters.EndpointBuilder().addPathPartAsIs("_snapshot").addPathPart(restoreSnapshotRequest.repository()).addPathPart(restoreSnapshotRequest.snapshot()).addPathPartAsIs("_restore").build();
    Request request = new Request(HttpPost.METHOD_NAME, endpoint);
    RequestConverters.Params parameters = new RequestConverters.Params();
    parameters.withMasterTimeout(restoreSnapshotRequest.masterNodeTimeout());
    parameters.withWaitForCompletion(restoreSnapshotRequest.waitForCompletion());
    request.addParameters(parameters.asMap());
    request.setEntity(RequestConverters.createEntity(restoreSnapshotRequest, RequestConverters.REQUEST_BODY_CONTENT_TYPE));
    return request;
}

Example 2 with RestoreSnapshotRequest

use of org.opensearch.action.admin.cluster.snapshots.restore.RestoreSnapshotRequest in project OpenSearch by opensearch-project.

the class SnapshotRequestConvertersTests method testRestoreSnapshot.

public void testRestoreSnapshot() throws IOException {
    Map<String, String> expectedParams = new HashMap<>();
    String repository = RequestConvertersTests.randomIndicesNames(1, 1)[0];
    String snapshot = "snapshot-" + randomAlphaOfLengthBetween(2, 5).toLowerCase(Locale.ROOT);
    String endpoint = String.format(Locale.ROOT, "/_snapshot/%s/%s/_restore", repository, snapshot);
    RestoreSnapshotRequest restoreSnapshotRequest = new RestoreSnapshotRequest(repository, snapshot);
    RequestConvertersTests.setRandomMasterTimeout(restoreSnapshotRequest, expectedParams);
    boolean waitForCompletion = randomBoolean();
    restoreSnapshotRequest.waitForCompletion(waitForCompletion);
    expectedParams.put("wait_for_completion", Boolean.toString(waitForCompletion));
    if (randomBoolean()) {
        String timeout = randomTimeValue();
        restoreSnapshotRequest.masterNodeTimeout(timeout);
        expectedParams.put("master_timeout", timeout);
    }
    Request request = SnapshotRequestConverters.restoreSnapshot(restoreSnapshotRequest);
    assertThat(request.getEndpoint(), equalTo(endpoint));
    assertThat(request.getMethod(), equalTo(HttpPost.METHOD_NAME));
    assertThat(request.getParameters(), equalTo(expectedParams));
    RequestConvertersTests.assertToXContentBody(restoreSnapshotRequest, request.getEntity());
}

Example 3 with RestoreSnapshotRequest

use of org.opensearch.action.admin.cluster.snapshots.restore.RestoreSnapshotRequest in project OpenSearch by opensearch-project.

the class SnapshotResiliencyTests method testSuccessfulSnapshotAndRestore.

public void testSuccessfulSnapshotAndRestore() {
    setupTestCluster(randomFrom(1, 3, 5), randomIntBetween(2, 10));
    String repoName = "repo";
    String snapshotName = "snapshot";
    final String index = "test";
    final int shards = randomIntBetween(1, 10);
    final int documents = randomIntBetween(0, 100);
    final TestClusterNodes.TestClusterNode masterNode = testClusterNodes.currentMaster(testClusterNodes.nodes.values().iterator().next().clusterService.state());
    final StepListener<CreateSnapshotResponse> createSnapshotResponseListener = new StepListener<>();
    continueOrDie(createRepoAndIndex(repoName, index, shards), createIndexResponse -> {
        final Runnable afterIndexing = () -> client().admin().cluster().prepareCreateSnapshot(repoName, snapshotName).setWaitForCompletion(true).execute(createSnapshotResponseListener);
        if (documents == 0) {
            afterIndexing.run();
        } else {
            final BulkRequest bulkRequest = new BulkRequest().setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE);
            for (int i = 0; i < documents; ++i) {
                bulkRequest.add(new IndexRequest(index).source(Collections.singletonMap("foo", "bar" + i)));
            }
            final StepListener<BulkResponse> bulkResponseStepListener = new StepListener<>();
            client().bulk(bulkRequest, bulkResponseStepListener);
            continueOrDie(bulkResponseStepListener, bulkResponse -> {
                assertFalse("Failures in bulk response: " + bulkResponse.buildFailureMessage(), bulkResponse.hasFailures());
                assertEquals(documents, bulkResponse.getItems().length);
                afterIndexing.run();
            });
        }
    });
    final StepListener<AcknowledgedResponse> deleteIndexListener = new StepListener<>();
    continueOrDie(createSnapshotResponseListener, createSnapshotResponse -> client().admin().indices().delete(new DeleteIndexRequest(index), deleteIndexListener));
    final StepListener<RestoreSnapshotResponse> restoreSnapshotResponseListener = new StepListener<>();
    continueOrDie(deleteIndexListener, ignored -> client().admin().cluster().restoreSnapshot(new RestoreSnapshotRequest(repoName, snapshotName).waitForCompletion(true), restoreSnapshotResponseListener));
    final StepListener<SearchResponse> searchResponseListener = new StepListener<>();
    continueOrDie(restoreSnapshotResponseListener, restoreSnapshotResponse -> {
        assertEquals(shards, restoreSnapshotResponse.getRestoreInfo().totalShards());
        client().search(new SearchRequest(index).source(new SearchSourceBuilder().size(0).trackTotalHits(true)), searchResponseListener);
    });
    final AtomicBoolean documentCountVerified = new AtomicBoolean();
    continueOrDie(searchResponseListener, r -> {
        assertEquals(documents, Objects.requireNonNull(r.getHits().getTotalHits()).value);
        documentCountVerified.set(true);
    });
    runUntil(documentCountVerified::get, TimeUnit.MINUTES.toMillis(5L));
    assertNotNull(createSnapshotResponseListener.result());
    assertNotNull(restoreSnapshotResponseListener.result());
    assertTrue(documentCountVerified.get());
    SnapshotsInProgress finalSnapshotsInProgress = masterNode.clusterService.state().custom(SnapshotsInProgress.TYPE);
    assertFalse(finalSnapshotsInProgress.entries().stream().anyMatch(entry -> entry.state().completed() == false));
    final Repository repository = masterNode.repositoriesService.repository(repoName);
    Collection<SnapshotId> snapshotIds = getRepositoryData(repository).getSnapshotIds();
    assertThat(snapshotIds, hasSize(1));
    final SnapshotInfo snapshotInfo = repository.getSnapshotInfo(snapshotIds.iterator().next());
    assertEquals(SnapshotState.SUCCESS, snapshotInfo.state());
    assertThat(snapshotInfo.indices(), containsInAnyOrder(index));
    assertEquals(shards, snapshotInfo.successfulShards());
    assertEquals(0, snapshotInfo.failedShards());
}

Example 4 with RestoreSnapshotRequest

use of org.opensearch.action.admin.cluster.snapshots.restore.RestoreSnapshotRequest in project OpenSearch by opensearch-project.

the class RestoreServiceTests method testPrefixNotChanged.

public void testPrefixNotChanged() {
    String dataStreamName = "ds-000001";
    String renamedDataStreamName = "ds2-000001";
    String backingIndexName = DataStream.getDefaultBackingIndexName(dataStreamName, 1);
    String renamedBackingIndexName = DataStream.getDefaultBackingIndexName(renamedDataStreamName, 1);
    List<Index> indices = Collections.singletonList(new Index(backingIndexName, "uuid"));
    DataStream dataStream = new DataStream(dataStreamName, createTimestampField("@timestamp"), indices);
    Metadata.Builder metadata = mock(Metadata.Builder.class);
    IndexMetadata indexMetadata = mock(IndexMetadata.class);
    when(metadata.get(eq(renamedBackingIndexName))).thenReturn(indexMetadata);
    Index renamedIndex = new Index(renamedBackingIndexName, "uuid2");
    when(indexMetadata.getIndex()).thenReturn(renamedIndex);
    RestoreSnapshotRequest request = new RestoreSnapshotRequest().renamePattern("ds-").renameReplacement("ds2-");
    DataStream renamedDataStream = RestoreService.updateDataStream(dataStream, metadata, request);
    assertEquals(renamedDataStreamName, renamedDataStream.getName());
    assertEquals(Collections.singletonList(renamedIndex), renamedDataStream.getIndices());
    request = new RestoreSnapshotRequest().renamePattern("ds-000001").renameReplacement("ds2-000001");
    renamedDataStream = RestoreService.updateDataStream(dataStream, metadata, request);
    assertEquals(renamedDataStreamName, renamedDataStream.getName());
    assertEquals(Collections.singletonList(renamedIndex), renamedDataStream.getIndices());
}

Example 5 with RestoreSnapshotRequest

use of org.opensearch.action.admin.cluster.snapshots.restore.RestoreSnapshotRequest in project OpenSearch by opensearch-project.

the class RestoreServiceTests method testUpdateDataStreamRename.

public void testUpdateDataStreamRename() {
    String dataStreamName = "data-stream-1";
    String renamedDataStreamName = "data-stream-2";
    String backingIndexName = DataStream.getDefaultBackingIndexName(dataStreamName, 1);
    String renamedBackingIndexName = DataStream.getDefaultBackingIndexName(renamedDataStreamName, 1);
    List<Index> indices = Collections.singletonList(new Index(backingIndexName, "uuid"));
    DataStream dataStream = new DataStream(dataStreamName, createTimestampField("@timestamp"), indices);
    Metadata.Builder metadata = mock(Metadata.Builder.class);
    IndexMetadata indexMetadata = mock(IndexMetadata.class);
    when(metadata.get(eq(renamedBackingIndexName))).thenReturn(indexMetadata);
    Index renamedIndex = new Index(renamedBackingIndexName, "uuid2");
    when(indexMetadata.getIndex()).thenReturn(renamedIndex);
    RestoreSnapshotRequest request = new RestoreSnapshotRequest().renamePattern("data-stream-1").renameReplacement("data-stream-2");
    DataStream renamedDataStream = RestoreService.updateDataStream(dataStream, metadata, request);
    assertEquals(renamedDataStreamName, renamedDataStream.getName());
    assertEquals(Collections.singletonList(renamedIndex), renamedDataStream.getIndices());
}