Remove NIOAsyncChannel

Sources/Valkey/Connection/ValkeyConnection.swift

@@ -13,6 +13,7 @@
 //===----------------------------------------------------------------------===//
 
 import Logging
+import NIOConcurrencyHelpers
 import NIOCore
 import NIOPosix
 import NIOSSL
@@ -46,6 +47,7 @@ public struct ValkeyConnection: Sendable {
     let logger: Logger
     let channel: Channel
     let configuration: ValkeyClientConfiguration
+    let isClosed: NIOLockedValueBox<Bool>
 
     /// Initialize Client
     private init(
@@ -56,6 +58,7 @@ public struct ValkeyConnection: Sendable {
         self.channel = channel
         self.configuration = configuration
         self.logger = logger
+        self.isClosed = .init(false)
     }
 
     public static func connect(
@@ -70,6 +73,14 @@ public struct ValkeyConnection: Sendable {
         return connection
     }
 
+    public func close() -> EventLoopFuture<Void> {
+        if self.isClosed.withLockedValue({ $0 }) {
+            return channel.eventLoop.makeSucceededFuture(())
+        }
+        self.channel.close(mode: .all, promise: nil)
+        return self.channel.closeFuture
+    }
+
     @discardableResult public func send<Command: RESPCommand>(command: Command) async throws -> Command.Response {
         var encoder = RESPCommandEncoder()
         command.encode(into: &encoder)

Sources/Valkey/ValkeyClient.swift

@@ -70,6 +70,14 @@ extension ValkeyClient {
             eventLoopGroup: self.eventLoopGroup,
             logger: logger
         )
-        return try await operation(valkeyConnection)
+        let value: Value
+        do {
+            value = try await operation(valkeyConnection)
+        } catch {
+            try? await valkeyConnection.close().get()
+            throw error
+        }
+        try await valkeyConnection.close().get()
+        return value
     }
 }