UNPKG

@reactivex/ix-esnext-esm

Version:

The Interactive Extensions for JavaScript

55 lines (53 loc) 2.31 kB
import { AsyncIterableX } from '../asynciterablex'; import { identityAsync } from '../../util/identity'; import { createGrouping } from './_grouping'; import { throwIfAborted } from '../../aborterror'; export class GroupedAsyncIterable extends AsyncIterableX { constructor(key, source) { super(); this.key = key; this._source = source; } async *[Symbol.asyncIterator](signal) { throwIfAborted(signal); for (const item of this._source) { yield item; } } } export class GroupByAsyncIterable extends AsyncIterableX { constructor(source, keySelector, elementSelector) { super(); this._source = source; this._keySelector = keySelector; this._elementSelector = elementSelector; } async *[Symbol.asyncIterator](signal) { throwIfAborted(signal); const map = await createGrouping(this._source, this._keySelector, this._elementSelector, signal); for (const [key, values] of map) { yield new GroupedAsyncIterable(key, values); } } } /** * Groups the elements of an async-iterable sequence and selects the resulting elements by using a specified function. * * @export * @template TSource The type of the elements in the source sequence. * @template TKey The type of the grouping key computed for each element in the source sequence. * @template TValue The type of the elements within the groups computed for each element in the source sequence. * @param {((value: TSource, signal?: AbortSignal) => TKey | Promise<TKey>)} keySelector A function to extract the key for each element. * @param {(( * value: TSource, * signal?: AbortSignal * ) => TValue | Promise<TValue>)} [elementSelector=identityAsync] A function to map each source element to an element in an async-enumerable group. * @returns {OperatorAsyncFunction<TSource, TResult>} A sequence of async-iterable groups, each of which corresponds to a unique key value, * containing all elements that share that same key value. */ export function groupBy(keySelector, elementSelector = identityAsync) { return function groupByOperatorFunction(source) { return new GroupByAsyncIterable(source, keySelector, elementSelector); }; } //# sourceMappingURL=groupby.mjs.map