UNPKG

@reactivex/ix-esnext-esm

Version:

The Interactive Extensions for JavaScript

56 lines (54 loc) 1.96 kB
import { AsyncIterableX, interval, concat, of } from '../index.js'; import { map } from './map.js'; import { merge } from '../merge.js'; import { wrapWithAbort } from './withabort.js'; const timerEvent = {}; const ended = {}; class BufferCountOrTime extends AsyncIterableX { source; bufferSize; maxWaitTime; constructor(source, bufferSize, maxWaitTime) { super(); this.source = source; this.bufferSize = bufferSize; this.maxWaitTime = maxWaitTime; } async *[Symbol.asyncIterator](signal) { const buffer = []; const timer = interval(this.maxWaitTime).pipe(map(() => timerEvent)); const source = concat(this.source, of(ended)); const merged = merge(source, timer); for await (const item of wrapWithAbort(merged, signal)) { if (item === ended) { break; } if (item !== timerEvent) { buffer.push(item); } if (buffer.length >= this.bufferSize || (buffer.length && item === timerEvent)) { yield buffer.slice(); buffer.length = 0; } } if (buffer.length) { yield buffer; } } } /** * Projects each element of an async-iterable sequence into consecutive buffers * which are emitted when either the threshold count or time is met. * * @template TSource The type of elements in the source sequence. * @param {number} count The size of the buffer. * @param {number} time The threshold number of milliseconds to wait before flushing a non-full buffer * @returns {OperatorAsyncFunction<TSource, TSource[]>} An operator which returns an async-iterable sequence * of buffers */ export function bufferCountOrTime(count, time) { return function bufferOperatorFunction(source) { return new BufferCountOrTime(source, count, time); }; } //# sourceMappingURL=buffercountortime.js.map