@MainActor类中异步任务为何未在主线程执行？

核心原因 #

你遇到的问题本质是Actor上下文不会自动传递给非隔离的扩展方法：

• asyncTest 是 @MainActor 类的实例方法，默认继承类的Actor隔离，所以运行在主线程。
• asyncMap 是 Sequence 的扩展方法，没有标注任何Actor隔离修饰符，属于nonisolated（非隔离）方法。当在MainActor上下文中调用非隔离异步方法时，系统会将其调度到任意后台线程执行，不会继承调用者的Actor上下文。

解决方案 #

有两种常用方式让asyncMap继承调用者的Actor上下文：

方案1：给扩展方法显式标注目标Actor

如果你希望所有Sequence的asyncMap都强制在主线程执行，可以直接给扩展方法加上@MainActor修饰：

import PlaygroundSupport
import Foundation

@MainActor
class AsyncTest {
    nonisolated init() {}
    
    func asyncTest() async {
        print(#function, "isMainThread:", Thread.isMainThread)
        
        let test: [String] = ["Hello", "World"]
        
        await test.asyncMap({ string in
            return string.uppercased()
        })
    }
}

extension Sequence {
    // 标注@MainActor，强制方法在主线程执行
    @MainActor
    func asyncMap<T>(
        _ transform: (Element) async throws -> T
    ) async rethrows -> [T] {
        print(#function, "isMainThread:", Thread.isMainThread)
        
        var values = [T]()
        
        for element in self {
            try await values.append(transform(element))
        }
        
        return values
    }
}

let asyncTest = AsyncTest()
Task {
    await asyncTest.asyncTest()
}

PlaygroundPage.current.needsIndefiniteExecution = true

输出会变成：

asyncTest() isMainThread: true
asyncMap(_:) isMainThread: true

方案2：让方法继承调用者的Actor上下文

如果你希望asyncMap能灵活继承调用者的任意Actor上下文（比如从其他Actor调用时，跑在对应Actor的线程），可以使用Swift的实验性特性@_inheritActorContext（注意下划线开头表示是未正式稳定的API）：

extension Sequence {
    // 继承调用者的Actor上下文
    @_inheritActorContext
    func asyncMap<T>(
        _ transform: (Element) async throws -> T
    ) async rethrows -> [T] {
        print(#function, "isMainThread:", Thread.isMainThread)
        
        var values = [T]()
        
        for element in self {
            try await values.append(transform(element))
        }
        
        return values
    }
}

这样无论从哪个Actor上下文调用asyncMap，方法都会运行在对应Actor的执行上下文里。

补充说明 #

Actor隔离的作用范围仅针对：

• Actor类型本身的成员方法/属性
• 显式标注了Actor修饰符的扩展方法

默认情况下，针对非Actor类型的扩展方法都是非隔离的，不会自动继承调用者的Actor上下文，需要手动通过上述方式指定。

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