91 lines
3.1 KiB
JavaScript
91 lines
3.1 KiB
JavaScript
|
import { OuterSubscriber } from '../OuterSubscriber';
|
||
|
|
import { subscribeToResult } from '../util/subscribeToResult';
|
||
|
|
export function expand(project, concurrent = Number.POSITIVE_INFINITY, scheduler = undefined) {
|
||
|
|
concurrent = (concurrent || 0) < 1 ? Number.POSITIVE_INFINITY : concurrent;
|
||
|
|
return (source) => source.lift(new ExpandOperator(project, concurrent, scheduler));
|
||
|
|
}
|
||
|
|
export class ExpandOperator {
|
||
|
|
constructor(project, concurrent, scheduler) {
|
||
|
|
this.project = project;
|
||
|
|
this.concurrent = concurrent;
|
||
|
|
this.scheduler = scheduler;
|
||
|
|
}
|
||
|
|
call(subscriber, source) {
|
||
|
|
return source.subscribe(new ExpandSubscriber(subscriber, this.project, this.concurrent, this.scheduler));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
export class ExpandSubscriber extends OuterSubscriber {
|
||
|
|
constructor(destination, project, concurrent, scheduler) {
|
||
|
|
super(destination);
|
||
|
|
this.project = project;
|
||
|
|
this.concurrent = concurrent;
|
||
|
|
this.scheduler = scheduler;
|
||
|
|
this.index = 0;
|
||
|
|
this.active = 0;
|
||
|
|
this.hasCompleted = false;
|
||
|
|
if (concurrent < Number.POSITIVE_INFINITY) {
|
||
|
|
this.buffer = [];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
static dispatch(arg) {
|
||
|
|
const { subscriber, result, value, index } = arg;
|
||
|
|
subscriber.subscribeToProjection(result, value, index);
|
||
|
|
}
|
||
|
|
_next(value) {
|
||
|
|
const destination = this.destination;
|
||
|
|
if (destination.closed) {
|
||
|
|
this._complete();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const index = this.index++;
|
||
|
|
if (this.active < this.concurrent) {
|
||
|
|
destination.next(value);
|
||
|
|
try {
|
||
|
|
const { project } = this;
|
||
|
|
const result = project(value, index);
|
||
|
|
if (!this.scheduler) {
|
||
|
|
this.subscribeToProjection(result, value, index);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
const state = { subscriber: this, result, value, index };
|
||
|
|
const destination = this.destination;
|
||
|
|
destination.add(this.scheduler.schedule(ExpandSubscriber.dispatch, 0, state));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch (e) {
|
||
|
|
destination.error(e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
this.buffer.push(value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
subscribeToProjection(result, value, index) {
|
||
|
|
this.active++;
|
||
|
|
const destination = this.destination;
|
||
|
|
destination.add(subscribeToResult(this, result, value, index));
|
||
|
|
}
|
||
|
|
_complete() {
|
||
|
|
this.hasCompleted = true;
|
||
|
|
if (this.hasCompleted && this.active === 0) {
|
||
|
|
this.destination.complete();
|
||
|
|
}
|
||
|
|
this.unsubscribe();
|
||
|
|
}
|
||
|
|
notifyNext(outerValue, innerValue, outerIndex, innerIndex, innerSub) {
|
||
|
|
this._next(innerValue);
|
||
|
|
}
|
||
|
|
notifyComplete(innerSub) {
|
||
|
|
const buffer = this.buffer;
|
||
|
|
const destination = this.destination;
|
||
|
|
destination.remove(innerSub);
|
||
|
|
this.active--;
|
||
|
|
if (buffer && buffer.length > 0) {
|
||
|
|
this._next(buffer.shift());
|
||
|
|
}
|
||
|
|
if (this.hasCompleted && this.active === 0) {
|
||
|
|
this.destination.complete();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
//# sourceMappingURL=expand.js.map
|