diff --git a/Sources/ConnectionPoolModule/ConnectionPool.swift b/Sources/ConnectionPoolModule/ConnectionPool.swift index ee72337d..40d52a5a 100644 --- a/Sources/ConnectionPoolModule/ConnectionPool.swift +++ b/Sources/ConnectionPoolModule/ConnectionPool.swift @@ -283,12 +283,15 @@ public final class ConnectionPool< await self.run(in: &taskGroup) } } onCancel: { - let actions = self.stateBox.withLockedValue { state in - state.stateMachine.triggerForceShutdown() - } + self.triggerForceShutdown() + } + } - self.runStateMachineActions(actions) + public func triggerForceShutdown() { + let actions = self.stateBox.withLockedValue { state in + state.stateMachine.triggerForceShutdown() } + self.runStateMachineActions(actions) } // MARK: - Private Methods -