Swift中能否扩展函数？寻求简化异步函数调用的实现方案

要实现像doSomethingAsync().makeItAsync()这样的链式调用，核心在于Swift中函数作为一等公民，无法直接扩展泛化的函数类型，但可以通过以下几种方式达成需求：

方案1：扩展具体的异步函数类型 #

针对你需要的特定异步函数签名（比如无参无返回值的() async -> Void）直接扩展：

func doSomethingAsync() async -> Void {
    print("执行异步任务")
}

// 扩展无参无返回值的异步函数类型
extension () async -> Void {
    func makeItAsync() {
        Task {
            await self()
        }
    }
}

// 直接调用
doSomethingAsync().makeItAsync()

优缺点 #

• 优点：实现简单，调用方式完全符合预期
• 缺点：扩展性差，每个不同的函数签名（比如() async -> Int、(String) async -> Bool）都需要单独编写扩展

方案2：用自定义结构体封装异步函数（通用方案） #

如果需要支持多种类型的异步函数，可以封装一个通用结构体来包裹函数，再扩展这个结构体：

// 通用异步函数封装结构体
struct AsyncFunc<Input, Output> {
    private let function: (Input) async -> Output
    
    // 带参数的初始化方法
    init(_ function: @escaping (Input) async -> Output) {
        self.function = function
    }
    
    // 无参函数的简化初始化（Input为Void）
    init(_ function: @escaping () async -> Output) where Input == Void {
        self.function = { _ in await function() }
    }
}

// 扩展结构体，添加同步触发异步任务的方法
extension AsyncFunc {
    func makeItAsync(with input: Input = ()) {
        Task {
            await self.function(input)
        }
    }
}

// 使用示例
// 无参异步函数
let doSomething = AsyncFunc(doSomethingAsync)
doSomething.makeItAsync()

// 带参数的异步函数
func fetchUser(id: Int) async -> String {
    return "用户ID：\(id)"
}

let fetchUserFunc = AsyncFunc(fetchUser)
fetchUserFunc.makeItAsync(with: 1001)

优缺点 #

• 优点：完全通用，支持任意参数和返回值的异步函数
• 缺点：需要先将函数封装为AsyncFunc实例，调用前多一步初始化操作

方案3：利用协议扩展（半通用） #

定义一个协议，让特定的异步函数类型遵守该协议，再扩展协议添加方法：

// 定义异步可调用协议
protocol AsyncInvocable {
    associatedtype Output
    func callAsFunction() async -> Output
}

// 让不同签名的异步函数类型遵守协议
extension () async -> Void: AsyncInvocable {}
extension () async -> Int: AsyncInvocable {}
extension (String) async -> Bool: AsyncInvocable {}

// 扩展协议，添加makeItAsync方法
extension AsyncInvocable {
    func makeItAsync() {
        Task {
            await self()
        }
    }
}

// 调用示例
doSomethingAsync().makeItAsync()

func getRandomNumber() async -> Int {
    return Int.random(in: 1...100)
}

getRandomNumber().makeItAsync()

优缺点 #

• 优点：调用方式简洁，支持你想要的链式写法
• 缺点：需要手动让每种需要的函数类型遵守协议，无法自动覆盖所有可能的签名

关于你尝试的类型别名扩展无效的原因 #

你定义的typealias AsnyFunction = () async -> Any?（注意拼写错误：Asny应为Async）之所以无效，是因为：

1. 类型别名只是现有类型的别称，扩展别名本质是扩展其对应的底层类型
2. 你的doSomethingAsync()是() async -> Void类型，和() async -> Any?不是同一类型，所以扩展对它不生效

如果修正别名并匹配函数类型，其实是可以工作的：

typealias AsyncVoidFunc = () async -> Void
extension AsyncVoidFunc {
    func makeItAsync() {
        Task {
            await self()
        }
    }
}

doSomethingAsync().makeItAsync() // 此时可以正常调用

内容的提问来源于stack exchange，提问作者Mehmet Karanlık