Add callback-oriented version of asyncPeriodically

Alkenso · Dec 15, 2024 · 345fe4c · 345fe4c
1 parent 8cc9e10
commit 345fe4c
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/Sources/SpellbookFoundation/Threading & Concurrency/Extensions - DispatchQueue.swift b/Sources/SpellbookFoundation/Threading & Concurrency/Extensions - DispatchQueue.swift
@@ -80,6 +80,23 @@ extension DispatchQueue {
         }
         schedule(firstRun: true)
     }
+
+    public func asyncPeriodically(
+        interval: TimeInterval,
+        immediately: Bool,
+        qos: DispatchQoS = .unspecified,
+        flags: DispatchWorkItemFlags = [],
+        execute: @escaping (@escaping () -> Void) -> Void
+    ) {
+        func schedule(firstRun: Bool) {
+            asyncAfter(delay: (firstRun && immediately) ? 0 : interval, qos: qos, flags: flags) {
+                execute {
+                    schedule(firstRun: false)
+                }
+            }
+        }
+        schedule(firstRun: true)
+    }
 }
 
 extension DispatchQueue {

diff --git a/Tests/SpellbookTests/Threading & Concurrency/DispatchQueueExtensionsTests.swift b/Tests/SpellbookTests/Threading & Concurrency/DispatchQueueExtensionsTests.swift
@@ -19,4 +19,21 @@ class DispatchQueueExtensionsTests: XCTestCase {
         XCTAssertEqual(count, limit)
         waitForExpectations()
     }
+
+    func test_asyncPeriodically_async() {
+        var count: Int = 0
+        let limit = 5
+        let exp = expectation(description: "Repeated action")
+        exp.expectedFulfillmentCount = limit
+        DispatchQueue.global().asyncPeriodically(interval: 0.01, immediately: true) {
+            count += 1
+            exp.fulfill()
+            if count < limit {
+                $0()
+            }
+        }
+        Thread.sleep(forTimeInterval: 0.1 * Self.waitRate)
+        XCTAssertEqual(count, limit)
+        waitForExpectations()
+    }
 }