ill-contrib-flyway_2.13.0.10.0-M3-4-38105f.source-code.FlywayModule.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mill-contrib-flyway_2.13 Show documentation

mill-contrib-flyway

There is a newer version: 0.12.4

Show newest version

package mill
package contrib.flyway

import java.net.URLClassLoader

import mill.contrib.flyway.ConsoleLog.Level
import mill.scalalib.Lib.resolveDependencies
import mill.scalalib._
import org.flywaydb.core.Flyway
import org.flywaydb.core.api.MigrationVersion
import org.flywaydb.core.api.logging.LogFactory
import org.flywaydb.core.internal.configuration.{ConfigUtils => flyway}
import org.flywaydb.core.internal.info.MigrationInfoDumper

import scala.collection.JavaConverters._

trait FlywayModule extends JavaModule {

  def flywayUrl: T[String]
  def flywayUser: T[String] = T("")
  def flywayPassword: T[String] = T("")
  def flywayFileLocations: T[Seq[PathRef]] =
    T(resources().map(pr => PathRef(pr.path / "db" / "migration", pr.quick)))
  def flywayDriverDeps: T[Agg[Dep]]
  def jdbcClasspath = T(resolveDependencies(
    repositoriesTask(),
    Lib.depToDependencyJava(_),
    flywayDriverDeps()
  ))

  private def strToOptPair[A](key: String, v: String) =
    Option(v)
      .filter(_.nonEmpty)
      .map(key -> _)

  def flywayInstance = T.worker {
    val jdbcClassloader = new URLClassLoader(jdbcClasspath().map(_.path.toIO.toURI.toURL).toArray)

    val configProps = Map(flyway.URL -> flywayUrl()) ++
      strToOptPair(flyway.USER, flywayUser()) ++
      strToOptPair(flyway.PASSWORD, flywayPassword())

    LogFactory.setLogCreator(new ConsoleLogCreator(Level.INFO))

    Flyway
      .configure(jdbcClassloader)
      .locations(flywayFileLocations().map("filesystem:" + _.path): _*)
      .configuration(configProps.asJava)
      .load
  }

  def flywayMigrate() = T.command(flywayInstance().migrate())
  def flywayClean() = T.command(flywayInstance().clean())
  def flywayBaseline() = T.command(flywayInstance().baseline())
  def flywayInfo() = T.command {
    val log = T.log
    val info = flywayInstance().info
    val current = info.current
    val currentSchemaVersion =
      if (current == null) MigrationVersion.EMPTY
      else current.getVersion
    log.info("Schema version: " + currentSchemaVersion)
    log.info(MigrationInfoDumper.dumpToAsciiTable(info.all))
  }
}