blob: f73e8e7b944d33cb78246272d914d0656e7efe71 (
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
65
66
67
68
69
70
71
72
|
package mill
package contrib.flyway
import java.net.URLClassLoader
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 org.flywaydb.core.internal.logging.console.ConsoleLog.Level
import org.flywaydb.core.internal.logging.console.ConsoleLogCreator
import upickle.default.writer
import scala.collection.JavaConverters._
trait FlywayModule extends JavaModule {
def flywayUrl: T[String]
def flywayUser: T[String] = T("")
def flywayPassword: T[String] = T("")
//def flywayClassLocationDeps: T[Agg[Dep]] = Agg.empty[Dep]
//def flywayClassLocations: T[Seq[String]] = T(Nil)
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 {
case a: String => a.nonEmpty
case _ => true
}
.map(key -> _)
private implicit val flywayWriter = writer[String].comap[Flyway](_.toString)
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))
}
}
|