Port transactionIsolationLevel to the Vitess database gradle plugin

GitOrigin-RevId: c9df6f520823f4263385f00f05b7cdcc3277ff8d

Author: aparajon

misk-vitess-database-gradle-plugin/api/misk-vitess-database-gradle-plugin.api

@@ -11,6 +11,7 @@ public abstract class misk/vitess/gradle/StartVitessDatabaseTask : org/gradle/ap
 	public abstract fun getPort ()Lorg/gradle/api/provider/Property;
 	public abstract fun getSchemaDir ()Lorg/gradle/api/provider/Property;
 	public abstract fun getSqlMode ()Lorg/gradle/api/provider/Property;
+	public abstract fun getTransactionIsolationLevel ()Lorg/gradle/api/provider/Property;
 	public abstract fun getTransactionTimeoutSeconds ()Lorg/gradle/api/provider/Property;
 	public abstract fun getVitessImage ()Lorg/gradle/api/provider/Property;
 	public abstract fun getVitessVersion ()Lorg/gradle/api/provider/Property;

misk-vitess-database-gradle-plugin/build.gradle.kts

@@ -17,7 +17,8 @@ gradlePlugin {
 }
 
 dependencies {
-  implementation(testFixtures(project(":misk-vitess")))
+  api(testFixtures(project(":misk-vitess")))
+
   testImplementation(gradleTestKit())
   testImplementation(libs.assertj)
   testImplementation(libs.junitApi)

misk-vitess-database-gradle-plugin/src/main/kotlin/misk/vitess/gradle/StartVitessDatabaseTask.kt

@@ -1,5 +1,6 @@
 package misk.vitess.gradle
 
+import misk.vitess.testing.TransactionIsolationLevel
 import misk.vitess.testing.VitessTestDb
 import org.gradle.api.DefaultTask
 import org.gradle.api.provider.Property
@@ -44,6 +45,9 @@ abstract class StartVitessDatabaseTask: DefaultTask() {
   @get:Input
   abstract val sqlMode: Property<String>
 
+  @get:Input
+  abstract val transactionIsolationLevel: Property<TransactionIsolationLevel>
+
   @get:Input
   abstract val transactionTimeoutSeconds: Property<Duration>
 
@@ -67,6 +71,7 @@ abstract class StartVitessDatabaseTask: DefaultTask() {
       port = port.get(),
       schemaDir = schemaDir.get(),
       sqlMode = sqlMode.get(),
+      transactionIsolationLevel = transactionIsolationLevel.get(),
       transactionTimeoutSeconds = transactionTimeoutSeconds.get(),
       vitessImage = vitessImage.get(),
       vitessVersion = vitessVersion.get()

misk-vitess-database-gradle-plugin/src/main/kotlin/misk/vitess/gradle/VitessDatabasePlugin.kt

@@ -18,6 +18,7 @@ class VitessDatabasePlugin: Plugin<Project> {
       it.port.convention(DefaultSettings.PORT)
       it.schemaDir.convention("filesystem:${project.layout.projectDirectory.dir("src/main/resources/vitess/schema").asFile.absolutePath}")
       it.sqlMode.convention(DefaultSettings.SQL_MODE)
+      it.transactionIsolationLevel.convention(DefaultSettings.TRANSACTION_ISOLATION_LEVEL)
       it.transactionTimeoutSeconds.convention(DefaultSettings.TRANSACTION_TIMEOUT_SECONDS)
       it.vitessImage.convention(DefaultSettings.VITESS_IMAGE)
       it.vitessVersion.convention(DefaultSettings.VITESS_VERSION)

misk-vitess-database-gradle-plugin/src/test/kotlin/misk/vitess/gradle/VitessDatabasePluginTest.kt

@@ -25,13 +25,18 @@ class VitessDatabasePluginTest {
 
     assertThat(result.task(":startVitessDatabase")!!.outcome).isEqualTo(TaskOutcome.SUCCESS)
 
-    val resultSet: ResultSet = executeQuery("SHOW KEYSPACES;")
+    val keyspaceResults: ResultSet = executeQuery("SHOW KEYSPACES;")
     var rowCount = 0
-    while (resultSet.next()) {
+    while (keyspaceResults.next()) {
       rowCount++
     }
 
     assertEquals(2, rowCount)
+
+    val txnIsoLevelResults = executeQuery("SELECT @@global.transaction_ISOLATION;")
+    txnIsoLevelResults.next()
+    val actualTransactionIsolationLevel = txnIsoLevelResults.getString(1)
+    assertEquals("READ-COMMITTED", actualTransactionIsolationLevel)
   }
 
   private fun executeQuery(query: String): ResultSet {

misk-vitess-database-gradle-plugin/src/test/resources/vitess-database-plugin-test/build.gradle.kts

@@ -1,4 +1,5 @@
 import misk.vitess.gradle.StartVitessDatabaseTask
+import misk.vitess.testing.TransactionIsolationLevel
 
 plugins {
   id("com.squareup.misk.vitess.vitess-database")
@@ -9,4 +10,5 @@ val startVitessDatabase = tasks.named("startVitessDatabase", StartVitessDatabase
   lintSchema.set(true)
   port.set(31503)
   schemaDir.set("filesystem:${layout.projectDirectory.dir("src/main/resources/vitess/schema")}")
+  transactionIsolationLevel.set(TransactionIsolationLevel.READ_COMMITTED)
 }