Skip to content

Commit

Permalink
feat(windowToggle): add higher-order lettable version of windowToggle
Browse files Browse the repository at this point in the history
  • Loading branch information
benlesh committed Jun 25, 2017
1 parent 9e963aa commit 81ec389
Show file tree
Hide file tree
Showing 3 changed files with 210 additions and 161 deletions.
164 changes: 3 additions & 161 deletions src/operator/windowToggle.ts
Original file line number Diff line number Diff line change
@@ -1,15 +1,6 @@
import { Operator } from '../Operator';
import { Subscriber } from '../Subscriber';
import { Observable } from '../Observable';
import { Subject } from '../Subject';
import { Subscription } from '../Subscription';

import { tryCatch } from '../util/tryCatch';
import { errorObject } from '../util/errorObject';

import { OuterSubscriber } from '../OuterSubscriber';
import { InnerSubscriber } from '../InnerSubscriber';
import { subscribeToResult } from '../util/subscribeToResult';
import { Observable } from '../Observable';
import { windowToggle as higherOrder } from '../operators';

/**
* Branch out the source Observable values as a nested Observable starting from
Expand Down Expand Up @@ -54,154 +45,5 @@ import { subscribeToResult } from '../util/subscribeToResult';
*/
export function windowToggle<T, O>(this: Observable<T>, openings: Observable<O>,
closingSelector: (openValue: O) => Observable<any>): Observable<Observable<T>> {
return this.lift(new WindowToggleOperator<T, O>(openings, closingSelector));
}

class WindowToggleOperator<T, O> implements Operator<T, Observable<T>> {

constructor(private openings: Observable<O>,
private closingSelector: (openValue: O) => Observable<any>) {
}

call(subscriber: Subscriber<Observable<T>>, source: any): any {
return source.subscribe(new WindowToggleSubscriber(
subscriber, this.openings, this.closingSelector
));
}
}

interface WindowContext<T> {
window: Subject<T>;
subscription: Subscription;
}

/**
* We need this JSDoc comment for affecting ESDoc.
* @ignore
* @extends {Ignored}
*/
class WindowToggleSubscriber<T, O> extends OuterSubscriber<T, any> {
private contexts: WindowContext<T>[] = [];
private openSubscription: Subscription;

constructor(destination: Subscriber<Observable<T>>,
private openings: Observable<O>,
private closingSelector: (openValue: O) => Observable<any>) {
super(destination);
this.add(this.openSubscription = subscribeToResult(this, openings, openings));
}

protected _next(value: T) {
const { contexts } = this;
if (contexts) {
const len = contexts.length;
for (let i = 0; i < len; i++) {
contexts[i].window.next(value);
}
}
}

protected _error(err: any) {

const { contexts } = this;
this.contexts = null;

if (contexts) {
const len = contexts.length;
let index = -1;

while (++index < len) {
const context = contexts[index];
context.window.error(err);
context.subscription.unsubscribe();
}
}

super._error(err);
}

protected _complete() {
const { contexts } = this;
this.contexts = null;
if (contexts) {
const len = contexts.length;
let index = -1;
while (++index < len) {
const context = contexts[index];
context.window.complete();
context.subscription.unsubscribe();
}
}
super._complete();
}

protected _unsubscribe() {
const { contexts } = this;
this.contexts = null;
if (contexts) {
const len = contexts.length;
let index = -1;
while (++index < len) {
const context = contexts[index];
context.window.unsubscribe();
context.subscription.unsubscribe();
}
}
}

notifyNext(outerValue: any, innerValue: any,
outerIndex: number, innerIndex: number,
innerSub: InnerSubscriber<T, any>): void {

if (outerValue === this.openings) {

const { closingSelector } = this;
const closingNotifier = tryCatch(closingSelector)(innerValue);

if (closingNotifier === errorObject) {
return this.error(errorObject.e);
} else {
const window = new Subject<T>();
const subscription = new Subscription();
const context = { window, subscription };
this.contexts.push(context);
const innerSubscription = subscribeToResult(this, closingNotifier, context);

if (innerSubscription.closed) {
this.closeWindow(this.contexts.length - 1);
} else {
(<any> innerSubscription).context = context;
subscription.add(innerSubscription);
}

this.destination.next(window);

}
} else {
this.closeWindow(this.contexts.indexOf(outerValue));
}
}

notifyError(err: any): void {
this.error(err);
}

notifyComplete(inner: Subscription): void {
if (inner !== this.openSubscription) {
this.closeWindow(this.contexts.indexOf((<any> inner).context));
}
}

private closeWindow(index: number): void {
if (index === -1) {
return;
}

const { contexts } = this;
const context = contexts[index];
const { window, subscription } = context;
contexts.splice(index, 1);
window.complete();
subscription.unsubscribe();
}
return higherOrder(openings, closingSelector)(this);
}
1 change: 1 addition & 0 deletions src/operators/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -22,3 +22,4 @@ export { switchAll } from './switchAll';
export { switchMap } from './switchMap';
export { takeLast } from './takeLast';
export { tap } from './tap';
export { windowToggle } from './windowToggle';
206 changes: 206 additions & 0 deletions src/operators/windowToggle.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,206 @@
import { Operator } from '../Operator';
import { Subscriber } from '../Subscriber';
import { Observable } from '../Observable';
import { Subject } from '../Subject';
import { Subscription } from '../Subscription';
import { tryCatch } from '../util/tryCatch';
import { errorObject } from '../util/errorObject';
import { OuterSubscriber } from '../OuterSubscriber';
import { InnerSubscriber } from '../InnerSubscriber';
import { subscribeToResult } from '../util/subscribeToResult';
import { OperatorFunction } from '../interfaces';

