42 lines
1.3 KiB
JavaScript
42 lines
1.3 KiB
JavaScript
|
import { Collector } from "./collector";
|
||
|
|
export const streamCollector = (stream) => {
|
||
|
|
if (isReadableStreamInstance(stream)) {
|
||
|
|
return collectReadableStream(stream);
|
||
|
|
}
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
const collector = new Collector();
|
||
|
|
stream.pipe(collector);
|
||
|
|
stream.on("error", (err) => {
|
||
|
|
collector.end();
|
||
|
|
reject(err);
|
||
|
|
});
|
||
|
|
collector.on("error", reject);
|
||
|
|
collector.on("finish", function () {
|
||
|
|
const bytes = new Uint8Array(Buffer.concat(this.bufferedBytes));
|
||
|
|
resolve(bytes);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
const isReadableStreamInstance = (stream) => typeof ReadableStream === "function" && stream instanceof ReadableStream;
|
||
|
|
async function collectReadableStream(stream) {
|
||
|
|
const chunks = [];
|
||
|
|
const reader = stream.getReader();
|
||
|
|
let isDone = false;
|
||
|
|
let length = 0;
|
||
|
|
while (!isDone) {
|
||
|
|
const { done, value } = await reader.read();
|
||
|
|
if (value) {
|
||
|
|
chunks.push(value);
|
||
|
|
length += value.length;
|
||
|
|
}
|
||
|
|
isDone = done;
|
||
|
|
}
|
||
|
|
const collected = new Uint8Array(length);
|
||
|
|
let offset = 0;
|
||
|
|
for (const chunk of chunks) {
|
||
|
|
collected.set(chunk, offset);
|
||
|
|
offset += chunk.length;
|
||
|
|
}
|
||
|
|
return collected;
|
||
|
|
}
|