| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- 'use strict'
- const { SymbolDispose } = require('../../ours/primordials')
- const { AbortError, codes } = require('../../ours/errors')
- const { isNodeStream, isWebStream, kControllerErrorFunction } = require('./utils')
- const eos = require('./end-of-stream')
- const { ERR_INVALID_ARG_TYPE } = codes
- let addAbortListener
- // This method is inlined here for readable-stream
- // It also does not allow for signal to not exist on the stream
- // https://github.com/nodejs/node/pull/36061#discussion_r533718029
- const validateAbortSignal = (signal, name) => {
- if (typeof signal !== 'object' || !('aborted' in signal)) {
- throw new ERR_INVALID_ARG_TYPE(name, 'AbortSignal', signal)
- }
- }
- module.exports.addAbortSignal = function addAbortSignal(signal, stream) {
- validateAbortSignal(signal, 'signal')
- if (!isNodeStream(stream) && !isWebStream(stream)) {
- throw new ERR_INVALID_ARG_TYPE('stream', ['ReadableStream', 'WritableStream', 'Stream'], stream)
- }
- return module.exports.addAbortSignalNoValidate(signal, stream)
- }
- module.exports.addAbortSignalNoValidate = function (signal, stream) {
- if (typeof signal !== 'object' || !('aborted' in signal)) {
- return stream
- }
- const onAbort = isNodeStream(stream)
- ? () => {
- stream.destroy(
- new AbortError(undefined, {
- cause: signal.reason
- })
- )
- }
- : () => {
- stream[kControllerErrorFunction](
- new AbortError(undefined, {
- cause: signal.reason
- })
- )
- }
- if (signal.aborted) {
- onAbort()
- } else {
- addAbortListener = addAbortListener || require('../../ours/util').addAbortListener
- const disposable = addAbortListener(signal, onAbort)
- eos(stream, disposable[SymbolDispose])
- }
- return stream
- }
|