Gradle script snippet for flyway to load config from Spring Boot

build.gradle

def getAppProperties() {
    // from yaml
    return new org.yaml.snakeyaml.Yaml().load(
            file("${project.buildDir}/resources/main/application.yml").newDataInputStream())
    // from properties
//    Properties appProp = new Properties()
//        appProp.load(file("$buildDir/resources/main/application.properties").newDataInputStream())
//    return appProp;
}

task("doMigration") {
    dependsOn processResources
    doFirst {
        def appProp = getAppProperties()
        def dbUrl = appProp.spring.datasource.url
        println ">>>>>>>>>>> doing db_migration on $dbUrl"

        flyway {
            url = dbUrl
            user = appProp.spring.datasource.username
            password = appProp.spring.datasource.password
            driver = appProp.spring.datasource["driver-class-name"]
            schemas = ["pf4j"]
            baselineOnMigrate = true
            table = "_db_migration"
            locations = ["filesystem:$buildDir/resources/main/db_migration"]
        }

    }
    finalizedBy flywayMigrate, flywayInfo
}