Skip to content

Commit b0f0cb2

Browse files
phauslerphausler
authored
Audit and correct AsyncSequence Sendability (#41253)
* Audit and correct AsyncSequence Sendability * Annotate availability on Sendable conformances for AsyncSequences * Fix typo of Sendable.Iterator * Correct tests to use sendable builders * Add @preconcurrency annotations and fixup one remaining missing @sendable case * Move preconcurrency to correct syntactical position * Use unchecked Sendable conditional conformance instead of marking all internals as preconcurrency and sendable * Use unchecked Sendable conditional conformance instead of marking all internals as preconcurrency and sendable for dropWhile * claim ABI changes for marking of @preconcurrency as expected
1 parent 72c7284 commit b0f0cb2

16 files changed

+191
-15
lines changed

‎stdlib/public/Concurrency/AsyncCompactMapSequence.swift‎

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,10 @@ extension AsyncSequence {
4545
/// same or of a different type.
4646
/// - Returns: An asynchronous sequence that contains, in order, the
4747
/// non-`nil` elements produced by the `transform` closure.
48+
@preconcurrency
4849
@inlinable
4950
public __consuming func compactMap<ElementOfResult>(
50-
_ transform: @escaping (Element) async -> ElementOfResult?
51+
_ transform: @Sendable @escaping (Element) async -> ElementOfResult?
5152
) -> AsyncCompactMapSequence<Self, ElementOfResult> {
5253
return AsyncCompactMapSequence(self, transform: transform)
5354
}
@@ -129,3 +130,15 @@ extension AsyncCompactMapSequence: AsyncSequence {
129130
return Iterator(base.makeAsyncIterator(), transform: transform)
130131
}
131132
}
133+
134+
@available(SwiftStdlib 5.1, *)
135+
extension AsyncCompactMapSequence: @unchecked Sendable
136+
where Base: Sendable,
137+
Base.Element: Sendable,
138+
ElementOfResult: Sendable { }
139+
140+
@available(SwiftStdlib 5.1, *)
141+
extension AsyncCompactMapSequence.Iterator: @unchecked Sendable
142+
where Base.AsyncIterator: Sendable,
143+
Base.Element: Sendable,
144+
ElementOfResult: Sendable { }

‎stdlib/public/Concurrency/AsyncDropFirstSequence.swift‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,3 +135,13 @@ extension AsyncDropFirstSequence {
135135
return AsyncDropFirstSequence(base, dropping: self.count + count)
136136
}
137137
}
138+
139+
@available(SwiftStdlib 5.1, *)
140+
extension AsyncDropFirstSequence: Sendable
141+
where Base: Sendable,
142+
Base.Element: Sendable { }
143+
144+
@available(SwiftStdlib 5.1, *)
145+
extension AsyncDropFirstSequence.Iterator: Sendable
146+
where Base.AsyncIterator: Sendable,
147+
Base.Element: Sendable { }

‎stdlib/public/Concurrency/AsyncDropWhileSequence.swift‎

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,10 @@ extension AsyncSequence {
4141
/// modified sequence.
4242
/// - Returns: An asynchronous sequence that skips over values from the
4343
/// base sequence until the provided closure returns `false`.
44+
@preconcurrency
4445
@inlinable
4546
public __consuming func drop(
46-
while predicate: @escaping (Element) async -> Bool
47+
while predicate: @Sendable @escaping (Element) async -> Bool
4748
) -> AsyncDropWhileSequence<Self> {
4849
AsyncDropWhileSequence(self, predicate: predicate)
4950
}
@@ -127,3 +128,13 @@ extension AsyncDropWhileSequence: AsyncSequence {
127128
return Iterator(base.makeAsyncIterator(), predicate: predicate)
128129
}
129130
}
131+
132+
@available(SwiftStdlib 5.1, *)
133+
extension AsyncDropWhileSequence: @unchecked Sendable
134+
where Base: Sendable,
135+
Base.Element: Sendable { }
136+
137+
@available(SwiftStdlib 5.1, *)
138+
extension AsyncDropWhileSequence.Iterator: @unchecked Sendable
139+
where Base.AsyncIterator: Sendable,
140+
Base.Element: Sendable { }

