org.nocrala.tools.database.db.source.Source Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of db-maven-plugin Show documentation
This Maven plugin automates the build, rebuild, and clean operations on a project database based on the SQL source code available in a strict, versioned structure in the the SCM.
There is a newer version: 1.0.3
Show newest version
package org.nocrala.tools.database.db.source;

import java.io.File;
import java.util.TreeMap;

import org.nocrala.tools.database.db.executor.Feedback;
import org.nocrala.tools.database.db.executor.SQLExecutor;
import org.nocrala.tools.database.db.executor.SQLExecutor.CouldNotReadSQLScriptException;
import org.nocrala.tools.database.db.executor.SQLExecutor.SQLScriptAbortedException;
import org.nocrala.tools.database.db.utils.VersionNumber;

public class Source {

  private File sourceDir;
  private boolean layeredBuild;
  private boolean layeredScenarios;
  private boolean buildOnErrorContinue;
  private boolean cleanOnErrorContinue;

  private Feedback feedback;

  private TreeMap layers;

  public Source(final File sourceDir, final boolean layeredBuild, final boolean layeredScenarios,
      final boolean buildOnErrorContinue, final boolean cleanOnErrorContinue, final Feedback feedback)
      throws InvalidDatabaseSourceException {

    this.feedback = feedback;

    if (sourceDir == null) {
      feedback.error("Mandatory property (sourcedir) is not specified.");
      throw new InvalidDatabaseSourceException("Mandatory property (sourcedir) is not specified");
    }
    if (!sourceDir.exists()) {
      this.feedback.error("sourcedir does not exist: " + sourceDir);
      throw new InvalidDatabaseSourceException("sourcedir does not exist: " + sourceDir);
    }
    if (!sourceDir.isDirectory()) {
      this.feedback.error("sourcedir is not a directory: " + sourceDir);
      throw new InvalidDatabaseSourceException("sourcedir is not a directory: " + sourceDir);
    }

    if (!layeredBuild && layeredScenarios) {
      this.feedback.error("layered scenarios cannot be enabled when builds are not layered");
      throw new InvalidDatabaseSourceException("layered scenarios cannot be enabled when builds are not layered");
    }

    this.sourceDir = sourceDir;
    this.layeredBuild = layeredBuild;
    this.layeredScenarios = layeredScenarios;
    this.buildOnErrorContinue = buildOnErrorContinue;
    this.cleanOnErrorContinue = cleanOnErrorContinue;

    File[] dirs = sourceDir.listFiles(f -> f.isDirectory());

    this.layers = new TreeMap<>();
    for (File f : dirs) {
      VersionNumber v = new VersionNumber(f.getName());
      Layer l = new Layer(v, f);
      this.layers.put(v, l);
    }

  }

  public void build(final VersionNumber currentVersion, final String scenarioName, final SQLExecutor sqlExecutor)
      throws CouldNotReadSQLScriptException, SQLScriptAbortedException {
    Layer l = this.layers.get(currentVersion);
    if (l == null) {
      throw new IllegalArgumentException(
          "Could build database; version " + currentVersion + " does not exist in: " + this.sourceDir);
    }

    if (this.layeredBuild) {
      for (VersionNumber v : this.layers.keySet()) {
        if (!v.after(currentVersion)) {
          buildLayer(this.layers.get(v), sqlExecutor);
          if (this.layeredScenarios || v.same(currentVersion)) {
            buildScenario(this.layers.get(v), scenarioName, sqlExecutor);
          }
        }
      }
    } else {
      buildLayer(l, sqlExecutor);
      buildScenario(l, scenarioName, sqlExecutor);
    }

  }

  public void clean(final VersionNumber currentVersion, final SQLExecutor sqlExecutor)
      throws CouldNotReadSQLScriptException, SQLScriptAbortedException {
    Layer l = this.layers.get(currentVersion);
    if (l == null) {
      throw new IllegalArgumentException(
          "Could build database; version " + currentVersion + " does not exist in: " + this.sourceDir);
    }

    if (this.layeredBuild) {
      for (VersionNumber v : this.layers.descendingKeySet()) {
        if (!v.after(currentVersion)) {
          cleanLayer(this.layers.get(v), sqlExecutor);
        }
      }
    } else {
      cleanLayer(l, sqlExecutor);
    }

  }

  public void rebuild(final VersionNumber currentVersion, final String scenarioName, final SQLExecutor sqlExecutor)
      throws CouldNotReadSQLScriptException, SQLScriptAbortedException {
    clean(currentVersion, sqlExecutor);
    build(currentVersion, scenarioName, sqlExecutor);
  }

  // Delegate methods

  private void buildLayer(final Layer l, final SQLExecutor sqlExecutor)
      throws CouldNotReadSQLScriptException, SQLScriptAbortedException {
    l.build(sqlExecutor, this.buildOnErrorContinue);
  }

  private void buildScenario(final Layer l, final String scenarioName, final SQLExecutor sqlExecutor)
      throws CouldNotReadSQLScriptException, SQLScriptAbortedException {
    l.buildScenario(scenarioName, sqlExecutor, this.buildOnErrorContinue);
  }

  private void cleanLayer(final Layer l, final SQLExecutor sqlExecutor)
      throws CouldNotReadSQLScriptException, SQLScriptAbortedException {
    l.clean(sqlExecutor, this.cleanOnErrorContinue);
  }

  // toString()

  public String toString() {
    StringBuilder sb = new StringBuilder();
    sb.append("layers=" + this.layers.size() + "\n");
    for (Layer l : this.layers.values()) {
      sb.append(" * layer '" + l.getVersionNumber() + "' - " + l + "\n");
    }
    return sb.toString();
  }

  // Classes

  public static class InvalidDatabaseSourceException extends Exception {

    private static final long serialVersionUID = 1L;

    public InvalidDatabaseSourceException(final String message) {
      super(message);
    }

  }

}