Swift 中实现 Promise 模式

在异步编程中,除了竟态处理、资源利用以外,另外一个难点就是流程管理。在拥有匿名函数、闭包这些特性的编程语言中,我们通常可以使用回调函数来做一个异步任务完成或失败时的处理。但当我们的业务逻辑逐渐复杂时,就会产生回调嵌套,整个事件流将十分混乱。相信大家对 Node.js 的回调陷阱一定有所耳闻了。于是各种各样事件流处理的库就产生了,比如 NPM.js 社区中著名的 qbluebird 都是用来解决回调陷阱的,它们所采用的模式就是我们所说的 Promise,也是我们今天要谈的模式。当然异步流程处理的方式远不止 Promise 这一种,在支持生成器语义的语言里我们还能用 yield + generator 来实现协程,co就是这种思想。但是 Swift 并不支持,所以我们就不讨论了。

Getting Started

相信大家在日常开发中对下面这种模式已经不陌生了:

dispatch_async(dispatch_get_global_queue(QOS_CLASS_USER_INITIATED, 0)) { 
    // Do something in background.
    dispatch_async(dispatch_get_main_queue()) {
        // Update the UI.
    }
}

类似用法的还有诸如权限认证NSURLSession,都有这样的回调,如果逻辑稍微复杂一些就会出现回调陷阱。下面我们应用 Promise 来将它们平坦化,通过连接就可以实现链式的事件响应。

我们先看看用了 Promise 后是一番怎样的景象:

doSomething()
    .then( doAnotherThing )
    .then( doSomethingElse )
    .success( someHandler )

是不是瞬间清晰了很多?我们来分析一下如何实现。
首先,doSomething() 会返回一个 Promise 对象,Promise 对象在构造函数中接受一个函数作为参数,这个函数就是用来启动异步任务的。而这个函数又有两个参数,分别是成功时的回调和失败时的回调。

我们下面就来实现这个 Promise 类:

class Promise<T> {
    typealias ResolveCallback = (T) -> Void
    typealias RejectCallback = (ErrorType) -> Void
    typealias AsyncTask = (ResolveCallback, RejectCallback) -> Void
    
    let task: AsyncTask
    
    var resolveCallback: ResolveCallback?
    var rejectCallback: RejectCallback?
    
    init(_ task: AsyncTask) {
        self.task = task
    }
}

通过泛型表示异步任务最终得到的结果的类型,对于失败的情况,传递一个实现 ErrorType 协议的对象作为错误的原因。

下面我们看如何响应结果并启动任务:

    ...

    func success(callback: ResolveCallback) {
        self.resolveCallback = callback
        self.task({ self.resolve($0) }, { self.reject($0) })
    }
    
    func failed(callback: RejectCallback) {
        self.rejectCallback = callback
    }

    ...

我这里采用了冷启动的方式,也就是说仅当有响应函数被挂载时才启动异步任务,如果你想在 Promise 一被创建时就启动异步任务(也就是热启动),就需要用一个属性来存放任务的结果,以免响应函数还没被挂载,这个异步任务就完成了,这样这个结果就丢失了。

异步任务启动的方式是将 resolvereject 作为参数传给异步任务启动函数,当异步任务自身的回调调用时,这个 Promise 对象就能做出响应处理,将这个事件传递给它的响应函数。

当然,resolvereject函数也很简单,就是执行回调:

    private func resolve(result: T) {
        self.resolveCallback?(result)
    }
    
    private func reject(error: ErrorType) {
        self.rejectCallback?(error)
    }

链式调用

到现在我们并没有实现 Promise 的精髓,还不能链式调用。所谓的链式调用就是当一个 Promise 完成时,立即用一个变换函数将结果传给下一个 Promise 去执行,以此类推,这样我们构造一串操作之后再一并启动,实现上面我说到的冷启动。

那么这个核心函数就是 then,这个函数接受一个变换函数作为参数,这个变换函数接受 Promise 结果然后构造出一个新的 Promise 对象,然后返回。由于 then 返回的也是 Promise 因此我们可以用链式语法不断连接多个操作,十分方便。

那么连接后的启动顺序回事怎样的呢?举个例子:

promise1.then(genPromise2).then(genPromise3).success(...)

那么最终得到的 Promise 对象就是这样的:

promise3( promise2( promise1 ) )

