Add public triggerForceShutdown to ConnectionPool (#572)

fabianfett · web-flow · commit 4ebc10cb4e8b · 2025-09-18T09:40:23.000Z
Add distinct method that we need in Valkey (valkey-io/valkey-swift#85)
diff --git 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 -

Original file line number	Diff line number	Diff line change
`@@ -283,12 +283,15 @@ public final class ConnectionPool<`
`283`	`283`	`await self.run(in: &taskGroup)`
`284`	`284`	`}`
`285`	`285`	`} onCancel: {`
`286`		`- let actions = self.stateBox.withLockedValue { state in`
`287`		`- state.stateMachine.triggerForceShutdown()`
`288`		`- }`
	`286`	`+ self.triggerForceShutdown()`
	`287`	`+ }`
	`288`	`+ }`
`289`	`289`
`290`		`- self.runStateMachineActions(actions)`
	`290`	`+ public func triggerForceShutdown() {`
	`291`	`+ let actions = self.stateBox.withLockedValue { state in`
	`292`	`+ state.stateMachine.triggerForceShutdown()`
`291`	`293`	`}`
	`294`	`+ self.runStateMachineActions(actions)`
`292`	`295`	`}`
`293`	`296`
`294`	`297`	`// MARK: - Private Methods -`