StatefulSagaBuilder

data class PaymentState(
    val paymentAmount: Long = 0L,
    val escrowCaptured: Boolean = false
)

val saga = sagaExecutor<OrderContext, OrderResult>(PaymentState()) {
    step("take-payment") { context ->
        val payment = paymentService.takePayment(context.amount)
        updateState { it.copy(paymentAmount = payment.amount) }
        OrderResult(payment.id)
    }
    compensate {
        // Smart compensation using state from later steps
        if (state.escrowCaptured) {
            paymentService.refund(result.paymentId)
        } else {
            paymentService.returnEscrow()
        }
    }

    step("capture-escrow") { context ->
        paymentService.captureEscrow()
        updateState { it.copy(escrowCaptured = true) }
        result!! // Pass through previous result
    }
    compensate {
        paymentService.refund(state.paymentAmount)
    }
}