最后一个 then 连接的 Promise 对象包裹了上一个对象,以此类推,那么在启动时,当然也是最后一个对象被启动,只不过由于这个 Promise 对象是被封装过的,所以它会先触发上一个对象,待拿到结果后再传递给自己执行,仍然以此类推,就能保证执行顺序是正确的。

说的很抽象,直接看代码吧:

    func then<U>(f: (T) -> Promise<U>) -> Promise<U> {
        return Promise<U> { (resolve, reject) in
            self.task(
                { (result) in
                    let wrapped = f(result)
                    wrapped.success { resolve($0) }
                },
                { (error) in
                    reject(error)
            })
        }
    }

可以看到我们在两个 Promise 对象之间又封装了一个 Promise 对象,作为协调者,它会先执行之前的异步任务,然后再传给下一个任务,那么这个函数就会返回封装后的 Promise 对象。

下面是完整的代码:

class Promise<T> {
    typealias ResolveCallback = (T) -> Void
    typealias RejectCallback = (ErrorType) -> Void
    typealias AsyncTask = (ResolveCallback, RejectCallback) -> Void
    
    let task: AsyncTask
    
    var resolveCallback: ResolveCallback?
    var rejectCallback: RejectCallback?
    
    init(_ task: AsyncTask) {
        self.task = task
    }
    
    private func resolve(result: T) {
        self.resolveCallback?(result)
    }
    
    private func reject(error: ErrorType) {
        self.rejectCallback?(error)
    }
    
    func success(callback: ResolveCallback) {
        self.resolveCallback = callback
        self.task({ self.resolve($0) }, { self.reject($0) })
    }
    
    func failed(callback: RejectCallback) {
        self.rejectCallback = callback
    }
    
    func then<U>(f: (T) -> Promise<U>) -> Promise<U> {
        return Promise<U> { (resolve, reject) in
            self.task(
                { (result) in
                    let wrapped = f(result)
                    wrapped.success { resolve($0) }
                },
                { (error) in
                    reject(error)
            })
        }
    }
}

使用示例

下面我们试试怎么使用这个 Promise 对象,我首先定义三个操作:

func delay(secs: UInt64) -> Promise<Void> {
    return Promise<Void> { (resolve, _) in
        let time = dispatch_time(DISPATCH_TIME_NOW, Int64(NSEC_PER_SEC * secs))
        dispatch_after(time, dispatch_get_main_queue()) {
            resolve()
        }
    }
}

func fetch(URL URL: NSURL) -> Promise<NSData?> {
    return Promise<NSData?> { (resolve, reject) in
        let task = NSURLSession.sharedSession().dataTaskWithURL(URL) { (data, response, error) in
            if (error != nil) {
                reject(error!)
            } else {
                resolve(data)
            }
        }
        task.resume()
    }
}

func decodeToString(data: NSData?) -> Promise<String> {
    return Promise<String> { (resolve, _) in
        if (data == nil) {
            resolve("")
        } else {
            resolve(String(data: data!, encoding: NSUTF8StringEncoding) ?? "")
        }
    }
}

这些操作里有异步操作,也有同步操作,但都能适应 Promise 模式,每个函数都返回一个 Promise 对象。

然后我们构造一个链式操作:

delay(5)
    .then { () -> Promise<NSData?> in
        fetch(URL: NSURL(string: "https://www.zhihu.com")!)
    }
    .then { (data) -> Promise<String> in
        decodeToString(data)
    }
    .success { (result) in
        print(result)
}

XCPlaygroundPage.currentPage.needsIndefiniteExecution = true

注意:我这个例子是在 Playground 里做的,所以为了让异步任务能够执行,我们需要设置一个属性,以免 Playground 在主线程完毕后结束程序。

结果符合我们的预期,在 5 秒的延时之后,网络请求被执行,然后转码,最终打印出来。

总结

事实上,Promise 模式被利用得最好的应该是 JavaScript,我们只不过在其他语言中借鉴这种做法,然后给出响应的实现罢了。当然本文只是简单剖析 Promise 的内部原理,很多细节可能没有太完善,如果你喜欢这种编程方式,可以尝试一下成熟的 PromiseKit,它有 Objective-C 和 Swift 的实现。
当然你也可以尝试一下具有类似思想的反应式函数式编程框架,比如 ReactiveX

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容