| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
1 parent 4bf3e1e commit 3191d29
3 files changed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1144,12 +1144,12 @@ OutgoingMessage.prototype._flush = function _flush() { | |||
| 1144 | 1144 | ||
| 1145 | 1145 | if (socket?.writable) { | |
| 1146 | 1146 | // There might be remaining data in this.output; write it out | |
| 1147 | - const ret = this._flushOutput(socket); | ||
| 1147 | + this._flushOutput(socket); | ||
| 1148 | 1148 | ||
| 1149 | 1149 | if (this.finished) { | |
| 1150 | 1150 | // This is a queue to the server or client to bring in the next this. | |
| 1151 | 1151 | this._finish(); | |
| 1152 | - } else if (ret && this[kNeedDrain]) { | ||
| 1152 | + } else if (this[kNeedDrain] && this.writableLength === 0) { | ||
| 1153 | 1153 | this[kNeedDrain] = false; | |
| 1154 | 1154 | this.emit('drain'); | |
| 1155 | 1155 | } | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -820,7 +820,13 @@ function socketOnDrain(socket, state) { | |||
| 820 | 820 | } | |
| 821 | 821 | ||
| 822 | 822 | const msg = socket._httpMessage; | |
| 823 | - if (msg && !msg.finished && msg[kNeedDrain]) { | ||
| 823 | + // Only emit 'drain' once the message has no data pending anywhere, so that | ||
| 824 | + // msg.writableLength === 0 when the event fires. socketOnDrain is called | ||
| 825 | + // synchronously from updateOutgoingData during _flushOutput, at which point | ||
| 826 | + // the bytes we just handed to the socket (or the stale outputSize) mean | ||
| 827 | + // the message is not actually drained yet - we wait for the socket's | ||
| 828 | + // own 'drain' event instead. | ||
| 829 | + if (msg && !msg.finished && msg[kNeedDrain] && msg.writableLength === 0) { | ||
| 824 | 830 | msg[kNeedDrain] = false; | |
| 825 | 831 | msg.emit('drain'); | |
| 826 | 832 | } | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -0,0 +1,58 @@ | |||
| 1 | + 'use strict'; | ||
| 2 | + // Regression test: when a pipelined ServerResponse (whose writes were | ||
| 3 | + // buffered in outputData while the socket belonged to a previous response) | ||
| 4 | + // is finally assigned its socket and flushed, 'drain' must not be emitted | ||
| 5 | + // until the socket's own buffer has actually drained. Previously, | ||
| 6 | + // socketOnDrain was called synchronously from _flushOutput via _onPendingData | ||
| 7 | + // and emitted 'drain' even though the bytes we just wrote were still sitting | ||
| 8 | + // in the socket's writable buffer, so res.writableLength was non-zero. | ||
| 9 | + | ||
| 10 | + const common = require('../common'); | ||
| 11 | + const http = require('http'); | ||
| 12 | + const net = require('net'); | ||
| 13 | + const assert = require('assert'); | ||
| 14 | + | ||
| 15 | + let step = 0; | ||
| 16 | + | ||
| 17 | + const server = http.createServer(common.mustCall((req, res) => { | ||
| 18 | + step++; | ||
| 19 | + | ||
| 20 | + if (step === 1) { | ||
| 21 | + // Keep the first response open briefly so the second is queued with | ||
| 22 | + // res.socket === null. | ||
| 23 | + res.writeHead(200, { 'Content-Type': 'text/plain' }); | ||
| 24 | + setTimeout(() => res.end('ok'), 50); | ||
| 25 | + return; | ||
| 26 | + } | ||
| 27 | + | ||
| 28 | + // Second (pipelined) response - queued in state.outgoing, no socket yet. | ||
| 29 | + assert.strictEqual(res.socket, null); | ||
| 30 | + | ||
| 31 | + res.writeHead(200, { 'Content-Type': 'text/plain' }); | ||
| 32 | + | ||
| 33 | + // Write past the response's highWaterMark so res.write() returns false | ||
| 34 | + // and kNeedDrain is set. Data is buffered in outputData. | ||
| 35 | + const chunk = Buffer.alloc(16 * 1024, 'x'); | ||
| 36 | + while (res.write(chunk)); | ||
| 37 | + assert.strictEqual(res.writableNeedDrain, true); | ||
| 38 | + | ||
| 39 | + res.on('drain', common.mustCall(() => { | ||
| 40 | + assert.strictEqual( | ||
| 41 | + res.writableLength, 0, | ||
| 42 | + `'drain' fired with writableLength=${res.writableLength}`, | ||
| 43 | + ); | ||
| 44 | + res.end(); | ||
| 45 | + server.close(); | ||
| 46 | + })); | ||
| 47 | + }, 2)); | ||
| 48 | + | ||
| 49 | + server.listen(0, common.mustCall(function() { | ||
| 50 | + const port = this.address().port; | ||
| 51 | + const client = net.connect(port); | ||
| 52 | + client.write( | ||
| 53 | + `GET /1 HTTP/1.1\r\nHost: localhost:${port}\r\n\r\n` + | ||
| 54 | + `GET /2 HTTP/1.1\r\nHost: localhost:${port}\r\n\r\n`, | ||
| 55 | + ); | ||
| 56 | + client.resume(); | ||
| 57 | + client.on('error', () => {}); | ||
| 58 | + })); | ||
| Back | FazBrowse Home | New Git URL |
0 commit comments