1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
'use strict';
const from = require('from2');
const pIsPromise = require('p-is-promise');
module.exports = x => {
if (Array.isArray(x)) {
x = x.slice();
}
let promise;
let iterator;
prepare(x);
function prepare(value) {
x = value;
promise = pIsPromise(x) ? x : null;
// we don't iterate on strings and buffers since slicing them is ~7x faster
const shouldIterate = !promise && x[Symbol.iterator] && typeof x !== 'string' && !Buffer.isBuffer(x);
iterator = shouldIterate ? x[Symbol.iterator]() : null;
}
return from(function reader(size, cb) {
if (promise) {
promise.then(prepare).then(() => reader.call(this, size, cb), cb);
return;
}
if (iterator) {
const obj = iterator.next();
setImmediate(cb, null, obj.done ? null : obj.value);
return;
}
if (x.length === 0) {
setImmediate(cb, null, null);
return;
}
const chunk = x.slice(0, size);
x = x.slice(size);
setImmediate(cb, null, chunk);
});
};
module.exports.obj = x => {
if (Array.isArray(x)) {
x = x.slice();
}
let promise;
let iterator;
prepare(x);
function prepare(value) {
x = value;
promise = pIsPromise(x) ? x : null;
iterator = !promise && x[Symbol.iterator] ? x[Symbol.iterator]() : null;
}
return from.obj(function reader(size, cb) {
if (promise) {
promise.then(prepare).then(() => reader.call(this, size, cb), cb);
return;
}
if (iterator) {
const obj = iterator.next();
setImmediate(cb, null, obj.done ? null : obj.value);
return;
}
this.push(x);
setImmediate(cb, null, null);
});
};
|
