@reactivex/ix-esnext-esm
Version:
The Interactive Extensions for JavaScript
55 lines (53 loc) • 2.31 kB
JavaScript
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