谈谈RxSwift中的错误处理

RxSwift中提供了多种不同的错误处理操作符,它们可以在链式操作中相互组合以实现复杂的处理逻辑,下面先简单介绍一下RxSwift提供的错误处理操作,然后通过一些具体的例子来看看如何在实际项目中应用。这里不会详细介绍RxSwift,阅读前需要对Rx的基础有一定了解。

错误处理操作符

throw

Rx中许多操作符中的闭包签名都是带有throws修饰符的,比如说这是map方法的签名:

func map<R>(_ transform: @escaping (E) throws -> R) -> Observable<R>

我们可以在这样的操作中抛出错误,抛出的错误会沿着链式操作一步步向下传递,比如下面的代码:

Observable.of(3, 2, 1) // 创建一个包含3个事件的Observable
    .map { (n) -> Int in
        if n < 2 {
            throw CustomError.tooSmall // 1. 抛出一个自定义的错误
        } else {
            return n * 2
        }
    }
    .subscribe { event in
        // 2. 这里会收到一个 CustomError.tooSmall 类型的error
    }
    ...

当数字小于2时,在mapthrow一个自定义的错误类型,这个错误会被传递到下面的subscribe中。

catchError

RxSwift可以在链式操作中捕获错误,不管是Observable自己产生的错误还是用户throw的错误,都可以在catchError操作中进行处理,接着上面map的代码:

Observable.of(3, 2, 1)
    .map { 
        ...
    }
    .catchError({ (error) -> Observable<Int> in
        if case CustomError.tooSmall = error {
            return .just(2) // 1. 在捕获到tooSmall错误后返回一个2
        }
        return .error(error) // 2. 其他错误不处理,继续沿着操作链传递
    })
    .subscribe { event in
        // 3. 当发生tooSmall错误时这里会收到2,最终的结果是 3, 2, 2
    }
    ...

这样的处理方式接近于语言本身的try…catch机制,使用起来十分方便。

retry

retry提供了出错重试的操作,在一个Observable后面加上retry,当这个Observable出错的时候订阅方不会收到.error事件,而是重新订阅这个Observable,这通常意味着这个Observable创建事件相关的操作都会重新执行一次,比如说这是一个网络请求相关的Observable,“重新订阅”会重发相关的网络请求:

moyaProvider.rx.request(.customData)
    .retry() // 1. 当发生错误时重试请求
    .subscribe { 
        // 2. 重试的过程对于订阅者来说是不可见的,请求成功后这里才会接收到事件
    }

retry方法还可以带一个Int类型的参数,表示重试的最大次数。

retryWhen

retryWhen就是带条件的retry,可以在特定条件下才进行重试。retryWhen的方法签名比较特别,它的闭包中接受的参数不是一个简单的Error类型,而是一个Observable<Error>类型,使用方法如下:

Observable.of(3, 2, 1)
    .map { 
        ...
    }
    // 1. retryWhen不关心返回的是什么类型,只关心事件本身,所以直接用Observable<()>即可
    .retryWhen({ (errorObservable) -> Observable<()> in
        // 2. 用flatMap将其转换成其他类型的Observable
        return errorObservable.flatMap({ (error) -> Observable<()> in
            if case CustomError.tooSmall = error {
                return .just(()) // 3. 返回一个next事件表示重试
            }
            return .error(error) // 4. 继续返回error表示不处理
        })
    })

闭包返回的Observable可以是任意类型,因为retryWhen只关心Observable中的事件本身,不关心其中承载的数据类型,所以这里直接用一个空类型即可,如果需要重试的话就将一个带有.next事件的Observable返回。

retryWhen这样设计的一个优点是在出错的时候可以将它重试的逻辑跟另外一个Observable事件流关联起来(后面我会演示一个例子)。但是在上面这样一个简单的场景中,使用起来未免过于麻烦了,这里可以做一个简单的封装,提供一个(Error) -> Bool类型的闭包来处理判断逻辑:

extension ObservableType {
    public func retryWhen<Error: Swift.Error>(_ shouldRetry: @escaping (Error) -> Bool) -> Observable<E> {
        return self.retryWhen({ (errorObserver: Observable<Error>) -> Observable<()> in
            return errorObserver.flatMap({ (error) -> Observable<()> in
                if shouldRetry(error) {
                    return .just(())
                }
                return .error(error)
            })
        })
    }
    
