|
import { Subject } from '../Subject'; |
|
import { Subscriber } from '../Subscriber'; |
|
import { Subscription } from '../Subscription'; |
|
import { Scheduler } from '../Scheduler'; |
|
import { TestMessage } from './TestMessage'; |
|
import { SubscriptionLog } from './SubscriptionLog'; |
|
import { SubscriptionLoggable } from './SubscriptionLoggable'; |
|
import { applyMixins } from '../util/applyMixins'; |
|
import { observeNotification } from '../Notification'; |
|
|
|
export class HotObservable<T> extends Subject<T> implements SubscriptionLoggable { |
|
public subscriptions: SubscriptionLog[] = []; |
|
scheduler: Scheduler; |
|
|
|
logSubscribedFrame: () => number; |
|
|
|
logUnsubscribedFrame: (index: number) => void; |
|
|
|
constructor(public messages: TestMessage[], scheduler: Scheduler) { |
|
super(); |
|
this.scheduler = scheduler; |
|
} |
|
|
|
|
|
protected _subscribe(subscriber: Subscriber<any>): Subscription { |
|
const subject: HotObservable<T> = this; |
|
const index = subject.logSubscribedFrame(); |
|
const subscription = new Subscription(); |
|
subscription.add( |
|
new Subscription(() => { |
|
subject.logUnsubscribedFrame(index); |
|
}) |
|
); |
|
subscription.add(super._subscribe(subscriber)); |
|
return subscription; |
|
} |
|
|
|
setup() { |
|
const subject = this; |
|
const messagesLength = subject.messages.length; |
|
|
|
for (let i = 0; i < messagesLength; i++) { |
|
(() => { |
|
const { notification, frame } = subject.messages[i]; |
|
|
|
subject.scheduler.schedule(() => { |
|
observeNotification(notification, subject); |
|
}, frame); |
|
})(); |
|
} |
|
} |
|
} |
|
applyMixins(HotObservable, [SubscriptionLoggable]); |
|
|