Skip to content

Commit

Permalink
feat(distinctUntilChanged): add higher-order lettable version of dist…
Browse files Browse the repository at this point in the history
…inctUntilChanged
  • Loading branch information
benlesh committed Jul 25, 2017
1 parent cb91c3f commit b2725e7
Show file tree
Hide file tree
Showing 3 changed files with 121 additions and 68 deletions.
71 changes: 3 additions & 68 deletions src/operator/distinctUntilChanged.ts
Original file line number Diff line number Diff line change
@@ -1,9 +1,6 @@
import { Operator } from '../Operator';
import { Subscriber } from '../Subscriber';
import { tryCatch } from '../util/tryCatch';
import { errorObject } from '../util/errorObject';

import { Observable } from '../Observable';
import { TeardownLogic } from '../Subscription';
import { distinctUntilChanged as higherOrder } from '../operators';

/* tslint:disable:max-line-length */
export function distinctUntilChanged<T>(this: Observable<T>, compare?: (x: T, y: T) => boolean): Observable<T>;
Expand Down Expand Up @@ -50,67 +47,5 @@ export function distinctUntilChanged<T, K>(this: Observable<T>, compare: (x: K,
* @owner Observable
*/
export function distinctUntilChanged<T, K>(this: Observable<T>, compare?: (x: K, y: K) => boolean, keySelector?: (x: T) => K): Observable<T> {
return this.lift(new DistinctUntilChangedOperator<T, K>(compare, keySelector));
}

class DistinctUntilChangedOperator<T, K> implements Operator<T, T> {
constructor(private compare: (x: K, y: K) => boolean,
private keySelector: (x: T) => K) {
}

call(subscriber: Subscriber<T>, source: any): TeardownLogic {
return source.subscribe(new DistinctUntilChangedSubscriber(subscriber, this.compare, this.keySelector));
}
}

/**
* We need this JSDoc comment for affecting ESDoc.
* @ignore
* @extends {Ignored}
*/
class DistinctUntilChangedSubscriber<T, K> extends Subscriber<T> {
private key: K;
private hasKey: boolean = false;

constructor(destination: Subscriber<T>,
compare: (x: K, y: K) => boolean,
private keySelector: (x: T) => K) {
super(destination);
if (typeof compare === 'function') {
this.compare = compare;
}
}

private compare(x: any, y: any): boolean {
return x === y;
}

protected _next(value: T): void {

const keySelector = this.keySelector;
let key: any = value;

if (keySelector) {
key = tryCatch(this.keySelector)(value);
if (key === errorObject) {
return this.destination.error(errorObject.e);
}
}

let result: any = false;

if (this.hasKey) {
result = tryCatch(this.compare)(this.key, key);
if (result === errorObject) {
return this.destination.error(errorObject.e);
}
} else {
this.hasKey = true;
}

if (Boolean(result) === false) {
this.key = key;
this.destination.next(value);
}
}
return higherOrder(compare, keySelector)(this);
}
117 changes: 117 additions & 0 deletions src/operators/distinctUntilChanged.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,117 @@
import { Operator } from '../Operator';
import { Subscriber } from '../Subscriber';
import { tryCatch } from '../util/tryCatch';
import { errorObject } from '../util/errorObject';
import { Observable } from '../Observable';
import { TeardownLogic } from '../Subscription';
import { MonoTypeOperatorFunction } from '../interfaces';

/* tslint:disable:max-line-length */
export function distinctUntilChanged<T>(compare?: (x: T, y: T) => boolean): MonoTypeOperatorFunction<T>;
export function distinctUntilChanged<T, K>(compare: (x: K, y: K) => boolean, keySelector: (x: T) => K): MonoTypeOperatorFunction<T>;
/* tslint:enable:max-line-length */

/**
* Returns an Observable that emits all items emitted by the source Observable that are distinct by comparison from the previous item.
*
* If a comparator function is provided, then it will be called for each item to test for whether or not that value should be emitted.
*
* If a comparator function is not provided, an equality check is used by default.
*
* @example <caption>A simple example with numbers</caption>
* Observable.of(1, 1, 2, 2, 2, 1, 1, 2, 3, 3, 4)
* .distinctUntilChanged()
* .subscribe(x => console.log(x)); // 1, 2, 1, 2, 3, 4
*
* @example <caption>An example using a compare function</caption>
* interface Person {
* age: number,
* name: string
* }
*
* Observable.of<Person>(
* { age: 4, name: 'Foo'},
* { age: 7, name: 'Bar'},
* { age: 5, name: 'Foo'})
* { age: 6, name: 'Foo'})
* .distinctUntilChanged((p: Person, q: Person) => p.name === q.name)
* .subscribe(x => console.log(x));
*
* // displays:
* // { age: 4, name: 'Foo' }
* // { age: 7, name: 'Bar' }
* // { age: 5, name: 'Foo' }
*
* @see {@link distinct}
* @see {@link distinctUntilKeyChanged}
*
* @param {function} [compare] Optional comparison function called to test if an item is distinct from the previous item in the source.
* @return {Observable} An Observable that emits items from the source Observable with distinct values.
* @method distinctUntilChanged
* @owner Observable
*/
export function distinctUntilChanged<T, K>(compare?: (x: K, y: K) => boolean, keySelector?: (x: T) => K): MonoTypeOperatorFunction<T> {
return (source: Observable<T>) => source.lift(new DistinctUntilChangedOperator<T, K>(compare, keySelector));
}

class DistinctUntilChangedOperator<T, K> implements Operator<T, T> {
constructor(private compare: (x: K, y: K) => boolean,
private keySelector: (x: T) => K) {
}

call(subscriber: Subscriber<T>, source: any): TeardownLogic {
return source.subscribe(new DistinctUntilChangedSubscriber(subscriber, this.compare, this.keySelector));
}
}

/**
* We need this JSDoc comment for affecting ESDoc.
* @ignore
* @extends {Ignored}
*/
class DistinctUntilChangedSubscriber<T, K> extends Subscriber<T> {
private key: K;
private hasKey: boolean = false;

constructor(destination: Subscriber<T>,
compare: (x: K, y: K) => boolean,
private keySelector: (x: T) => K) {
super(destination);
if (typeof compare === 'function') {
this.compare = compare;
}
}

private compare(x: any, y: any): boolean {
return x === y;
}

protected _next(value: T): void {

const keySelector = this.keySelector;
let key: any = value;

if (keySelector) {
key = tryCatch(this.keySelector)(value);
if (key === errorObject) {
return this.destination.error(errorObject.e);
}
}

let result: any = false;

if (this.hasKey) {
result = tryCatch(this.compare)(this.key, key);
if (result === errorObject) {
return this.destination.error(errorObject.e);
}
} else {
this.hasKey = true;
}

if (Boolean(result) === false) {
this.key = key;
this.destination.next(value);
}
}
}
1 change: 1 addition & 0 deletions src/operators/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ export { defaultIfEmpty } from './defaultIfEmpty';
export { delay } from './delay';
export { delayWhen } from './delayWhen';
export { dematerialize } from './dematerialize';
export { distinctUntilChanged } from './distinctUntilChanged';
export { filter } from './filter';
export { ignoreElements } from './ignoreElements';
export { map } from './map';
Expand Down

0 comments on commit b2725e7

Please sign in to comment.