    public func retryWhen(_ shouldRetry: @escaping (Swift.Error) -> Bool) -> Observable<E> {
        return self.retryWhen({ (errorObserver: Observable<Swift.Error>) -> Observable<()> in
            return errorObserver.flatMap({ (error) -> Observable<()> in
                if shouldRetry(error) {
                    return .just(())
                }
                return .error(error)
            })
        })
    }
}

将上面这段代码复制到你的项目中,之前的重试逻辑就变成了:

...
.retryWhen({ (error) -> Bool in
    if case CustomError.tooSmall = error {
        return true
    }
    return false
})
...

这样看起来清楚多了,减轻了思维负担。

实际应用

Moya是Swift常用的一个网络库,它提供了Rx的接口,下面的例子以Moya作为网络库来演示,Moya的一个核心协议是TargetType,不了解Moya的朋友可以看看它的文档,基本使用就不再详细介绍了。下面来看两个常见的实际应用场景

场景一:带交互的出错重试

在很多时候,用户的操作失败时不能直接重试,而是要给一个,让用户来决定下一步的操作。例如有一个文件下载的请求,当下载失败的时候需要弹框来询问是否重试。也就是说在出错到重试之间存在一个“中断”,只有当用户做出选择之后操作链才会继续向下执行。

解决方法是使用retryWhen,将参数中的的Observable<Error>与我们自己业务逻辑的Observable关联起来。

首先,我们假定有这样一个确认框的控件,它的签名如下:

class ConfirmView: UIView {
    /// 在视图中显示一个确认框,callback为点击的回调,点击确认回调true,点击取消回调false
    static func show(_ title: String, _ callback: (Bool) -> Void) {
        ...
    }
}

实际的项目中通常都会有很多封装好的控件类型,借助于RxSwift中所提供的扩展机制,只需要添加一个小小的扩展就可以与Rx的世界无缝对接起来:

extension Reactive where Base: ConfirmView {
    // 1. 在扩展中定义一个show方法,不同的是没有callback参数,而是返回一个Observable<Bool>
    static func show(_ title: String) -> Observable<Bool> {
        // 2. 创建一个Observable<Bool>
        return Observable<Bool>.create({ (observer) -> Disposable in
            // 3. 调用原始的show方法,并在回调中通过observer发送结果
            ConfirmView.show(title, { (confirm) in
                observer.onNext(confirm)
                observer.onCompleted()
            })
            return Disposables.create { 
                // do some cleanup
            }
        })
    }
}

之后就可以通过ConfirmView.rx.show(xxx)的方式来调用这个方法了,这个方法会弹出一个选择框等待用户的选择,选择的结果通过Observable的事件来进行通知。之后我们使用flatMap将这个ObservableretryWhen中的Obverable<Error>关联起来:

...
.retryWhen({ (errorO) -> Observable<()> in
    return errorO.flatMap({ (error) -> Observable<()> in
        if case CustomError.tooSmall = error {
            return ConfirmView.rx
                .show("是否重试?")
                .map {
                    if $0 { // 1. 如果选择了重试,则返回.next()表示重试
                        return ()
                    } else {
                        throw error // 2. 否则继续返回error将错误继续向下传递
                    }
                }
        }
        return .error(error)
    })
})
.subscribe {
    // 3. 如果上面选择了重试,这里不会接收到错误事件
}
...

类似的,将不同的操作封装成Observable这样简单的逻辑流,然后通过RxSwift提供的操作加以组合以实现更加复杂的逻辑,这也是Rx所提倡的函数式思想。

场景二:401认证

401错误是一种很常见应用场景,比如说在我们的应用中认证流程是这样的:当服务器需要重新认证用户登录信息时会返回一个401状态码,这时客户端将认证信息添加到请求头中并重发当前的请求,这一过程对上层的业务方应该是无感知的。

这跟之前的例子有一些不同的地方:当出错时我们不能直接retry整个请求,而是要修改原始请求添加自定义的Header,最简单粗暴的方法是在检测到401错误时发送一个通知,外面收到通知之后将Header添加到请求头里:

