mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
8578584f61
* Implementations for operators and async functions on AsyncSequence per SE-0298 * Add trailing new lines to AsyncSequence adopter files * Remove extra // from preambles * Transition sequence properties and sequence initializers to internal * Rename Upstream to Base * Rename "Failable" asynchronous sequences to "Throwing" * Make iteration consistent with closure lifetimes and track failure via state * Use for await syntax for async function extensions on AsyncSequence * Work-around rethrows calculations being incorrect for conformance + closure rethrowing * Update testing names to reflect Throwing versus Failable * Incorperate runAsyncAndBlock into the test function for suite calls * Remove @frozen from AsyncSequence operators * Back out reduce family functions for now due to IRGen bug * Additional corrections for throwing cases and fix up potential linux crasher * Add a upstream and throwing tests and correct flatMap's throwing behavior * Unify parameter names to count * Add a more helpful precondition on prefix * Add tests for throwing cases of non closure async functions on AsyncSequence * Correct missing count transition in dropFirst * Correct missing , in assertion * [NFC] add commentary on optimization in dropFirst on dropFirst * Disable collect tests for non macOS platforms for now * Disable all async sequence tests for non macOS platforms for now
87 lines
2.3 KiB
Swift
87 lines
2.3 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2021 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import Swift
|
|
|
|
extension AsyncSequence {
|
|
@inlinable
|
|
public __consuming func prefix(
|
|
while predicate: @escaping (Element) async throws -> Bool
|
|
) rethrows -> AsyncThrowingPrefixWhileSequence<Self> {
|
|
return AsyncThrowingPrefixWhileSequence(self, predicate: predicate)
|
|
}
|
|
}
|
|
|
|
public struct AsyncThrowingPrefixWhileSequence<Base: AsyncSequence> {
|
|
@usableFromInline
|
|
let base: Base
|
|
|
|
@usableFromInline
|
|
let predicate: (Base.Element) async throws -> Bool
|
|
|
|
@usableFromInline
|
|
init(
|
|
_ base: Base,
|
|
predicate: @escaping (Base.Element) async throws -> Bool
|
|
) {
|
|
self.base = base
|
|
self.predicate = predicate
|
|
}
|
|
}
|
|
|
|
extension AsyncThrowingPrefixWhileSequence: AsyncSequence {
|
|
public typealias Element = Base.Element
|
|
public typealias AsyncIterator = Iterator
|
|
|
|
public struct Iterator: AsyncIteratorProtocol {
|
|
@usableFromInline
|
|
var predicateHasFailed = false
|
|
|
|
@usableFromInline
|
|
var baseIterator: Base.AsyncIterator
|
|
|
|
@usableFromInline
|
|
let predicate: (Base.Element) async throws -> Bool
|
|
|
|
@usableFromInline
|
|
init(
|
|
_ baseIterator: Base.AsyncIterator,
|
|
predicate: @escaping (Base.Element) async throws -> Bool
|
|
) {
|
|
self.baseIterator = baseIterator
|
|
self.predicate = predicate
|
|
}
|
|
|
|
@inlinable
|
|
public mutating func next() async throws -> Base.Element? {
|
|
if !predicateHasFailed, let nextElement = try await baseIterator.next() {
|
|
do {
|
|
if try await predicate(nextElement) {
|
|
return nextElement
|
|
} else {
|
|
predicateHasFailed = true
|
|
}
|
|
} catch {
|
|
predicateHasFailed = true
|
|
throw error
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|
|
@inlinable
|
|
public __consuming func makeAsyncIterator() -> Iterator {
|
|
return Iterator(base.makeAsyncIterator(), predicate: predicate)
|
|
}
|
|
}
|