0d073fa89e
- Add GETTING_STARTED.md with quick start guide and development modes - Add INSTALL.sh automated installation script - Add INSTALLATION_CHECKLIST.md, INSTALLATION_SUCCESS.md, and INSTALLATION_SUMMARY.md - Add QUICK_REFERENCE.md for common commands - Add SETUP_GUIDE.md with detailed setup instructions - Update README.md with improved project overview - Add did-wallet app dependencies and node_modules
111 lines
2.4 KiB
TypeScript
111 lines
2.4 KiB
TypeScript
/**
|
|
* @packageDocumentation
|
|
*
|
|
* The final batch may be smaller than the max.
|
|
*
|
|
* @example
|
|
*
|
|
* ```javascript
|
|
* import batch from 'it-batch'
|
|
* import all from 'it-all'
|
|
*
|
|
* // This can also be an iterator, generator, etc
|
|
* const values = [0, 1, 2, 3, 4]
|
|
* const batchSize = 2
|
|
*
|
|
* const result = all(batch(values, batchSize))
|
|
*
|
|
* console.info(result) // [0, 1], [2, 3], [4]
|
|
* ```
|
|
*
|
|
* Async sources must be awaited:
|
|
*
|
|
* ```javascript
|
|
* import batch from 'it-batch'
|
|
* import all from 'it-all'
|
|
*
|
|
* const values = async function * () {
|
|
* yield * [0, 1, 2, 3, 4]
|
|
* }
|
|
*
|
|
* const batchSize = 2
|
|
* const result = await all(batch(values(), batchSize))
|
|
*
|
|
* console.info(result) // [0, 1], [2, 3], [4]
|
|
* ```
|
|
*/
|
|
|
|
function isAsyncIterable <T> (thing: any): thing is AsyncIterable<T> {
|
|
return thing[Symbol.asyncIterator] != null
|
|
}
|
|
|
|
/**
|
|
* Takes an (async) iterable that emits things and returns an async iterable that
|
|
* emits those things in fixed-sized batches
|
|
*/
|
|
function batch <T> (source: Iterable<T>, size?: number): Generator<T[], void, undefined>
|
|
function batch <T> (source: Iterable<T> | AsyncIterable<T>, size?: number): AsyncGenerator<T[], void, undefined>
|
|
function batch <T> (source: Iterable<T> | AsyncIterable<T>, size: number = 1): Generator<T[], void, undefined> | AsyncGenerator<T[], void, undefined> {
|
|
size = Number(size)
|
|
|
|
if (isAsyncIterable(source)) {
|
|
return (async function * () {
|
|
let things: T[] = []
|
|
|
|
if (size < 1) {
|
|
size = 1
|
|
}
|
|
|
|
if (size !== Math.round(size)) {
|
|
throw new Error('Batch size must be an integer')
|
|
}
|
|
|
|
for await (const thing of source) {
|
|
things.push(thing)
|
|
|
|
while (things.length >= size) {
|
|
yield things.slice(0, size)
|
|
|
|
things = things.slice(size)
|
|
}
|
|
}
|
|
|
|
while (things.length > 0) {
|
|
yield things.slice(0, size)
|
|
|
|
things = things.slice(size)
|
|
}
|
|
}())
|
|
}
|
|
|
|
return (function * () {
|
|
let things: T[] = []
|
|
|
|
if (size < 1) {
|
|
size = 1
|
|
}
|
|
|
|
if (size !== Math.round(size)) {
|
|
throw new Error('Batch size must be an integer')
|
|
}
|
|
|
|
for (const thing of source) {
|
|
things.push(thing)
|
|
|
|
while (things.length >= size) {
|
|
yield things.slice(0, size)
|
|
|
|
things = things.slice(size)
|
|
}
|
|
}
|
|
|
|
while (things.length > 0) {
|
|
yield things.slice(0, size)
|
|
|
|
things = things.slice(size)
|
|
}
|
|
}())
|
|
}
|
|
|
|
export default batch
|