moyaProvider.request(target)
    .map({ (response) -> Response in
        if response.statusCode == 401 { // 将401转换成自定义的错误类型
            // 先发送通知,之后再retry
            NotificationCenter.default.post(name: .AddAuthHeader, object: nil)
            throw NetworkError.needAuth
        } else {
            return response
        }
    })
    .retry()

这种做法其实并不好,因为Rx中强调的是事件流,原本应该是一个连贯的逻辑却被通知给打断了,当我们阅读到这里的时候还得停下来全局搜索通知的名字以查找响应的位置,这样不利于阅读,同时也违背了Rx的哲学。

我这里所采用的做法是捕获到错误时不进行retry,而是返回一个新的网络请求。为了让这个新的网络请求与之前的逻辑无缝连接起来,首先需要定义一个代理TargetType:

let ProxyProvider = NetworkProvider<ProxyTarget>()

enum ProxyTarget {
    // 添加Header
    case addHeader(target: TargetType, headers: [String: String]) 
    // ...
}

extension ProxyTarget: TargetType {
    var headers: [String: String]? {
        switch self {
        // 1. 将新增的Header添加到被代理的Target上
        case let .addHeader(target: target, headers: headers):
            return headers.merging(target.headers ?? [:], uniquingKeysWith: { (first, second) -> String in
                return first
            })
        }
    }
    
    // 2. 不需要吹的地方直接返回被代理Target的属性
    var task: Task {
        switch self {
        case let .addHeader(target: target, headers: _):
            return target.task
        }
    }
    
    // ...
}

ProxyTarget并没有定义新的网络请求,而是用来代理另外一个TargetType,这里我们只定义了一个addHeader操作,用来修改请求的Header。

最终的实现如下:

provider.request(target)
    .map({ (response) -> Response in
        if response.statusCode == 401 { // 1. 将401转换成自定义的错误类型
            throw NetworkError.needAuth
        } else {
            return response
        }
    })
    .catchError({ (error) -> Single<Response> in
        if case NetworkError.needAuth(let response) = error{
            // 2. 捕获未认证的错误,添加认证头后再次重试
            let authHeader = ... // 计算认证头
            let target = ProxyTarget.addHeader(target: token, headers: authHeader)
            return ProxyProvider.rx.request(target, callbackQueue: callbackQueue)
        }
        return Single.error(error)
    })
    .subscribe {
        // 3. 认证的过程对于上层的业务方是无感知的
    }
    ...

使用map将401转换成自定义的错误类型,之后在catchError中捕获这个错误,使用ProxyTarget加上认证头之后返回一个新的Observable,这样一来所有相关的逻辑都被集中在这一系列的链式调用中了。

当然在实际项目中不仅仅是401这类错误,可能还会有许多其他业务相关的错误类型,将它们全都放在map中处理显然不是一个好主意,最好的办法是将这部分逻辑抽离出来放在Moya的Plugin中,这里就不再演示了。

最后

Rx中对于事件流的抽象十分强大,可以用来描述各种复杂的场景,这里仅仅从错误处理的方面列举了一些简单的例子,可以看到Rx的思想跟我们平常所写的代码有很大不同,思维上的转变才是理解Rx的关键。

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

推荐阅读更多精彩内容

  • 心在慌乱迷雾在前方那十字的墓碑刻着曾经的辉煌那墓碑的主人持着荣耀的勋章亡灵在呻吟恶魔在低语他将要苏醒,他将要回归 ...
    但丁如是说阅读 372评论 0 0
  • 欢迎关注微信公众号:说歌词(shuogeci) 昨天看了《奇葩大会》新一期,“晚上12点好饿哦,我该不该吃宵夜?”...
    上帝掷色子吗阅读 742评论 0 1
  • “竹泫泫以垂露,柳依依而迎蝉,鸥双双以赴水,鹭轩轩而归田。” 前两天偶然看到一部《茅山》的纪录片,听到画外音读到陶...
    漏报阅读 553评论 0 0
  • 新闻,有数据显示,2017年上半年,上海市送餐外卖行业发生交通事故76起,平均两天半就有一人伤亡。在南京,上半年涉...
    田园听雨阅读 261评论 0 1