You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
788 B
40 lines
788 B
3 years ago
|
/* eslint no-constant-condition: 0 */
|
||
|
|
/**
|
||
|
|
* Obliterator Take Function
|
||
|
|
* ==========================
|
||
|
|
*
|
||
|
|
* Function taking n or every value of the given iterator and returns them
|
||
|
|
* into an array.
|
||
|
|
*/
|
||
|
|
var iter = require('./iter.js');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Take.
|
||
|
|
*
|
||
|
|
* @param {Iterable} iterable - Target iterable.
|
||
|
|
* @param {number} [n] - Optional number of items to take.
|
||
|
|
* @return {array}
|
||
|
|
*/
|
||
|
|
module.exports = function take(iterable, n) {
|
||
|
|
var l = arguments.length > 1 ? n : Infinity,
|
||
|
|
array = l !== Infinity ? new Array(l) : [],
|
||
|
|
step,
|
||
|
|
i = 0;
|
||
|
|
|
||
|
|
var iterator = iter(iterable);
|
||
|
|
|
||
|
|
while (true) {
|
||
|
|
if (i === l) return array;
|
||
|
|
|
||
|
|
step = iterator.next();
|
||
|
|
|
||
|
|
if (step.done) {
|
||
|
|
if (i !== n) array.length = i;
|
||
|
|
|
||
|
|
return array;
|
||
|
|
}
|
||
|
|
|
||
|
|
array[i++] = step.value;
|
||
|
|
}
|
||
|
|
};
|