com.noheltcj.rxcommon.operators.DoOnEach.kt Maven / Gradle / Ivy

Go to download
package com.noheltcj.rxcommon.operators

import com.noheltcj.rxcommon.Source
import com.noheltcj.rxcommon.disposables.Disposable
import com.noheltcj.rxcommon.disposables.Disposables
import com.noheltcj.rxcommon.emitters.ColdEmitter
import com.noheltcj.rxcommon.emitters.Emitter
import com.noheltcj.rxcommon.observers.AllObserver
import com.noheltcj.rxcommon.observers.Observer

class DoOnEach(
        private val upstream: Source,
        private val onEachObserver: Observer
) : Operator() {
    override val emitter: Emitter = ColdEmitter {}

    override fun subscribe(observer: Observer): Disposable {
        emitter.addObserver(observer)

        val upstreamDisposable = upstream.subscribe(AllObserver(
                onNext = {
                    onEachObserver.onNext(it)
                    emitter.next(it)
                },
                onError = {
                    onEachObserver.onError(it)
                    emitter.terminate(it)
                },
                onComplete = {
                    onEachObserver.onComplete()
                    emitter.complete()
                }
        ))

        return Disposables.create {
            emitter.removeObserver(observer)
            upstreamDisposable.dispose()
        }
    }
}