Add internal _send function that returns RESPToken (#91)

Author: fabianfett

Sources/Valkey/Connection/ValkeyConnection.swift

@@ -115,15 +115,20 @@ public final actor ValkeyConnection: ValkeyConnectionProtocol, Sendable {
     /// - Returns: The command response as defined in the ValkeyCommand
     @inlinable
     public func send<Command: ValkeyCommand>(command: Command) async throws -> Command.Response {
+        let result = try await self._send(command: command)
+        return try .init(fromRESP: result)
+    }
+
+    @inlinable
+    func _send<Command: ValkeyCommand>(command: Command) async throws -> RESPToken {
         let requestID = Self.requestIDGenerator.next()
         return try await withTaskCancellationHandler {
             if Task.isCancelled {
                 throw ValkeyClientError(.cancelled)
             }
-            let result = try await withCheckedThrowingContinuation { continuation in
+            return try await withCheckedThrowingContinuation { continuation in
                 self.channelHandler.write(command: command, continuation: continuation, requestID: requestID)
             }
-            return try .init(fromRESP: result)
         } onCancel: {
             self.cancel(requestID: requestID)
         }

Sources/Valkey/ValkeyClient.swift

@@ -155,9 +155,16 @@ extension ValkeyClient {
 
 /// Extend ValkeyClient so we can call commands directly from it
 extension ValkeyClient: ValkeyConnectionProtocol {
+    @inlinable
     public func send<Command: ValkeyCommand>(command: Command) async throws -> Command.Response {
-        try await self.withConnection {
-            try await $0.send(command: command)
+        let token = try await self._send(command)
+        return try Command.Response(fromRESP: token)
+    }
+
+    @inlinable
+    func _send<Command: ValkeyCommand>(_ command: Command) async throws -> RESPToken {
+        try await self.withConnection { connection in
+            try await connection._send(command: command)
         }
     }
 }