ios - 带有 nil 包装的自定义 flatMapLatest

Question

我有一些测试函数，观察者对其进行赋值并将其映射到另一个 Observable :

private func test1() {
    selectedTagsVariable
        .asObservable()
        .flatMapLatest { [weak self] tags -> Observable<PostSet> in
            guard let strongSelf = self else { return .empty() }
            return strongSelf.postSetObservable(for: tags)
        }
}

没关系，但我不想每次都检查 self 值为 nil。当然，我可以替换weak至 unowned , 但这是不正确的。

因此，我创建了自定义 flatMapLatest :

extension ObservableType {

func xflatMapLatest<A:AnyObject, O: ObservableType>(weak obj: A, selector: @escaping (A, Self.E) throws -> O) -> Observable<O.E> {
    return flatMapLatest { [weak obj] value -> Observable<O.E> in
        guard let strongObj = obj else {
            return Observable.empty()
        }
        return try selector(strongObj, value) as! Observable<O.E>
    }
}
}

它看起来像这样:

private func test2() {
    selectedTagsVariable
        .asObservable()
        .xflatMapLatest(weak: self) { obj, tags -> Observable<PostSet> in
            return obj.postSetObservable(for: tags)
        }
}

问题:我不喜欢 xflatMapLatest 中的类型转换返回类型 ( as! Observable<O.E> ) 中的函数。我可以以某种方式删除它吗？帮助重构这个方法:)

Answer 1

你想要的是镜像 flatMapLatest运营商与一个简单的修改。看看它的签名:

func flatMapLatest<O : ObservableConvertibleType>(_ selector: @escaping (Self.E) throws -> O) -> RxSwift.Observable<O.E>

如您所见，selector 的结果类型关闭必须符合 ObservableConvertibleType ，它的定义在这里:

public protocol ObservableConvertibleType {
    associatedtype E
    public func asObservable() -> RxSwift.Observable<Self.E>
}

知道了这一点，就很容易看出您需要做哪些小的改变才能让它发挥作用:

extension ObservableType {
    func flatMapLatestWeak<A: AnyObject, O: ObservableConvertibleType>(weak obj: A, selector: @escaping (A, E) throws -> O) -> Observable<O.E> {
        return flatMapLatest { [weak obj] value -> Observable<O.E> in
            guard let strongObj = obj else {
                return Observable.empty()
            }
            return try selector(strongObj, value).asObservable()
        }
    }
}

此外，您可以使用 Optional<T> 缩短此方法的 map :

extension ObservableType {
    func flatMapLatestWeak<A: AnyObject, O: ObservableConvertibleType>(weak obj: A, selector: @escaping (A, E) throws -> O) -> Observable<O.E> {
        return flatMapLatest { [weak obj] value in
            try obj.map{ try selector($0 , value).asObservable() } ?? .empty()
        }
    }
}