next
Version:
The React Framework
74 lines (73 loc) • 2.13 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.
*/ "use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
0 && (module.exports = {
CloseController: null,
trackBodyConsumed: null,
trackStreamConsumed: null
});
function _export(target, all) {
for(var name in all)Object.defineProperty(target, name, {
enumerable: true,
get: all[name]
});
}
_export(exports, {
CloseController: function() {
return CloseController;
},
trackBodyConsumed: function() {
return trackBodyConsumed;
},
trackStreamConsumed: function() {
return trackStreamConsumed;
}
});
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);
}
}
function trackStreamConsumed(stream, onEnd) {
const closePassThrough = new TransformStream({
flush: ()=>{
return onEnd();
}
});
return stream.pipeThrough(closePassThrough);
}
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