com.instaclustr.cassandra.backup.cli.RestoreApplication Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of instaclustr-backup-restore Show documentation

Backup and restoration tooling for Cassandra

There is a newer version: 2.0.0-alpha8

Show newest version

package com.instaclustr.cassandra.backup.cli;

import static com.instaclustr.cassandra.backup.cli.BackupRestoreCLI.init;
import static com.instaclustr.picocli.CLIApplication.execute;
import static com.instaclustr.picocli.JarManifestVersionProvider.logCommandVersionInformation;
import static java.util.Collections.singletonList;
import static org.awaitility.Awaitility.await;

import com.google.inject.Inject;
import com.instaclustr.cassandra.backup.impl.restore.RestoreModules.RestoreModule;
import com.instaclustr.cassandra.backup.impl.restore.RestoreOperationRequest;
import com.instaclustr.operations.Operation;
import com.instaclustr.operations.OperationsService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import picocli.CommandLine.Command;
import picocli.CommandLine.Mixin;
import picocli.CommandLine.Model.CommandSpec;
import picocli.CommandLine.Spec;

@Command(name = "restore",
    mixinStandardHelpOptions = true,
    description = "Restore the Cassandra data on this node to a specified point-in-time.",
    sortOptions = false,
    versionProvider = BackupRestoreCLI.class
)
public class RestoreApplication implements Runnable {

    private static final Logger logger = LoggerFactory.getLogger(RestoreApplication.class);

    @Spec
    private CommandSpec spec;

    @Mixin
    private RestoreOperationRequest request;

    @Inject
    private OperationsService operationsService;

    public static void main(String[] args) {
        System.exit(execute(new RestoreApplication(), args));
    }

    @Override
    public void run() {
        logCommandVersionInformation(spec);

        init(this, null, request, logger, singletonList(new RestoreModule()));

        final Operation operation = operationsService.submitOperationRequest(request);

        await().forever().until(() -> operation.state.isTerminalState());

        if (operation.state == Operation.State.FAILED) {
            throw new IllegalStateException("Restore operation was not successful.");
        }
    }
}