Swift AsyncStream迭代为何需等待流结束？执行逻辑存疑

你的预期没错，但代码实现方式不符合AsyncStream的运行机制，才导致输出和预期不一致。

问题根源 #

你在bar()的AsyncStream启动闭包里使用了阻塞线程的sleep(1)，而且整个循环是同步执行的：

• AsyncStream的启动闭包会在for await开始遍历流的瞬间执行，且默认在当前调用线程上同步运行。
• sleep(1)会直接卡住当前线程，闭包里的循环要把3次yield全部执行完，才会把线程控制权交还给for await循环，所以你会先看到三次Yielding...，再一次性输出所有yield的值。

AsyncStream的核心工作逻辑 #

1. 当你创建AsyncStream实例时，启动闭包并不会立刻执行，直到第一个迭代器被创建（也就是for await开始遍历的时候）。
2. 启动闭包默认在调用迭代器的线程上同步执行，如果闭包里有阻塞操作，会直接阻塞消费流的线程，直到闭包执行完毕。
3. 只有当闭包主动交出线程控制权（比如通过异步操作）时，yield的值才会被for await实时消费。

修正后的代码 #

把生产值的逻辑放到异步Task里，避免阻塞消费线程，就能得到你预期的输出：

func foo() async throws {
    print("Start")
    for await data in bar() {
        print(data)
    }
    print("End")
}

private func bar() -> AsyncStream<String> {
    return AsyncStream { continuation in
        print("Stream Started")
        // 用Task异步执行生产逻辑，不阻塞当前线程
        Task {
            for count in 0...2 {
                // 用Swift并发框架的sleep，不会阻塞线程
                try? await Task.sleep(nanoseconds: 1_000_000_000)
                print("Yielding...")
                continuation.yield("\(count)")
            }
            continuation.finish()
        }
    }
}

执行这段代码，输出就会和你预期一致：

Start
Stream Started
Yielding...
0
Yielding...
1
Yielding...
2
End

关键注意点 #

• 不要在AsyncStream的启动闭包里做同步阻塞操作，否则会导致所有值生产完毕后才被批量消费。
• 如果需要异步生成值，必须把生产逻辑放到独立的Task、DispatchQueue或者其他异步执行环境中，让yield能实时把值推送给消费者。

内容的提问来源于stack exchange，提问作者sbarow