org.mybatis.maven.mvnmigrate.ScriptCommandMojo Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of migrations-maven-plugin Show documentation

Migration-plugin is a Apache Maven 2 plugin that aims to integrate into the maven life cycle, the myBatis 3 Migration tool. MyBatis 3 Migration Schema is a tool that helps you to manage database schema changes.

There is a newer version: 1.2.0

Show newest version

/*
 *    Copyright 2010 The myBatis Team
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
package org.mybatis.maven.mvnmigrate;

import java.io.File;
import java.io.PrintStream;
import java.text.MessageFormat;

import org.apache.ibatis.migration.commands.ScriptCommand;
import org.apache.ibatis.migration.options.SelectedOptions;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;

/**
 * Goal which executes the ibatis migration script command.
 *
 * @version $Id$
 * @goal script
 */
public final class ScriptCommandMojo extends AbstractCommandMojo {

    /**
     * Initial version
     *
     * @parameter expression="${migration.v1}"
     */
    private String v1;

    /**
     * Final version.
     *
     * @parameter expression="${migration.v2}"
     * @required
     */
    private String v2;

    /**
     * The output file to be create.
     *
     * @parameter expression="${migration.output}"
     */
    private File output;

    /**
     * {@inheritDoc}
     * @param options
     */
    @Override
    protected ScriptCommand createCommandClass(SelectedOptions options) {
        return new ScriptCommand(options);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        if (isSkip()) return;

        try {
            init();

            if (this.output == null) {
                // Set the default System.out PrintStream
                this.getCommand().setPrintStream(System.out);
                if (this.getLog().isInfoEnabled()) {
                    String[] args = { this.v1, this.v2 };
                    MessageFormat format = new MessageFormat(getBundle(this.getLocale()).getString("migration.plugin.execution.command.script.sqlscript"));
                    this.getLog().info(format.format(args));
                }
                // Print out all generated script. This is the standard migration tool behavior.
                System.out.println("  --- CUT HERE ---");
            } else {
                if (!this.output.exists()) {
                    new File(this.output.getParent()).mkdirs();
                }
                this.getCommand().setPrintStream(new PrintStream(this.output));
            }

            this.getCommand().execute(this.v1
                    + " "
                    + this.v2);

            if (this.getLog().isInfoEnabled()) {
                if (this.output != null) {
                    File[] args = { this.output };
                    MessageFormat format = new MessageFormat(getBundle(this.getLocale()).getString("migration.plugin.execution.command.create.file"));
                    this.getLog().info(format.format(args));
                } else {
                    String[] args = { this.v1, this.v2 };
                    MessageFormat format = new MessageFormat(getBundle(this.getLocale()).getString("migration.plugin.execution.command.script.sqlscript"));
                    this.getLog().info(format.format(args));
                }
            }
        } catch (RuntimeException e) {
            throw e;
        } catch (Exception e) {
            throw new MojoExecutionException(e.getMessage(), e);
        }
    }

}