override fun restoreThreadContext()

in exposed-core/src/main/kotlin/org/jetbrains/exposed/v1/core/transactions/suspend/TransactionContextElement.kt [49:82]
28 lines of code
5 McCabe index (conditional complexity)

    override fun restoreThreadContext(context: CoroutineContext, oldState: Transaction?) {
        if (oldState == null) return

        // Check if stack is empty - this can happen if withThreadLocalTransaction already popped it
        val currentTransaction = ThreadLocalTransactionsStack.getTransactionOrNull()
        if (currentTransaction == null) {
            exposedLogger.warn(
                "restoreThreadContext called for transaction ${transaction.transactionId} but stack is already empty. " +
                    "This is likely because withThreadLocalTransaction already popped the transaction. Skipping restore."
            )
            return
        }

        // Check if the top of the stack is actually our transaction before popping
        if (currentTransaction.transactionId != transaction.transactionId) {
            exposedLogger.warn(
                "restoreThreadContext called for transaction ${transaction.transactionId} but top of stack is ${currentTransaction.transactionId}. " +
                    "Skipping restore to avoid corrupting the stack."
            )
            return
        }

        // Safe to pop
        val poppedTransaction = ThreadLocalTransactionsStack.popTransaction()
        if (poppedTransaction.transactionId != transaction.transactionId) {
            exposedLogger.warn(
                "The current thread local stack of transactions had a transaction ${poppedTransaction.transactionId} on the top. " +
                    "But it differs from the transaction ${transaction.transactionId} in the coroutine context. " +
                    "Normally it should not happen because the coroutine adds its transaction into the thread " +
                    "local stack on the start and removes it on the end of execution"
            )
            exposedLogger.warn("Popped transaction ${poppedTransaction.transactionId} doesn't match expected ${transaction.transactionId}")
        }
    }