flecomte/postgres-json
1
1
Fork 0
Code Issues 10 Pull Requests 2 Actions Packages Projects Releases 2 Wiki Activity

feature #9: add lock on execute migration

Browse Source
This commit is contained in:
Fabrice Lecomte
2019-07-05 21:09:07 +02:00
parent 9c02fd21ca
commit 6dc5e611aa
4 changed files with 48 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/main/kotlin/fr/postgresjson/migration/Migrations.kt
View File

@@ -159,7 +159,13 @@ class Migrations(directory: File, private val connection: Connection) {
}
}
fun up(): Map<String, Status> {
private fun lock() {
File(this::class.java.getResource("/sql/migration/lockMigrationTables.sql").toURI()).let {
connection.connect().sendQuery(it.readText()).join()
}
}
internal fun up(): Map<String, Status> {
val list: MutableMap<String, Status> = mutableMapOf()
queries.forEach {
it.value.let { query ->
@@ -184,7 +190,7 @@ class Migrations(directory: File, private val connection: Connection) {
return list.toMap()
}
fun down(force: Boolean = false): Map<String, Status> {
internal fun down(force: Boolean = false): Map<String, Status> {
val list: MutableMap<String, Status> = mutableMapOf()
queries.forEach {
it.value.let { query ->
@@ -209,6 +215,23 @@ class Migrations(directory: File, private val connection: Connection) {
return list.toMap()
}
fun run(): Map<Pair<String, Direction>, Status> {
val list: MutableMap<Pair<String, Direction>, Status> = mutableMapOf()
connection.connect().apply {
sendQuery("BEGIN").join()
lock()
up().map {
list[Pair(it.key, Direction.UP)] = it.value
}
down(true).map {
list[Pair(it.key, Direction.DOWN)] = it.value
}
sendQuery("COMMIT").join()
}
return list.toMap()
}
fun test(): Map<Pair<String, Direction>, Status> {
val list: MutableMap<Pair<String, Direction>, Status> = mutableMapOf()
connection.connect().apply {