重新组织链式可观察物

时间:2019-06-17 00:36:40

标签: swift rx-swift

当通过table.rx.modelSelected选择一个tableviews行时,会触发大量的Rx链式可观察对象。

我希望能够分解这种逻辑,因为我当前必须在flatMapLatest中执行业务逻辑,因为它是流程的“第一步”(感觉很不对劲),我必须在随后的subscribe中执行更多的业务逻辑(“步骤2”)。这是我正在使用的代码:

locationsTable.rx.modelSelected(Location.self)
    .flatMapLatest { [weak self] location -> Observable<[JobState]?> in
        guard let hubs = self?.viewModel.userInfo.authorizedHubLocations else { return .empty() }
        guard let hub = hubs.first(where: { $0.locationId == location.id }) else { return .empty() }
        guard let hubToken = hub.hubToken else { return .empty() }

        // save data in db
        self?.databaseService.persistHub(hubResult: hub, location: location)

        // make network call for the 2nd step (the subscribe)
        let networkService = NetworkService(plugins: [AuthPlugin(token: hubToken)])
        return networkService.jobStates(locationId: location.id)
    }
    .subscribe(onNext: { [weak self] jobState in
        if let jobState = jobState {
            self?.databaseService.persistJobStates(jobStates: jobState)
        }
        NavigationService.renderScreenB()
    }, onError: { error in
        Banner.showBanner(type: .error, title: "Whoops", message: "Something went wrong.")
    }).disposed(by: disposeBag)

此代码当前有效,但感觉很脏。任何有关如何清理此问题的建议将不胜感激。

1 个答案:

答案 0 :(得分:1)

您有几个独立且截然不同的逻辑和副作用,并且您正在尝试将它们全部填充到一个flatMap中。我建议将它们分解为各个组成部分。

此外,您的错误逻辑也不正确。如果您的网络服务发出错误消息,则将显示“ Whoops”标语,但也会打断您的链条,用户将无法选择其他位置。我下面的代码解决了这个问题。

以下功能均为免费功能。由于它们没有绑定到特定的视图控制器,因此可以独立使用和测试它们。还要注意,这些功能包含系统的逻辑 all only 。这使您可以测试无副作用的逻辑并提升良好的体系结构。另请注意,它们返回Driver。您可以确定这些函数都不会发出错误,该错误会破坏链和视图控制器的行为。

/// Emits hubs that need to be persisted.
func hubPersist(location: ControlEvent<Location>, userInfo: UserInfo) -> Driver<(location: Location, hub: Hub)> {
    let hub = getHub(location: location, userInfo: userInfo)
        .asDriver(onErrorRecover: { _ in fatalError("no errors are possible") })
    return Driver.combineLatest(location.asDriver(), hub) { (location: $0, hub: $1) }
}

/// Values emitted by this function are used to make the network request.
func networkInfo(location: ControlEvent<Location>, userInfo: UserInfo) -> Driver<(NetworkService, Int)> {
    let hub = getHub(location: location, userInfo: userInfo)
    return Observable.combineLatest(hub, location.asObservable())
        .compactMap { (hub, location) -> (NetworkService, Int)? in
            guard let hubToken = hub.hubToken else { return nil }
            return (NetworkService(plugins: [AuthPlugin(token: hubToken)]), location.id)
        }
        .asDriver(onErrorRecover: { _ in fatalError("no errors are possible") })
}

/// shared logic used by both of the above. Testing the above will test this by default.
func getHub(location: ControlEvent<Location>, userInfo: UserInfo) -> Observable<Hub> {
    return location
        .compactMap { location -> Hub? in
            let hubs = userInfo.authorizedHubLocations
            return hubs.first(where: { $0.locationId == location.id })
    }
}

下面的功能是对您的网络请求的包装,使错误更易于使用。

extension NetworkService {
    func getJobStates(locationId: Int) -> Driver<Result<[JobState], Error>> {
        return jobStates(locationId: locationId)
            .map { .success($0 ?? []) }
            .asDriver(onErrorRecover: { Driver.just(.failure($0)) })
    }
}

这是使用以上所有内容的视图控制器代码。它几乎完全由副作用组成。唯一的逻辑就是检查网络请求是否成功的两个保护措施。

func viewDidLoad() {
    super.viewDidLoad()

    hubPersist(location: locationsTable.rx.modelSelected(Location.self), userInfo: viewModel.userInfo)
        .drive(onNext: { [databaseService] location, hub in
            databaseService?.persistHub(hubResult: hub, location: location)
        })
        .disposed(by: disposeBag)

    let jobStates = networkInfo(location: locationsTable.rx.modelSelected(Location.self), userInfo: viewModel.userInfo)
        .flatMapLatest { networkService, locationId in
            return networkService.getJobStates(locationId: locationId)
        }

    jobStates
        .drive(onNext: { [databaseService] jobStates in
            guard case .success(let state) = jobStates else { return }
            databaseService?.persistJobStates(jobStates: state)
        })
        .disposed(by: disposeBag)

    jobStates
        .drive(onNext: { jobStates in
            guard case .success = jobStates else { return }
            NavigationService.renderScreenB()
        })
        .disposed(by: disposeBag)

    jobStates
        .drive(onNext: { jobStates in
            guard case .failure = jobStates else { return }
            Banner.showBanner(type: .error, title: "Whoops", message: "Something went wrong.")
        })
        .disposed(by: disposeBag)
}

仅供参考,以上代码使用Swift 5 / RxSwift 5。