next
Version:
The React Framework
48 lines (47 loc) • 1.54 kB
JavaScript
/** Monitor when the consumer finishes reading the response body.
that's as close as we can get to `res.on('close')` using web APIs.
*/ export function trackBodyConsumed(body, onEnd) {
if (typeof body === 'string') {
const generator = async function* generate() {
const encoder = new TextEncoder();
yield encoder.encode(body);
onEnd();
};
// @ts-expect-error BodyInit typings doesn't seem to include AsyncIterables even though it's supported in practice
return generator();
} else {
return trackStreamConsumed(body, onEnd);
}
}
export function trackStreamConsumed(stream, onEnd) {
const closePassThrough = new TransformStream({
flush: ()=>{
return onEnd();
}
});
return stream.pipeThrough(closePassThrough);
}
export class CloseController {
onClose(callback) {
if (this.isClosed) {
throw new Error('Cannot subscribe to a closed CloseController');
}
this.target.addEventListener('close', callback);
this.listeners++;
}
dispatchClose() {
if (this.isClosed) {
throw new Error('Cannot close a CloseController multiple times');
}
if (this.listeners > 0) {
this.target.dispatchEvent(new Event('close'));
}
this.isClosed = true;
}
constructor(){
this.target = new EventTarget();
this.listeners = 0;
this.isClosed = false;
}
}
//# sourceMappingURL=web-on-close.js.map