UNPKG

next

Version:

The React Framework

74 lines (73 loc) 2.13 kB
/** 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