Add DispatchWorkItem to async and sync methods

space-code · May 27, 2023 · 4897a73 · 4897a73
2 parents 7176c3c + ff7aa2a
commit 4897a73
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Sources/Concurrency/DispatchQueue/IDispatchQueue.swift b/Sources/Concurrency/DispatchQueue/IDispatchQueue.swift
@@ -81,6 +81,17 @@ public protocol IDispatchQueue: AnyObject {
     /// - SeeAlso: `async(execute:)`
     ///
     func sync(execute block: () -> Void)
+
+    /// Schedules a work item for immediate execution, and returns immediately.
+    ///
+    /// - Parameter execute: The work item containing the task to execute.
+    /// For information on how to create this work item.
+    func async(execute workItem: DispatchWorkItem)
+
+    /// Submits a block for synchronous execution on this queue.
+    ///
+    /// - Parameter workItem: The work item to be invoked on the queue.
+    func sync(execute workItem: DispatchWorkItem)
 }
 
 public extension IDispatchQueue {

diff --git a/Sources/TestConcurrency/TestDispatchQueue.swift b/Sources/TestConcurrency/TestDispatchQueue.swift
@@ -39,4 +39,12 @@ extension TestDispatchQueue: IDispatchQueue {
     public func async(execute work: @escaping @convention(block) () -> Void) {
         work()
     }
+
+    public func async(execute workItem: DispatchWorkItem) {
+        workItem.perform()
+    }
+
+    public func sync(execute workItem: DispatchWorkItem) {
+        workItem.perform()
+    }
 }