http2: reuse ._onTimeout() in Http2Session and Http2Stream classes

PR-URL: #33354 Reviewed-By: Benjamin Gruenbaum <[email protected]> Reviewed-By: Ruben Bridgewater <[email protected]>
nodejs · May 23, 2020 · 7a094dc · 7a094dc
1 parent 49745cd
commit 7a094dc
Showing 1 changed file with 26 additions and 40 deletions.
diff --git a/lib/internal/http2/core.js b/lib/internal/http2/core.js
@@ -1460,27 +1460,7 @@ class Http2Session extends EventEmitter {
   }
 
   _onTimeout() {
-    // If the session is destroyed, this should never actually be invoked,
-    // but just in case...
-    if (this.destroyed)
-      return;
-    // This checks whether a write is currently in progress and also whether
-    // that write is actually sending data across the write. The kHandle
-    // stored `chunksSentSinceLastWrite` is only updated when a timeout event
-    // happens, meaning that if a write is ongoing it should never equal the
-    // newly fetched, updated value.
-    if (this[kState].writeQueueSize > 0) {
-      const handle = this[kHandle];
-      const chunksSentSinceLastWrite = handle !== undefined ?
-        handle.chunksSentSinceLastWrite : null;
-      if (chunksSentSinceLastWrite !== null &&
-          chunksSentSinceLastWrite !== handle.updateChunksSent()) {
-        this[kUpdateTimer]();
-        return;
-      }
-    }
-
-    this.emit('timeout');
+    callTimeout(this);
   }
 
   ref() {
@@ -1909,25 +1889,7 @@ class Http2Stream extends Duplex {
   }
 
   _onTimeout() {
-    if (this.destroyed)
-      return;
-    // This checks whether a write is currently in progress and also whether
-    // that write is actually sending data across the write. The kHandle
-    // stored `chunksSentSinceLastWrite` is only updated when a timeout event
-    // happens, meaning that if a write is ongoing it should never equal the
-    // newly fetched, updated value.
-    if (this[kState].writeQueueSize > 0) {
-      const handle = this[kSession][kHandle];
-      const chunksSentSinceLastWrite = handle !== undefined ?
-        handle.chunksSentSinceLastWrite : null;
-      if (chunksSentSinceLastWrite !== null &&
-          chunksSentSinceLastWrite !== handle.updateChunksSent()) {
-        this[kUpdateTimer]();
-        return;
-      }
-    }
-
-    this.emit('timeout');
+    callTimeout(this, kSession);
   }
 
   // True if the HEADERS frame has been sent
@@ -2214,6 +2176,30 @@ class Http2Stream extends Duplex {
   }
 }
 
+function callTimeout(self, kSession) {
+  // If the session is destroyed, this should never actually be invoked,
+  // but just in case...
+  if (self.destroyed)
+    return;
+  // This checks whether a write is currently in progress and also whether
+  // that write is actually sending data across the write. The kHandle
+  // stored `chunksSentSinceLastWrite` is only updated when a timeout event
+  // happens, meaning that if a write is ongoing it should never equal the
+  // newly fetched, updated value.
+  if (self[kState].writeQueueSize > 0) {
+    const handle = kSession ? self[kSession][kHandle] : self[kHandle];
+    const chunksSentSinceLastWrite = handle !== undefined ?
+      handle.chunksSentSinceLastWrite : null;
+    if (chunksSentSinceLastWrite !== null &&
+      chunksSentSinceLastWrite !== handle.updateChunksSent()) {
+      self[kUpdateTimer]();
+      return;
+    }
+  }
+
+  self.emit('timeout');
+}
+
 function callStreamClose(stream) {
   stream.close();
 }