53 lines
1.6 KiB
Swift
53 lines
1.6 KiB
Swift
//
|
|
// ColdObservable.swift
|
|
// Rx
|
|
//
|
|
// Created by Krunoslav Zaher on 3/14/15.
|
|
// Copyright (c) 2015 Krunoslav Zaher. All rights reserved.
|
|
//
|
|
|
|
import RxSwift
|
|
|
|
class ColdObservable<Element: Equatable>: Observable<Element> {
|
|
typealias Events = Recorded<Element>
|
|
typealias Observer = ObserverOf<Element>
|
|
|
|
let testScheduler: TestScheduler
|
|
|
|
var subscriptions: [Subscription]
|
|
var recordedEvents: [Events]
|
|
var observers: Bag<Observer>
|
|
|
|
init(testScheduler: TestScheduler, recordedEvents: [Events]) {
|
|
self.testScheduler = testScheduler
|
|
|
|
self.recordedEvents = recordedEvents
|
|
self.subscriptions = []
|
|
self.observers = Bag()
|
|
|
|
super.init()
|
|
}
|
|
|
|
override func subscribe<O : ObserverType where O.E == E>(observer: O) -> Disposable {
|
|
let key = observers.put(ObserverOf(observer))
|
|
subscriptions.append(Subscription(self.testScheduler.now))
|
|
|
|
let i = self.subscriptions.count - 1
|
|
|
|
for recordedEvent in recordedEvents {
|
|
testScheduler.scheduleRelative((), dueTime: recordedEvent.time, action: { (Int) in
|
|
self.observers.forEach { $0.on(recordedEvent.event) }
|
|
return NopDisposableResult
|
|
})
|
|
}
|
|
|
|
return AnonymousDisposable {
|
|
let removed = self.observers.removeKey(key)
|
|
assert(removed != nil);
|
|
|
|
let existing = self.subscriptions[i]
|
|
self.subscriptions[i] = Subscription(existing.subscribe, self.testScheduler.now)
|
|
|
|
}
|
|
}
|
|
} |