Add Transaction to lifecycle hooks for JetBrains#1530

rsromanowski · Jun 13, 2022 · 320dbcb · 320dbcb
1 parent b456bbe
commit 320dbcb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/exposed-core/src/main/kotlin/org/jetbrains/exposed/sql/Transaction.kt b/exposed-core/src/main/kotlin/org/jetbrains/exposed/sql/Transaction.kt
@@ -72,17 +72,17 @@ open class Transaction(private val transactionImpl: TransactionInterface) : User
         }
         transactionImpl.commit()
         userdata.clear()
-        globalInterceptors.forEach { it.afterCommit() }
-        interceptors.forEach { it.afterCommit() }
+        globalInterceptors.forEach { it.afterCommit(this) }
+        interceptors.forEach { it.afterCommit(this) }
         userdata.putAll(dataToStore)
     }
 
     override fun rollback() {
         globalInterceptors.forEach { it.beforeRollback(this) }
         interceptors.forEach { it.beforeRollback(this) }
         transactionImpl.rollback()
-        globalInterceptors.forEach { it.afterRollback() }
-        interceptors.forEach { it.afterRollback() }
+        globalInterceptors.forEach { it.afterRollback(this) }
+        interceptors.forEach { it.afterRollback(this) }
         userdata.clear()
     }
 

diff --git a/exposed-core/src/main/kotlin/org/jetbrains/exposed/sql/statements/StatementInterceptor.kt b/exposed-core/src/main/kotlin/org/jetbrains/exposed/sql/statements/StatementInterceptor.kt
@@ -9,10 +9,10 @@ interface StatementInterceptor {
     fun afterExecution(transaction: Transaction, contexts: List<StatementContext>, executedStatement: PreparedStatementApi) {}
 
     fun beforeCommit(transaction: Transaction) {}
-    fun afterCommit() {}
+    fun afterCommit(transaction: Transaction) {}
 
     fun beforeRollback(transaction: Transaction) {}
-    fun afterRollback() {}
+    fun afterRollback(transaction: Transaction) {}
 
     fun keepUserDataInTransactionStoreOnCommit(userData: Map<Key<*>, Any?>): Map<Key<*>, Any?> = emptyMap()
 }