/**
* Branch out the source Observable values as a nested Observable starting from
* an emission from `openings` and ending when the output of `closingSelector`
* emits.
*
* <span class="informal">It's like {@link bufferToggle}, but emits a nested
* Observable instead of an array.</span>
*
* <img src="./img/windowToggle.png" width="100%">
*
* Returns an Observable that emits windows of items it collects from the source
* Observable. The output Observable emits windows that contain those items
* emitted by the source Observable between the time when the `openings`
* Observable emits an item and when the Observable returned by
* `closingSelector` emits an item.
*
* @example <caption>Every other second, emit the click events from the next 500ms</caption>
* var clicks = Rx.Observable.fromEvent(document, 'click');
* var openings = Rx.Observable.interval(1000);
* var result = clicks.windowToggle(openings, i =>
* i % 2 ? Rx.Observable.interval(500) : Rx.Observable.empty()
* ).mergeAll();
* result.subscribe(x => console.log(x));
*
* @see {@link window}
* @see {@link windowCount}
* @see {@link windowTime}
* @see {@link windowWhen}
* @see {@link bufferToggle}
*
* @param {Observable<O>} openings An observable of notifications to start new
* windows.
* @param {function(value: O): Observable} closingSelector A function that takes
* the value emitted by the `openings` observable and returns an Observable,
* which, when it emits (either `next` or `complete`), signals that the
* associated window should complete.
* @return {Observable<Observable<T>>} An observable of windows, which in turn
* are Observables.
* @method windowToggle
* @owner Observable
*/
export function windowToggle<T, O>(openings: Observable<O>,
closingSelector: (openValue: O) => Observable<any>): OperatorFunction<T, Observable<T>> {
return (source: Observable<T>) => source.lift(new WindowToggleOperator<T, O>(openings, closingSelector));
}

class WindowToggleOperator<T, O> implements Operator<T, Observable<T>> {

constructor(private openings: Observable<O>,
private closingSelector: (openValue: O) => Observable<any>) {
}

call(subscriber: Subscriber<Observable<T>>, source: any): any {
return source.subscribe(new WindowToggleSubscriber(
subscriber, this.openings, this.closingSelector
));
}
}

interface WindowContext<T> {
window: Subject<T>;
subscription: Subscription;
}

/**
* We need this JSDoc comment for affecting ESDoc.
* @ignore
* @extends {Ignored}
*/
class WindowToggleSubscriber<T, O> extends OuterSubscriber<T, any> {
private contexts: WindowContext<T>[] = [];
private openSubscription: Subscription;

constructor(destination: Subscriber<Observable<T>>,
private openings: Observable<O>,
private closingSelector: (openValue: O) => Observable<any>) {
super(destination);
this.add(this.openSubscription = subscribeToResult(this, openings, openings));
}

protected _next(value: T) {
const { contexts } = this;
if (contexts) {
const len = contexts.length;
for (let i = 0; i < len; i++) {
contexts[i].window.next(value);
}
}
}

protected _error(err: any) {

const { contexts } = this;
this.contexts = null;

if (contexts) {
const len = contexts.length;
let index = -1;

while (++index < len) {
const context = contexts[index];
context.window.error(err);
context.subscription.unsubscribe();
}
}

super._error(err);
}

protected _complete() {
const { contexts } = this;
this.contexts = null;
if (contexts) {
const len = contexts.length;
let index = -1;
while (++index < len) {
const context = contexts[index];
context.window.complete();
context.subscription.unsubscribe();
}
}
super._complete();
}

protected _unsubscribe() {
const { contexts } = this;
this.contexts = null;
if (contexts) {
const len = contexts.length;
let index = -1;
while (++index < len) {
const context = contexts[index];
context.window.unsubscribe();
context.subscription.unsubscribe();
}
}
}

notifyNext(outerValue: any, innerValue: any,
outerIndex: number, innerIndex: number,
innerSub: InnerSubscriber<T, any>): void {

if (outerValue === this.openings) {

const { closingSelector } = this;
const closingNotifier = tryCatch(closingSelector)(innerValue);

if (closingNotifier === errorObject) {
return this.error(errorObject.e);
} else {
const window = new Subject<T>();
const subscription = new Subscription();
const context = { window, subscription };
this.contexts.push(context);
const innerSubscription = subscribeToResult(this, closingNotifier, context);

if (innerSubscription.closed) {
this.closeWindow(this.contexts.length - 1);
} else {
(<any> innerSubscription).context = context;
subscription.add(innerSubscription);
}

this.destination.next(window);

}
} else {
this.closeWindow(this.contexts.indexOf(outerValue));
}
}

notifyError(err: any): void {
this.error(err);
}

notifyComplete(inner: Subscription): void {
if (inner !== this.openSubscription) {
this.closeWindow(this.contexts.indexOf((<any> inner).context));
}
}

private closeWindow(index: number): void {
if (index === -1) {
return;
}

const { contexts } = this;
const context = contexts[index];
const { window, subscription } = context;
contexts.splice(index, 1);
window.complete();
subscription.unsubscribe();
}
}

0 comments on commit 81ec389

Please sign in to comment.