61 lines
1.4 KiB
JavaScript
61 lines
1.4 KiB
JavaScript
// src/utils/stream.ts
|
|
var StreamingApi = class {
|
|
writer;
|
|
encoder;
|
|
writable;
|
|
abortSubscribers = [];
|
|
responseReadable;
|
|
constructor(writable, _readable) {
|
|
this.writable = writable;
|
|
this.writer = writable.getWriter();
|
|
this.encoder = new TextEncoder();
|
|
const reader = _readable.getReader();
|
|
this.abortSubscribers.push(async () => {
|
|
await reader.cancel();
|
|
});
|
|
this.responseReadable = new ReadableStream({
|
|
async pull(controller) {
|
|
const { done, value } = await reader.read();
|
|
done ? controller.close() : controller.enqueue(value);
|
|
},
|
|
cancel: () => {
|
|
this.abortSubscribers.forEach((subscriber) => subscriber());
|
|
}
|
|
});
|
|
}
|
|
async write(input) {
|
|
try {
|
|
if (typeof input === "string") {
|
|
input = this.encoder.encode(input);
|
|
}
|
|
await this.writer.write(input);
|
|
} catch (e) {
|
|
}
|
|
return this;
|
|
}
|
|
async writeln(input) {
|
|
await this.write(input + "\n");
|
|
return this;
|
|
}
|
|
sleep(ms) {
|
|
return new Promise((res) => setTimeout(res, ms));
|
|
}
|
|
async close() {
|
|
try {
|
|
await this.writer.close();
|
|
} catch (e) {
|
|
}
|
|
}
|
|
async pipe(body) {
|
|
this.writer.releaseLock();
|
|
await body.pipeTo(this.writable, { preventClose: true });
|
|
this.writer = this.writable.getWriter();
|
|
}
|
|
onAbort(listener) {
|
|
this.abortSubscribers.push(listener);
|
|
}
|
|
};
|
|
export {
|
|
StreamingApi
|
|
};
|