H2 Databaseのデータだけバックアップ&リストアするGradleのタスク

DBスキーマはFlywayで管理する前提で、テスト中の一時的なデータをバックアップ＆リストアするために、Gradleのタスクを作りました。

./gradlew backupH2 flywayClean flywayMigrate restoreH2

みたいにしたかったのですが、Flywayの管理テーブル（schema_version）やFlywayのスクリプトで管理しているデータがあるため、プライマリキー違反になってしまいダメでした。。。

とりあえず、データだけバックアップする方法としてメモを残しておこう。

バックアップとリストアのタスク

バックアップは「Backup using the Script Tool」で作成したスクリプトファイルからCREATE文とALTER文を取り除いているだけです。

build.gradle

apply plugin: 'java'

repositories {
    jcenter()
}


dependencies {
    runtime('com.h2database:h2:1.4.196')
}

// H2を使用するアプリのbuild.gradleが既にある場合は、ここから下を追記するだけで大丈夫なはず。
// スクリプトファイル名、DB接続URLなどは適宜変更する。

// H2のデータのバックアップ
task backupH2(type: JavaExec) {
    classpath = sourceSets.main.runtimeClasspath
    main = 'org.h2.tools.Script'
    args = ['-url', 'jdbc:h2:file:./h2/hoge', '-user', 'sa', '-password', '', '-script', 'data.sql']
    doLast {
        file('data.sql').text = file('data.sql').text.replaceAll(/(CREATE|ALTER)[^;]+;\s*\R*/, '')
    }
}

// H2のデータのリストア
task restoreH2(type: JavaExec) {
    classpath = sourceSets.main.runtimeClasspath
    main = 'org.h2.tools.RunScript'
    args = ['-url', 'jdbc:h2:file:./h2/hoge', '-user', 'sa', '-password', '', '-script', 'data.sql']
}

（追記）改良版

バックアップとリストアの部分の改良版です。
スクリプトファイルに対して、Flywayの管理テーブル（schema_version）のINSERT文を削除し、その他のテーブルのDELETE文を追加しています。

def h2Url = 'jdbc:h2:file:./h2/hoge'
def h2User = 'sa'
def h2Password = ''
def h2ScriptFileName = 'data.sql'

// H2のデータのバックアップ
task backupH2(type: JavaExec) {
    classpath = sourceSets.main.runtimeClasspath
    main = 'org.h2.tools.Script'
    args = ['-url', h2Url, '-user', h2User, '-password', h2Password, '-script', h2ScriptFileName]
    doLast {
        def insertScript = file(h2ScriptFileName).text
        insertScript = insertScript.replaceAll(/(CREATE|ALTER)[^;]+;\s*\R*/, '')
        insertScript = insertScript.replaceAll(/(--|INSERT)[^;]+schema_version[^;]+;\s*\R*/, '')

        def deleteScript = ';\n'
        Set<String> tables = []
        insertScript.findAll(/INSERT INTO ([^\s]+)\([^;]+;\s*\R*/){ g0, g1 -> tables.add(g1) }
        tables.forEach() { table -> deleteScript += 'DELETE FROM ' + table + ';\n' }

        file(h2ScriptFileName).text = deleteScript + insertScript
    }
}

// H2のデータのリストア
task restoreH2(type: JavaExec) {
    classpath = sourceSets.main.runtimeClasspath
    main = 'org.h2.tools.RunScript'
    args = ['-url', h2Url, '-user', h2User, '-password', h2Password, '-script', h2ScriptFileName]
}

自分の環境では

./gradlew backupH2 flywayClean flywayMigrate restoreH2

ができたので、ひとまず良しとしよう。