+++ /dev/null
-import { Subscriber } from '../Subscriber';
-import { Subscription } from '../Subscription';
-import { Observable } from '../Observable';
-import { Operator } from '../Operator';
-import { Subject } from '../Subject';
-import { OperatorFunction } from '../types';
-
-/* tslint:disable:max-line-length */
-export function groupBy<T, K>(keySelector: (value: T) => K): OperatorFunction<T, GroupedObservable<K, T>>;
-export function groupBy<T, K>(keySelector: (value: T) => K, elementSelector: void, durationSelector: (grouped: GroupedObservable<K, T>) => Observable<any>): OperatorFunction<T, GroupedObservable<K, T>>;
-export function groupBy<T, K, R>(keySelector: (value: T) => K, elementSelector?: (value: T) => R, durationSelector?: (grouped: GroupedObservable<K, R>) => Observable<any>): OperatorFunction<T, GroupedObservable<K, R>>;
-export function groupBy<T, K, R>(keySelector: (value: T) => K, elementSelector?: (value: T) => R, durationSelector?: (grouped: GroupedObservable<K, R>) => Observable<any>, subjectSelector?: () => Subject<R>): OperatorFunction<T, GroupedObservable<K, R>>;
-/* tslint:enable:max-line-length */
-
-/**
- * Groups the items emitted by an Observable according to a specified criterion,
- * and emits these grouped items as `GroupedObservables`, one
- * {@link GroupedObservable} per group.
- *
- * ![](groupBy.png)
- *
- * When the Observable emits an item, a key is computed for this item with the keySelector function.
- *
- * If a {@link GroupedObservable} for this key exists, this {@link GroupedObservable} emits. Elsewhere, a new
- * {@link GroupedObservable} for this key is created and emits.
- *
- * A {@link GroupedObservable} represents values belonging to the same group represented by a common key. The common
- * key is available as the key field of a {@link GroupedObservable} instance.
- *
- * The elements emitted by {@link GroupedObservable}s are by default the items emitted by the Observable, or elements
- * returned by the elementSelector function.
- *
- * ## Examples
- *
- * ### Group objects by id and return as array
- *
- * ```ts
- * import { of } from 'rxjs';
- * import { mergeMap, groupBy, reduce } from 'rxjs/operators';
- *
- * of(
- * {id: 1, name: 'JavaScript'},
- * {id: 2, name: 'Parcel'},
- * {id: 2, name: 'webpack'},
- * {id: 1, name: 'TypeScript'},
- * {id: 3, name: 'TSLint'}
- * ).pipe(
- * groupBy(p => p.id),
- * mergeMap((group$) => group$.pipe(reduce((acc, cur) => [...acc, cur], []))),
- * )
- * .subscribe(p => console.log(p));
- *
- * // displays:
- * // [ { id: 1, name: 'JavaScript'},
- * // { id: 1, name: 'TypeScript'} ]
- * //
- * // [ { id: 2, name: 'Parcel'},
- * // { id: 2, name: 'webpack'} ]
- * //
- * // [ { id: 3, name: 'TSLint'} ]
- * ```
- *
- * ### Pivot data on the id field
- *
- * ```ts
- * import { of } from 'rxjs';
- * import { groupBy, map, mergeMap, reduce } from 'rxjs/operators';
- *
- * of(
- * { id: 1, name: 'JavaScript' },
- * { id: 2, name: 'Parcel' },
- * { id: 2, name: 'webpack' },
- * { id: 1, name: 'TypeScript' },
- * { id: 3, name: 'TSLint' }
- * )
- * .pipe(
- * groupBy(p => p.id, p => p.name),
- * mergeMap(group$ =>
- * group$.pipe(reduce((acc, cur) => [...acc, cur], [`${group$.key}`]))
- * ),
- * map(arr => ({ id: parseInt(arr[0], 10), values: arr.slice(1) }))
- * )
- * .subscribe(p => console.log(p));
- *
- * // displays:
- * // { id: 1, values: [ 'JavaScript', 'TypeScript' ] }
- * // { id: 2, values: [ 'Parcel', 'webpack' ] }
- * // { id: 3, values: [ 'TSLint' ] }
- * ```
- *
- * @param {function(value: T): K} keySelector A function that extracts the key
- * for each item.
- * @param {function(value: T): R} [elementSelector] A function that extracts the
- * return element for each item.
- * @param {function(grouped: GroupedObservable<K,R>): Observable<any>} [durationSelector]
- * A function that returns an Observable to determine how long each group should
- * exist.
- * @return {Observable<GroupedObservable<K,R>>} An Observable that emits
- * GroupedObservables, each of which corresponds to a unique key value and each
- * of which emits those items from the source Observable that share that key
- * value.
- * @method groupBy
- * @owner Observable
- */
-export function groupBy<T, K, R>(keySelector: (value: T) => K,
- elementSelector?: ((value: T) => R) | void,
- durationSelector?: (grouped: GroupedObservable<K, R>) => Observable<any>,
- subjectSelector?: () => Subject<R>): OperatorFunction<T, GroupedObservable<K, R>> {
- return (source: Observable<T>) =>
- source.lift(new GroupByOperator(keySelector, elementSelector, durationSelector, subjectSelector));
-}
-
-export interface RefCountSubscription {
- count: number;
- unsubscribe: () => void;
- closed: boolean;
- attemptedToUnsubscribe: boolean;
-}
-
-class GroupByOperator<T, K, R> implements Operator<T, GroupedObservable<K, R>> {
- constructor(private keySelector: (value: T) => K,
- private elementSelector?: ((value: T) => R) | void,
- private durationSelector?: (grouped: GroupedObservable<K, R>) => Observable<any>,
- private subjectSelector?: () => Subject<R>) {
- }
-
- call(subscriber: Subscriber<GroupedObservable<K, R>>, source: any): any {
- return source.subscribe(new GroupBySubscriber(
- subscriber, this.keySelector, this.elementSelector, this.durationSelector, this.subjectSelector
- ));
- }
-}
-
-/**
- * We need this JSDoc comment for affecting ESDoc.
- * @ignore
- * @extends {Ignored}
- */
-class GroupBySubscriber<T, K, R> extends Subscriber<T> implements RefCountSubscription {
- private groups: Map<K, Subject<T | R>> = null;
- public attemptedToUnsubscribe: boolean = false;
- public count: number = 0;
-
- constructor(destination: Subscriber<GroupedObservable<K, R>>,
- private keySelector: (value: T) => K,
- private elementSelector?: ((value: T) => R) | void,
- private durationSelector?: (grouped: GroupedObservable<K, R>) => Observable<any>,
- private subjectSelector?: () => Subject<R>) {
- super(destination);
- }
-
- protected _next(value: T): void {
- let key: K;
- try {
- key = this.keySelector(value);
- } catch (err) {
- this.error(err);
- return;
- }
-
- this._group(value, key);
- }
-
- private _group(value: T, key: K) {
- let groups = this.groups;
-
- if (!groups) {
- groups = this.groups = new Map<K, Subject<T | R>>();
- }
-
- let group = groups.get(key);
-
- let element: R;
- if (this.elementSelector) {
- try {
- element = this.elementSelector(value);
- } catch (err) {
- this.error(err);
- }
- } else {
- element = <any>value;
- }
-
- if (!group) {
- group = (this.subjectSelector ? this.subjectSelector() : new Subject<R>()) as Subject<T | R>;
- groups.set(key, group);
- const groupedObservable = new GroupedObservable(key, group, this);
- this.destination.next(groupedObservable);
- if (this.durationSelector) {
- let duration: any;
- try {
- duration = this.durationSelector(new GroupedObservable<K, R>(key, <Subject<R>>group));
- } catch (err) {
- this.error(err);
- return;
- }
- this.add(duration.subscribe(new GroupDurationSubscriber(key, group, this)));
- }
- }
-
- if (!group.closed) {
- group.next(element);
- }
- }
-
- protected _error(err: any): void {
- const groups = this.groups;
- if (groups) {
- groups.forEach((group, key) => {
- group.error(err);
- });
-
- groups.clear();
- }
- this.destination.error(err);
- }
-
- protected _complete(): void {
- const groups = this.groups;
- if (groups) {
- groups.forEach((group, key) => {
- group.complete();
- });
-
- groups.clear();
- }
- this.destination.complete();
- }
-
- removeGroup(key: K): void {
- this.groups.delete(key);
- }
-
- unsubscribe() {
- if (!this.closed) {
- this.attemptedToUnsubscribe = true;
- if (this.count === 0) {
- super.unsubscribe();
- }
- }
- }
-}
-
-/**
- * We need this JSDoc comment for affecting ESDoc.
- * @ignore
- * @extends {Ignored}
- */
-class GroupDurationSubscriber<K, T> extends Subscriber<T> {
- constructor(private key: K,
- private group: Subject<T>,
- private parent: GroupBySubscriber<any, K, T | any>) {
- super(group);
- }
-
- protected _next(value: T): void {
- this.complete();
- }
-
- /** @deprecated This is an internal implementation detail, do not use. */
- _unsubscribe() {
- const { parent, key } = this;
- this.key = this.parent = null;
- if (parent) {
- parent.removeGroup(key);
- }
- }
-}
-
-/**
- * An Observable representing values belonging to the same group represented by
- * a common key. The values emitted by a GroupedObservable come from the source
- * Observable. The common key is available as the field `key` on a
- * GroupedObservable instance.
- *
- * @class GroupedObservable<K, T>
- */
-export class GroupedObservable<K, T> extends Observable<T> {
- /** @deprecated Do not construct this type. Internal use only */
- constructor(public key: K,
- private groupSubject: Subject<T>,
- private refCountSubscription?: RefCountSubscription) {
- super();
- }
-
- /** @deprecated This is an internal implementation detail, do not use. */
- _subscribe(subscriber: Subscriber<T>) {
- const subscription = new Subscription();
- const { refCountSubscription, groupSubject } = this;
- if (refCountSubscription && !refCountSubscription.closed) {
- subscription.add(new InnerRefCountSubscription(refCountSubscription));
- }
- subscription.add(groupSubject.subscribe(subscriber));
- return subscription;
- }
-}
-
-/**
- * We need this JSDoc comment for affecting ESDoc.
- * @ignore
- * @extends {Ignored}
- */
-class InnerRefCountSubscription extends Subscription {
- constructor(private parent: RefCountSubscription) {
- super();
- parent.count++;
- }
-
- unsubscribe() {
- const parent = this.parent;
- if (!parent.closed && !this.closed) {
- super.unsubscribe();
- parent.count -= 1;
- if (parent.count === 0 && parent.attemptedToUnsubscribe) {
- parent.unsubscribe();
- }
- }
- }
-}