cn.dreampie.common.plugin.db.FlywayPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfinal-dreampie Show documentation
jfinal shiro-freemarker plugins
There is a newer version: 1.2.2
package cn.dreampie.common.plugin.db;

import com.google.common.collect.Maps;
import com.jfinal.plugin.IPlugin;
import org.apache.commons.lang3.StringUtils;
import org.flywaydb.core.Flyway;
import org.flywaydb.core.api.MigrationInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;
import java.util.Map;

/**
 * Created by wangrenhui on 14-4-21.
 */
public class FlywayPlugin implements IPlugin {

  private Logger logger = LoggerFactory.getLogger(getClass());

  private String flywayPrefixToMigrationScript = "/db/migration/";


  private DbConfig dbConfig;

  public FlywayPlugin() {
    dbConfig = new DbConfig();
  }

  public FlywayPlugin(String config) {
    dbConfig = new DbConfig(config);
  }


  @Override
  public boolean start() {
    List dbNames = dbConfig.getAllDbNames();
    boolean dev = dbConfig.isDev();
    for (String dbName : dbNames) {
      boolean auto = dbConfig.migrateAuto(dbName);
      if (dev || auto) {
        migrateAutomatically(dbName);
      } else {
        checkState(dbName);
      }
    }
    logger.info("flyway inited!");
    return true;
  }

  @Override
  public boolean stop() {
    List dbNames = dbConfig.getAllDbNames();
    for (String dbName : dbNames) {
      cleanAutomatically(dbName);
    }
    logger.info("flyway stoped!");
    return true;
  }

  private Map flyways() {
    Map flywayMap = Maps.newHashMap();
    Map dbSourceMap = dbConfig.getAllDbSources();
    DbSource dbSource = null;
    String migrationFilesLocation = null;
    Flyway flyway = null;
    for (String dbName : dbSourceMap.keySet()) {
      migrationFilesLocation = flywayPrefixToMigrationScript + dbName;
      if (dbConfig.migrationFileDirectoryExists(migrationFilesLocation)) {
        dbSource = dbSourceMap.get(dbName);
        flyway = new Flyway();
        flyway.setDataSource(dbSource.url, dbSource.user, dbSource.password);
        flyway.setLocations(migrationFilesLocation);
        if (dbConfig.isClean(dbName)) {
          flyway.setCleanOnValidationError(true);
        }
        if (dbConfig.initOnMigrate(dbName)) {
          flyway.setInitOnMigrate(true);
        }
        flywayMap.put(dbName, flyway);
      }
    }
    return flywayMap;
  }

  private void migrateAutomatically(String dbName) {
    Map flywayMap = flyways();
    flywayMap.get(dbName).migrate();
  }

  private void cleanAutomatically(String dbName) {
    Map flywayMap = flyways();
    flywayMap.get(dbName).clean();
  }

  private void checkState(String dbName) {
    Map flywayMap = flyways();

    MigrationInfo[] pendingMigrations = flywayMap.get(dbName).info().pending();

    if (pendingMigrations != null) {
      throw new RuntimeException(dbName + "-" + StringUtils.join(pendingMigrations, ","));
    }
  }
}