tigorn / cooperation Goto Github PK

import Foundation
import CompoundOperations

/// Chainable operation that performs network request
class NetworkRequestChainableOperation: ChainableOperationBase {
    
    /// Network client (Core-component)
    private let networkClient: NetworkClient
    
    init(networkClient: NetworkClient) {
        self.networkClient = networkClient
        
        super.init()
    }
    
    // MARK: Executing
    
    override func inputDataClass() -> AnyClass? {
        return NSURLRequest.self
    }
    
    override func processInputData(inputData: AnyObject?,
                                   completionBlock: ChainableOperationBaseOutputDataBlock) {
        
        let inputRequest = inputData as! NSURLRequest
        
        networkClient.performRequest(inputRequest) { (data, error) in
            completionBlock(data, error)
        }
    }
}

    func obtainDataCompoundOperation(withResultBlock resultBlock: CompoundOperationResultBlock?) -> CompoundOperation {
        let networkRequestOperation = NetworkRequestChainableOperation(networkClient: NetworkClientImplementation())
        let deserializationOperation = DeserializationChainableOperation(deserializer: JSONDeserializer)

        let chainableOperations = [
            networkRequestOperation,
            deserializationOperation
        ]
        
        let operation = CompoundOperation.defaultCompoundOperation()
        operation.configureWithChainableOperations(chainableOperations,
                                                   resultBlock: resultBlock)
        
        return operation
    }

    let compoundOperation = obtainDataCompoundOperation { (data, error) in
        // Process the result
    }
    
    queue.addOperation(compoundOperation) // OR compoundOperation.start()