connection

Author: purocean

build.gradle

@@ -10,7 +10,7 @@ buildscript {
 }
 
 group 'info.purocean'
-version '1.0'
+version '1.1.0'
 
 apply plugin: 'kotlin'
 apply plugin: 'maven-publish'

src/main/kotlin/info/purocean/dbmigration/CodeMigration.kt

@@ -1,5 +1,7 @@
 package info.purocean.dbmigration
 
+import java.sql.Connection
+
 interface CodeMigration <in T> {
-    fun run(context: T)
-}
+    fun run(context: T, connection: Connection)
+}

src/main/kotlin/info/purocean/dbmigration/DB.kt

@@ -88,6 +88,10 @@ class DB(dbUrl: String, dbUsername: String, dbPassword: String) {
         this.conn.close()
     }
 
+    fun getConnection (): Connection {
+        return this.conn
+    }
+
     private fun getStmt (): Statement {
         return this.conn.createStatement()
     }

src/main/kotlin/info/purocean/dbmigration/Migrate.kt

@@ -68,7 +68,7 @@ class Migrate(dbUrl: String, dbUsername: String, dbPassword: String, private var
 
             val obj = cls.newInstance()
 
-            cls.getMethod("run", Object::class.java).invoke(obj, context)
+            cls.getMethod("run", Object::class.java).invoke(obj, context, this.db.getConnection())
         }
 
         this.db.writeRecord(migration)