‎stdlib/public/Concurrency/AsyncFilterSequence.swift‎

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,10 @@ extension AsyncSequence {
3333
/// that indicates whether to include the element in the filtered sequence.
3434
/// - Returns: An asynchronous sequence that contains, in order, the elements
3535
/// of the base sequence that satisfy the given predicate.
36+
@preconcurrency
3637
@inlinable
3738
public __consuming func filter(
38-
_ isIncluded: @escaping (Element) async -> Bool
39+
_ isIncluded: @Sendable @escaping (Element) async -> Bool
3940
) -> AsyncFilterSequence<Self> {
4041
return AsyncFilterSequence(self, isIncluded: isIncluded)
4142
}
@@ -113,3 +114,13 @@ extension AsyncFilterSequence: AsyncSequence {
113114
return Iterator(base.makeAsyncIterator(), isIncluded: isIncluded)
114115
}
115116
}
117+
118+
@available(SwiftStdlib 5.1, *)
119+
extension AsyncFilterSequence: @unchecked Sendable
120+
where Base: Sendable,
121+
Base.Element: Sendable { }
122+
123+
@available(SwiftStdlib 5.1, *)
124+
extension AsyncFilterSequence.Iterator: @unchecked Sendable
125+
where Base.AsyncIterator: Sendable,
126+
Base.Element: Sendable { }

‎stdlib/public/Concurrency/AsyncFlatMapSequence.swift‎

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,10 @@ extension AsyncSequence {
3939
/// of this sequence as its parameter and returns an `AsyncSequence`.
4040
/// - Returns: A single, flattened asynchronous sequence that contains all
4141
/// elements in all the asychronous sequences produced by `transform`.
42-
@inlinable
42+
@preconcurrency
43+
@inlinable
4344
public __consuming func flatMap<SegmentOfResult: AsyncSequence>(
44-
_ transform: @escaping (Element) async -> SegmentOfResult
45+
_ transform: @Sendable @escaping (Element) async -> SegmentOfResult
4546
) -> AsyncFlatMapSequence<Self, SegmentOfResult> {
4647
return AsyncFlatMapSequence(self, transform: transform)
4748
}
@@ -154,3 +155,18 @@ extension AsyncFlatMapSequence: AsyncSequence {
154155
return Iterator(base.makeAsyncIterator(), transform: transform)
155156
}
156157
}
158+
159+
@available(SwiftStdlib 5.1, *)
160+
extension AsyncFlatMapSequence: @unchecked Sendable
161+
where Base: Sendable,
162+
Base.Element: Sendable,
163+
SegmentOfResult: Sendable,
164+
SegmentOfResult.Element: Sendable { }
165+
166+
@available(SwiftStdlib 5.1, *)
167+
extension AsyncFlatMapSequence.Iterator: @unchecked Sendable
168+
where Base.AsyncIterator: Sendable,
169+
Base.Element: Sendable,
170+
SegmentOfResult: Sendable,
171+
SegmentOfResult.Element: Sendable,
172+
SegmentOfResult.AsyncIterator: Sendable { }

‎stdlib/public/Concurrency/AsyncMapSequence.swift‎

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,10 @@ extension AsyncSequence {
4343
/// same or of a different type.
4444
/// - Returns: An asynchronous sequence that contains, in order, the elements
4545
/// produced by the `transform` closure.
46+
@preconcurrency
4647
@inlinable
4748
public __consuming func map<Transformed>(
48-
_ transform: @escaping (Element) async -> Transformed
49+
_ transform: @Sendable @escaping (Element) async -> Transformed
4950
) -> AsyncMapSequence<Self, Transformed> {
5051
return AsyncMapSequence(self, transform: transform)
5152
}
@@ -117,3 +118,15 @@ extension AsyncMapSequence: AsyncSequence {
117118
return Iterator(base.makeAsyncIterator(), transform: transform)
118119
}
119120
}
121+
122+
@available(SwiftStdlib 5.1, *)
123+
extension AsyncMapSequence: @unchecked Sendable
124+
where Base: Sendable,
125+
Base.Element: Sendable,
126+
Transformed: Sendable { }
127+
128+
@available(SwiftStdlib 5.1, *)
129+
extension AsyncMapSequence.Iterator: @unchecked Sendable
130+
where Base.AsyncIterator: Sendable,
131+
Base.Element: Sendable,
132+
Transformed: Sendable { }

‎stdlib/public/Concurrency/AsyncPrefixSequence.swift‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,3 +109,13 @@ extension AsyncPrefixSequence: AsyncSequence {
109109
return Iterator(base.makeAsyncIterator(), count: count)
110110
}
111111
}
112+
113+
@available(SwiftStdlib 5.1, *)
114+
extension AsyncPrefixSequence: Sendable
115+
where Base: Sendable,
116+
Base.Element: Sendable { }
117+
118+
@available(SwiftStdlib 5.1, *)
119+
extension AsyncPrefixSequence.Iterator: Sendable
120+
where Base.AsyncIterator: Sendable,
121+
Base.Element: Sendable { }

