ServiceType1+FriendRequests

ServiceType1+FriendRequests.swift

import RxSwift

protocol MyViewModelServiceType {
    func observeCurrentNumberOfFriendRequests() -> Observable<Int>
}

struct MyViewModelService: MyViewModelServiceType {
    
    private let usersBaseService = UsersBaseService.shared
    
    init() {
        // implement mock case for testing
    }
    
    func observeCurrentNumberOfFriendRequests() -> Observable<Int> {
        return usersBaseService.observeCurrentFriendRequests()
            .map { friendRequestsResult in
                switch friendRequestsResult {
                case .success(let userIds):
                    return userIds.count
                case .failure:
                    return 0
                }
        }
    }
}