add receive variant that returns async sequence of [UInt8]

PADL · Aug 2, 2024 · 68c6aa6 · 68c6aa6
1 parent f708ec3
commit 68c6aa6
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 6 deletions.
diff --git a/Examples/IORingDeviceSpy/IORingDeviceSpy.swift b/Examples/IORingDeviceSpy/IORingDeviceSpy.swift
@@ -30,12 +30,10 @@ public struct IORingDeviceSpy {
       exit(1)
     }
 
-    let blockSize: Int
-
-    if CommandLine.arguments.count > 2 {
-      blockSize = Int(CommandLine.arguments[2])!
+    let blockSize = if CommandLine.arguments.count > 2 {
+      Int(CommandLine.arguments[2])!
     } else {
-      blockSize = 1
+      1
     }
 
     let spy = try await IORingDeviceSpy(blockSize: blockSize)

diff --git a/Examples/IORingTCPEcho/IORingTCPEcho.swift b/Examples/IORingTCPEcho/IORingTCPEcho.swift
@@ -74,7 +74,7 @@ public struct IORingTCPEcho {
   func receiveSendEcho(client: Socket) async {
     do {
       repeat {
-        let data = try await client.receive(count: bufferSize)
+        let data: [UInt8] = try await client.receive(count: bufferSize)
         try await client.send(data)
       } while true
     } catch {

diff --git a/Sources/IORingUtils/Socket.swift b/Sources/IORingUtils/Socket.swift
@@ -348,6 +348,15 @@ public struct Socket: CustomStringConvertible, Equatable, Hashable, Sendable {
     )
   }
 
+  @IORingActor
+  public func receive(count: Int) async throws -> AnyAsyncSequence<[UInt8]> {
+    guard let fileHandle else { throw Errno.badFileDescriptor }
+    return try ring.receive(
+      count: count,
+      from: fileHandle
+    )
+  }
+
   @IORingActor
   public func send(_ data: [UInt8]) async throws {
     guard let fileHandle else { throw Errno.badFileDescriptor }