Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Concurrency #21

Merged
merged 18 commits into from
Oct 26, 2024
Merged
Show file tree
Hide file tree
Changes from 6 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
60 changes: 47 additions & 13 deletions Package.resolved
Original file line number Diff line number Diff line change
@@ -1,16 +1,50 @@
{
"object": {
"pins": [
{
"package": "swift-nio",
"repositoryURL": "https://github.com/apple/swift-nio.git",
"state": {
"branch": null,
"revision": "51c3fc2e4a0fcdf4a25089b288dd65b73df1b0ef",
"version": "2.37.0"
}
"pins" : [
{
"identity" : "async-collections",
"kind" : "remoteSourceControl",
"location" : "https://github.com/adam-fowler/async-collections",
"state" : {
"revision" : "726af96095a19df6b8053ddbaed0a727aa70ccb2",
"version" : "0.1.0"
}
]
},
"version": 1
},
{
"identity" : "swift-algorithms",
"kind" : "remoteSourceControl",
"location" : "https://github.com/apple/swift-algorithms.git",
"state" : {
"revision" : "f6919dfc309e7f1b56224378b11e28bab5bccc42",
"version" : "1.2.0"
}
},
{
"identity" : "swift-atomics",
"kind" : "remoteSourceControl",
"location" : "https://github.com/apple/swift-atomics.git",
"state" : {
"revision" : "cd142fd2f64be2100422d658e7411e39489da985",
"version" : "1.2.0"
}
},
{
"identity" : "swift-collections",
"kind" : "remoteSourceControl",
"location" : "https://github.com/apple/swift-collections.git",
"state" : {
"revision" : "671108c96644956dddcd89dd59c203dcdb36cec7",
"version" : "1.1.4"
}
},
{
"identity" : "swift-numerics",
"kind" : "remoteSourceControl",
"location" : "https://github.com/apple/swift-numerics.git",
"state" : {
"revision" : "0a5bc04095a675662cf24757cc0640aa2204253b",
"version" : "1.0.2"
}
}
],
"version" : 2
}
17 changes: 12 additions & 5 deletions Package.swift
Original file line number Diff line number Diff line change
@@ -1,19 +1,26 @@
// swift-tools-version:5.0
// swift-tools-version:5.10
// The swift-tools-version declares the minimum version of Swift required to build this package.

import PackageDescription

let package = Package(
name: "DataLoader",
platforms: [.macOS(.v12), .iOS(.v15), .tvOS(.v15), .watchOS(.v8)],
products: [
.library(name: "DataLoader", targets: ["DataLoader"]),
],
dependencies: [
.package(url: "https://github.com/apple/swift-nio.git", from: "2.0.0"),
.package(url: "https://github.com/apple/swift-algorithms.git", from: "1.0.0"),
.package(url: "https://github.com/adam-fowler/async-collections", from: "0.0.1"),
],
targets: [
.target(name: "DataLoader", dependencies: ["NIO", "NIOConcurrencyHelpers"]),
.target(
name: "DataLoader",
dependencies: [
.product(name: "Algorithms", package: "swift-algorithms"),
.product(name: "AsyncCollections", package: "async-collections"),
]
),
.testTarget(name: "DataLoaderTests", dependencies: ["DataLoader"]),
],
swiftLanguageVersions: [.v5]
]
)
55 changes: 55 additions & 0 deletions Sources/DataLoader/Channel/Channel.swift
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
public actor Channel<Success: Sendable, Failure: Error>: Sendable {
private var state = State<Success, Failure>()
}

public extension Channel {
@discardableResult
func fulfill(_ value: Success) async -> Bool {
if await state.result == nil {
await state.setResult(result: value)

for waiters in await state.waiters {
waiters.resume(returning: value)
}

await state.removeAllWaiters()

return false
}

return true
}

@discardableResult
func fail(_ failure: Failure) async -> Bool {
if await state.failure == nil {
await state.setFailure(failure: failure)

for waiters in await state.waiters {
waiters.resume(throwing: failure)
}

await state.removeAllWaiters()

return false
}

return true
}

var value: Success {
get async throws {
try await withCheckedThrowingContinuation { continuation in
Task {
if let result = await state.result {
continuation.resume(returning: result)
} else if let failure = await self.state.failure {
continuation.resume(throwing: failure)
} else {
await state.appendWaiters(waiters: continuation)
}
}
}
}
}
}
25 changes: 25 additions & 0 deletions Sources/DataLoader/Channel/State.swift
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
typealias Waiter<Success, Failure> = CheckedContinuation<Success, Error>

actor State<Success, Failure> {
var waiters = [Waiter<Success, Failure>]()
var result: Success?
var failure: Failure?
}

extension State {
func setResult(result: Success) {
self.result = result
}

func setFailure(failure: Failure) {
self.failure = failure
}

func appendWaiters(waiters: Waiter<Success, Failure>...) {
self.waiters.append(contentsOf: waiters)
}

func removeAllWaiters() {
waiters.removeAll()
}
}
Loading
Loading