diff options
Diffstat (limited to 'lib/_stream_writable.js')
-rw-r--r-- | lib/_stream_writable.js | 37 |
1 files changed, 21 insertions, 16 deletions
diff --git a/lib/_stream_writable.js b/lib/_stream_writable.js index c6d895ff5d..9b75b672cb 100644 --- a/lib/_stream_writable.js +++ b/lib/_stream_writable.js @@ -158,6 +158,11 @@ function WritableState(options, stream, isDuplex) { // Should .destroy() be called after 'finish' (and potentially 'end') this.autoDestroy = !!(options && options.autoDestroy); + // Indicates whether the stream has errored. When true all write() calls + // should return false. This is needed since when autoDestroy + // is disabled we need a way to tell whether the stream has failed. + this.errored = false; + // Count buffered requests this.bufferedRequestCount = 0; @@ -401,7 +406,7 @@ function writeOrBuffer(stream, state, isBuf, chunk, encoding, cb) { if (!ret) state.needDrain = true; - if (state.writing || state.corked) { + if (state.writing || state.corked || state.errored) { var last = state.lastBufferedRequest; state.lastBufferedRequest = { chunk, @@ -420,7 +425,9 @@ function writeOrBuffer(stream, state, isBuf, chunk, encoding, cb) { doWrite(stream, state, false, len, chunk, encoding, cb); } - return ret; + // Return false if errored or destroyed in order to break + // any synchronous while(stream.write(data)) loops. + return ret && !state.errored && !state.destroyed; } function doWrite(stream, state, writev, len, chunk, encoding, cb) { @@ -437,18 +444,11 @@ function doWrite(stream, state, writev, len, chunk, encoding, cb) { state.sync = false; } -function onwriteError(stream, state, sync, er, cb) { +function onwriteError(stream, state, er, cb) { --state.pendingcb; - if (sync) { - // Defer the callback if we are being called synchronously - // to avoid piling up things on the stack - process.nextTick(cb, er); - } else { - // The caller expect this to happen before if - // it is async - cb(er); - } + cb(er); + // This can emit error, but error must always follow cb. errorOrDestroy(stream, er); } @@ -465,9 +465,14 @@ function onwrite(stream, er) { state.length -= state.writelen; state.writelen = 0; - if (er) - onwriteError(stream, state, sync, er, cb); - else { + if (er) { + state.errored = true; + if (sync) { + process.nextTick(onwriteError, stream, state, er, cb); + } else { + onwriteError(stream, state, er, cb); + } + } else { // Check if we're actually ready to finish, but don't emit yet var finished = needFinish(state) || stream.destroyed; @@ -622,7 +627,7 @@ Object.defineProperty(Writable.prototype, 'writableLength', { function needFinish(state) { return (state.ending && state.length === 0 && - !state.errorEmitted && + !state.errored && state.bufferedRequest === null && !state.finished && !state.writing); |