diff --git a/node/src/main/kotlin/net/corda/node/internal/RpcExceptionHandlingProxy.kt b/node/src/main/kotlin/net/corda/node/internal/RpcExceptionHandlingProxy.kt index f8434b9b52..60f285ba2b 100644 --- a/node/src/main/kotlin/net/corda/node/internal/RpcExceptionHandlingProxy.kt +++ b/node/src/main/kotlin/net/corda/node/internal/RpcExceptionHandlingProxy.kt @@ -5,6 +5,7 @@ import net.corda.core.contracts.ContractState import net.corda.core.crypto.SecureHash import net.corda.core.doOnError import net.corda.core.flows.FlowLogic +import net.corda.core.flows.StateMachineRunId import net.corda.core.identity.AbstractParty import net.corda.core.identity.CordaX500Name import net.corda.core.internal.concurrent.doOnError @@ -128,6 +129,8 @@ class RpcExceptionHandlingProxy(private val delegate: SecureCordaRPCOps) : Corda override fun isFlowsDrainingModeEnabled() = wrap(delegate::isFlowsDrainingModeEnabled) + override fun killFlow(id: StateMachineRunId) = wrap { delegate.killFlow(id) } + override fun shutdown() = wrap(delegate::shutdown) private fun wrap(call: () -> RESULT): RESULT {