com.instaclustr.cassandra.sidecar.Sidecar Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassandra-sidecar Show documentation
Sidecar for Apache Cassandra
There is a newer version: 2.0.0-alpha8
package com.instaclustr.cassandra.sidecar;

import static com.google.inject.Guice.createInjector;
import static com.google.inject.Stage.PRODUCTION;
import static com.instaclustr.operations.OperationBindings.installOperationBindings;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;

import com.google.inject.AbstractModule;
import com.google.inject.Injector;
import com.instaclustr.cassandra.CassandraModule;
import com.instaclustr.cassandra.backup.guice.StorageModules;
import com.instaclustr.cassandra.backup.impl._import.ImportModule;
import com.instaclustr.cassandra.backup.impl.backup.BackupModules;
import com.instaclustr.cassandra.backup.impl.backup.BackupModules.BackupModule;
import com.instaclustr.cassandra.backup.impl.backup.BackupModules.CommitlogBackupModule;
import com.instaclustr.cassandra.backup.impl.restore.RestoreModules;
import com.instaclustr.cassandra.backup.impl.restore.RestoreModules.RestorationStrategyModule;
import com.instaclustr.cassandra.backup.impl.restore.RestoreModules.RestoreCommitlogModule;
import com.instaclustr.cassandra.backup.impl.restore.RestoreModules.RestoreModule;
import com.instaclustr.cassandra.backup.impl.truncate.TruncateOperation;
import com.instaclustr.cassandra.backup.impl.truncate.TruncateOperationRequest;
import com.instaclustr.cassandra.sidecar.operations.cleanup.CleanupsModule;
import com.instaclustr.cassandra.sidecar.operations.decommission.DecommissioningModule;
import com.instaclustr.cassandra.sidecar.operations.drain.DrainModule;
import com.instaclustr.cassandra.sidecar.operations.flush.FlushModule;
import com.instaclustr.cassandra.sidecar.operations.rebuild.RebuildModule;
import com.instaclustr.cassandra.sidecar.operations.refresh.RefreshModule;
import com.instaclustr.cassandra.sidecar.operations.restart.RestartModule;
import com.instaclustr.cassandra.sidecar.operations.scrub.ScrubModule;
import com.instaclustr.cassandra.sidecar.operations.upgradesstables.UpgradeSSTablesModule;
import com.instaclustr.cassandra.sidecar.service.ServicesModule;
import com.instaclustr.guice.Application;
import com.instaclustr.guice.ServiceManagerModule;
import com.instaclustr.jackson.JacksonModule;
import com.instaclustr.operations.OperationsModule;
import com.instaclustr.picocli.CLIApplication;
import com.instaclustr.picocli.CassandraJMXSpec;
import com.instaclustr.sidecar.http.JerseyHttpServerModule;
import com.instaclustr.sidecar.picocli.SidecarSpec;
import com.instaclustr.threading.ExecutorsModule;
import com.instaclustr.version.VersionModule;
import jmx.org.apache.cassandra.CassandraJMXConnectionInfo;
import picocli.CommandLine.Command;
import picocli.CommandLine.Mixin;
import picocli.CommandLine.Model.CommandSpec;
import picocli.CommandLine.Option;
import picocli.CommandLine.Spec;

@Command(name = "cassandra-sidecar",
         description = "Sidecar management application for Cassandra.",
         versionProvider = Sidecar.class,
         sortOptions = false,
         usageHelpWidth = 128,
         mixinStandardHelpOptions = true
)
public final class Sidecar extends CLIApplication implements Callable {

    @Mixin
    public SidecarSpec sidecarSpec;

    @Mixin
    public CassandraJMXSpec jmxSpec;

    @Spec
    private CommandSpec commandSpec;

    @Option(names = {"--enable-truncate"},
            description = "If enabled, sidecar will expose truncate operation on REST, defaults to false.")
    private boolean enableTruncateOperation;

    public static void main(String[] args) {
        main(args, true);
    }

    public static void mainWithoutExit(String[] args) {
        main(args, false);
    }

    public static void main(final String[] args, boolean exit) {
        int exitCode = execute(new Sidecar(), args);

        if (exit) {
            System.exit(exitCode);
        }
    }

    @Override
    public Void call() throws Exception {

        logCommandVersionInformation(commandSpec);

        // production binds singletons as eager by default
        final Injector injector = createInjector(PRODUCTION, getModules(sidecarSpec, jmxSpec, enableTruncateOperation));

        return injector.getInstance(Application.class).call();
    }

    @Override
    public String getImplementationTitle() {
        return "cassandra-sidecar";
    }

    public List getModules(SidecarSpec sidecarSpec,
                                           CassandraJMXSpec jmxSpec,
                                           final boolean enableTruncateOperation) throws Exception {
        List modules = new ArrayList<>();

        modules.addAll(backupRestoreModules());
        modules.addAll(operationModules());
        modules.addAll(sidecarModules(sidecarSpec, jmxSpec));

        if (enableTruncateOperation) {
            modules.add(new AbstractModule() {
                @Override
                protected void configure() {
                    installOperationBindings(binder(),
                                             "truncate",
                                             TruncateOperationRequest.class,
                                             TruncateOperation.class);
                }
            });
        }

        return modules;
    }

    public List sidecarModules(SidecarSpec sidecarSpec, CassandraJMXSpec jmxSpec) throws Exception {
        return new ArrayList() {{
            add(new VersionModule(getVersion()));
            add(new ServiceManagerModule());
            add(new CassandraModule(new CassandraJMXConnectionInfo(jmxSpec.jmxPassword,
                                                                   jmxSpec.jmxUser,
                                                                   jmxSpec.jmxServiceURL,
                                                                   jmxSpec.trustStore,
                                                                   jmxSpec.trustStorePassword,
                                                                   jmxSpec.keyStore,
                                                                   jmxSpec.keyStorePassword,
                                                                   jmxSpec.jmxClientAuth)));
            add(new JerseyHttpServerModule(sidecarSpec.httpServerAddress));
            add(new OperationsModule(sidecarSpec.operationsExpirationPeriod));
            add(new ExecutorsModule());
            add(new JacksonModule());
            add(new ServicesModule());
            add(new AbstractModule() {
                @Override
                protected void configure() {
                    bind(SidecarSpec.class).toInstance(sidecarSpec);
                }
            });
        }};
    }

    public static List backupRestoreModules() {
        return new ArrayList() {{
            add(new StorageModules());
            add(new BackupModule());
            add(new CommitlogBackupModule());
            add(new RestoreModule());
            add(new RestoreCommitlogModule());
            add(new RestorationStrategyModule());
            add(new BackupModules.UploadingModule());
            add(new RestoreModules.DownloadingModule());
        }};
    }

    public static List operationModules() {
        return new ArrayList() {{
            add(new DecommissioningModule());
            add(new CleanupsModule());
            add(new UpgradeSSTablesModule());
            add(new RebuildModule());
            add(new ScrubModule());
            add(new DrainModule());
            add(new RestartModule());
            add(new RefreshModule());
            add(new FlushModule());
            add(new ImportModule());
        }};
    }
}