转载

一个 ViewModel 的修改建议

有位朋友问了下关于这段 ViewModel 的问题。

在 init 中如果用 subscribeNext 在用到 weak self 是有错误，怎可以避免这个错误呢

一个 ViewModel 的修改建议

import Result
import ObjectMapper
class PhgViewModel: NSObject {
    let refreshTrigger = PublishSubject<Void>()
    let loadNextPageTrigger = PublishSubject<Void>()
    let elements = Variable<[Brand]>([])
    let page = Variable(1)
    let hasNextPage = Variable(true)
    let refreshing = Variable(false)
    let loading = Variable<Bool>(true)
    let error = PublishSubject<ErrorType>()
    init(
        input: (
        refreshTriger: Observable<Void>,
        loadMoreTriger: Observable<Void>,
        )
    ) {
        let refreshRequest :Observable<Result<[Brand],NSError>>  = input.refreshTriger
            .map {1}
            .flatMapLatest { page -> Observable<Result<[String : AnyObject],NSError>> in
                EmeAPI.sharedInstance.request(PhgApi.getList(begin: page , length: 10))
                .observeOn(Mainscheduler.instance)
                .mapRegMessageData([String : AnyObject])
                .map{Result.Success($0)}
                .catchError{ error in
                    let errir  = error as NSError
                    let right = Result<[String : AnyObject], NSError>> in
                    return Observable.just(right)
                }
            }.flatMapLatest { response -> Observable<Result<[Brand],NSError>> in
                return Observable.create { observer -> Disposable in
                    switch response {
                    case .Success(let relDict):
                        if let brand = Mapper<Brand>().mapArray(relDict["list"]){
                            observer.onNext(Result.Success(brand))
                        }
                        observer.onNext(Result.Failure(NSError.init(domain: "获取数据错误", code: 100, userInfo: nil)))
                    case .Failure(let error):
                        observer.onNext(Result.Failure(error))
                    }
                    return NoDisposable.instance
                }
            }
            refreshRequest.subscribeNext{ [weak self]  result in
                switch result {
                case .Success(let branch):
                    self?.page.value  = 2
                    self?.elements.value = brands
                case .Failure(  let error) :
                    print(error)
                }
            }
        ...

编译错误

先来解决编译错误吧。

不能调用 self 是因为初始化并没有完成，可能这个问题不会被一下子发现。仔细看下代码，会发现 RhgViewModel 继承自 NSObject ，也就是说还没有调用 super.init 方法。在初始化最开始加入 super.init() 就可以了。

注：从这段代码中可以看到 NSObejct 是没有存在的必要，我们可以考虑去掉。如果是因为这里面还有一些 Objective-C 的 delegate 之类的代码，可以尝试实现一个 ProxyDelegate 解决。

Driver

Driver 不是必备的，当然可以不用，具体可以参考 [Driver vs Observable] 。

逻辑错误

代码中有一个严重逻辑错误（54 - 57 行）：

if let brand = Mapper<Brand>().mapArray(relDict["list"]){
  observer.onNext(Result.Success(brand))
}
observer.onNext(Result.Failure(NSError.init(domain: "获取数据错误", code: 100, userInfo: nil)))