blob: e10b463f3b3f73cdd3c035e5401f7290d5941296 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
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(
repositories,
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.ctx().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))
}
}
|
