stream: simplify howMuchToRead

nodejs · Aug 16, 2019 · df1756e · df1756e
1 parent a15cd9d
commit df1756e
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/lib/_stream_readable.js b/lib/_stream_readable.js
@@ -383,17 +383,9 @@ function howMuchToRead(n, state) {
     else
       return state.length;
   }
-  // If we're asking for more than the current hwm, then raise the hwm.
-  if (n > state.highWaterMark)
-    state.highWaterMark = computeNewHighWaterMark(n);
   if (n <= state.length)
     return n;
-  // Don't have enough
-  if (!state.ended) {
-    state.needReadable = true;
-    return 0;
-  }
-  return state.length;
+  return state.ended ? state.length : 0;
 }
 
 // You can override either this method, or the async _read(n) below.
@@ -409,6 +401,10 @@ Readable.prototype.read = function(n) {
   const state = this._readableState;
   const nOrig = n;
 
+  // If we're asking for more than the current hwm, then raise the hwm.
+  if (n > state.highWaterMark)
+    state.highWaterMark = computeNewHighWaterMark(n);
+
   if (n !== 0)
     state.emittedReadable = false;