‎stdlib/public/Concurrency/AsyncPrefixWhileSequence.swift‎

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,10 @@ extension AsyncSequence {
3838
/// included in the modified sequence.
3939
/// - Returns: An asynchronous sequence of the initial, consecutive
4040
/// elements that satisfy `predicate`.
41+
@preconcurrency
4142
@inlinable
4243
public __consuming func prefix(
43-
while predicate: @escaping (Element) async -> Bool
44+
while predicate: @Sendable @escaping (Element) async -> Bool
4445
) rethrows -> AsyncPrefixWhileSequence<Self> {
4546
return AsyncPrefixWhileSequence(self, predicate: predicate)
4647
}
@@ -120,3 +121,13 @@ extension AsyncPrefixWhileSequence: AsyncSequence {
120121
return Iterator(base.makeAsyncIterator(), predicate: predicate)
121122
}
122123
}
124+
125+
@available(SwiftStdlib 5.1, *)
126+
extension AsyncPrefixWhileSequence: @unchecked Sendable
127+
where Base: Sendable,
128+
Base.Element: Sendable { }
129+
130+
@available(SwiftStdlib 5.1, *)
131+
extension AsyncPrefixWhileSequence.Iterator: @unchecked Sendable
132+
where Base.AsyncIterator: Sendable,
133+
Base.Element: Sendable { }

‎stdlib/public/Concurrency/AsyncThrowingCompactMapSequence.swift‎

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,10 @@ extension AsyncSequence {
5757
/// non-`nil` elements produced by the `transform` closure. The sequence
5858
/// ends either when the base sequence ends or when `transform` throws an
5959
/// error.
60+
@preconcurrency
6061
@inlinable
6162
public __consuming func compactMap<ElementOfResult>(
62-
_ transform: @escaping (Element) async throws -> ElementOfResult?
63+
_ transform: @Sendable @escaping (Element) async throws -> ElementOfResult?
6364
) -> AsyncThrowingCompactMapSequence<Self, ElementOfResult> {
6465
return AsyncThrowingCompactMapSequence(self, transform: transform)
6566
}
@@ -151,3 +152,14 @@ extension AsyncThrowingCompactMapSequence: AsyncSequence {
151152
return Iterator(base.makeAsyncIterator(), transform: transform)
152153
}
153154
}
155+
156+
@available(SwiftStdlib 5.1, *)
157+
extension AsyncThrowingCompactMapSequence: @unchecked Sendable
158+
where Base: Sendable,
159+
Base.Element: Sendable { }
160+
161+
@available(SwiftStdlib 5.1, *)
162+
extension AsyncThrowingCompactMapSequence.Iterator: @unchecked Sendable
163+
where Base.AsyncIterator: Sendable,
164+
Base.Element: Sendable { }
165+

‎stdlib/public/Concurrency/AsyncThrowingDropWhileSequence.swift‎

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,10 @@ extension AsyncSequence {
5454
/// element from the modified sequence.
5555
/// - Returns: An asynchronous sequence that skips over values until the
5656
/// provided closure returns `false` or throws an error.
57+
@preconcurrency
5758
@inlinable
5859
public __consuming func drop(
59-
while predicate: @escaping (Element) async throws -> Bool
60+
while predicate: @Sendable @escaping (Element) async throws -> Bool
6061
) -> AsyncThrowingDropWhileSequence<Self> {
6162
AsyncThrowingDropWhileSequence(self, predicate: predicate)
6263
}
@@ -154,3 +155,13 @@ extension AsyncThrowingDropWhileSequence: AsyncSequence {
154155
return Iterator(base.makeAsyncIterator(), predicate: predicate)
155156
}
156157
}
158+
159+
@available(SwiftStdlib 5.1, *)
160+
extension AsyncThrowingDropWhileSequence: @unchecked Sendable
161+
where Base: Sendable,
162+
Base.Element: Sendable { }
163+
164+
@available(SwiftStdlib 5.1, *)
165+
extension AsyncThrowingDropWhileSequence.Iterator: @unchecked Sendable
166+
where Base.AsyncIterator: Sendable,
167+
Base.Element: Sendable { }

0 commit comments

Comments
 (0)