io.dropwizard.flyway.cli.DbCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dropwizard-flyway Show documentation

Addon bundle for Dropwizard to support Flyway for database migrations

There is a newer version: 4.0.1-2

package io.dropwizard.flyway.cli;

import io.dropwizard.flyway.FlywayConfiguration;
import io.dropwizard.Configuration;
import io.dropwizard.db.DatabaseConfiguration;
import net.sourceforge.argparse4j.inf.Namespace;
import net.sourceforge.argparse4j.inf.Subparser;
import org.flywaydb.core.Flyway;

import java.util.SortedMap;
import java.util.TreeMap;

public class DbCommand extends AbstractFlywayCommand {
    private static final String COMMAND_NAME_ATTR = "subCommand";
    private final SortedMap> subCommands = new TreeMap<>();

    public DbCommand(final DatabaseConfiguration databaseConfiguration,
                     final FlywayConfiguration flywayConfiguration,
                     final Class configurationClass) {
        super("db", "Run database migration tasks",
                databaseConfiguration, flywayConfiguration, configurationClass);

        addSubCommand(new DbMigrateCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
        addSubCommand(new DbCleanCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
        addSubCommand(new DbInitCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
        addSubCommand(new DbValidateCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
        addSubCommand(new DbInfoCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
        addSubCommand(new DbRepairCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
    }

    private void addSubCommand(final AbstractFlywayCommand subCommand) {
        subCommands.put(subCommand.getName(), subCommand);
    }

    @Override
    public void configure(final Subparser subparser) {
        for (AbstractFlywayCommand subCommand : subCommands.values()) {
            final Subparser cmdParser = subparser.addSubparsers()
                    .addParser(subCommand.getName())
                    .setDefault(COMMAND_NAME_ATTR, subCommand.getName())
                    .description(subCommand.getDescription());
            subCommand.configure(cmdParser);
        }
    }

    @Override
    public void run(final Namespace namespace, final Flyway flyway) throws Exception {
        subCommands.get(namespace.getString(COMMAND_NAME_ATTR)).run(namespace, flyway);
    }
}