call onUnexpectedConnectionClose on closeFuture

swift-server · Jun 13, 2023 · 43d9445 · 43d9445
1 parent 2fd4db6
commit 43d9445
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Sources/RediStack/RedisConnectionPool.swift b/Sources/RediStack/RedisConnectionPool.swift
@@ -208,9 +208,8 @@ extension RedisConnectionPool {
             .map { connection in
                 // disallow subscriptions on all connections by default to enforce our management of PubSub state
                 connection.allowSubscriptions = false
-                connection.onUnexpectedClosure = { [weak self, weak connection] in
-                    guard let connection = connection else { return }
-                    self?.configuration.onUnexpectedConnectionClose?(connection)
+                connection.channel.closeFuture.whenComplete { _ in
+                    self.configuration.onUnexpectedConnectionClose?(connection)
                 }
                 return connection
             }