stream: move legacy to sep file

nodejs · Jan 6, 2017 · 625bf71 · 625bf71
1 parent f43ea2a
commit 625bf71
Show file tree

Hide file tree

Showing 3 changed files with 101 additions and 92 deletions.
diff --git a/lib/_stream_legacy.js b/lib/_stream_legacy.js
@@ -0,0 +1,91 @@
+'use strict';
+
+module.exports = function(Stream) {
+  // old-style streams.  Note that the pipe method (the only relevant
+  // part of this class) is overridden in the Readable class.
+
+  Stream.prototype.pipe = function(dest, options) {
+    var source = this;
+
+    function ondata(chunk) {
+      if (dest.writable) {
+        if (false === dest.write(chunk) && source.pause) {
+          source.pause();
+        }
+      }
+    }
+
+    source.on('data', ondata);
+
+    function ondrain() {
+      if (source.readable && source.resume) {
+        source.resume();
+      }
+    }
+
+    dest.on('drain', ondrain);
+
+    // If the 'end' option is not supplied, dest.end() will be called when
+    // source gets the 'end' or 'close' events.  Only dest.end() once.
+    if (!dest._isStdio && (!options || options.end !== false)) {
+      source.on('end', onend);
+      source.on('close', onclose);
+    }
+
+    var didOnEnd = false;
+    function onend() {
+      if (didOnEnd) return;
+      didOnEnd = true;
+
+      dest.end();
+    }
+
+
+    function onclose() {
+      if (didOnEnd) return;
+      didOnEnd = true;
+
+      if (typeof dest.destroy === 'function') dest.destroy();
+    }
+
+    // don't leave dangling pipes when there are errors.
+    function onerror(er) {
+      cleanup();
+      if (EE.listenerCount(this, 'error') === 0) {
+        throw er; // Unhandled stream error in pipe.
+      }
+    }
+
+    source.on('error', onerror);
+    dest.on('error', onerror);
+
+    // remove all the event listeners that were added.
+    function cleanup() {
+      source.removeListener('data', ondata);
+      dest.removeListener('drain', ondrain);
+
+      source.removeListener('end', onend);
+      source.removeListener('close', onclose);
+
+      source.removeListener('error', onerror);
+      dest.removeListener('error', onerror);
+
+      source.removeListener('end', cleanup);
+      source.removeListener('close', cleanup);
+
+      dest.removeListener('close', cleanup);
+    }
+
+    source.on('end', cleanup);
+    source.on('close', cleanup);
+
+    dest.on('close', cleanup);
+
+    dest.emit('pipe', source);
+
+    // Allow for unix-like usage: A.pipe(B).pipe(C)
+    return dest;
+  };
+
+  return Stream;
+};
diff --git a/lib/stream.js b/lib/stream.js
@@ -1,11 +1,16 @@
 'use strict';
 
-module.exports = Stream;
-
 const EE = require('events');
 const util = require('util');
 
+function Stream() {
+  EE.call(this);
+}
 util.inherits(Stream, EE);
+
+// wrap the old-style stream
+require('_stream_legacy')(Stream);
+
 Stream.Readable = require('_stream_readable');
 Stream.Writable = require('_stream_writable');
 Stream.Duplex = require('_stream_duplex');
@@ -15,93 +20,5 @@ Stream.PassThrough = require('_stream_passthrough');
 // Backwards-compat with node 0.4.x
 Stream.Stream = Stream;
 
-
-// old-style streams.  Note that the pipe method (the only relevant
-// part of this class) is overridden in the Readable class.
-
-function Stream() {
-  EE.call(this);
-}
-
-Stream.prototype.pipe = function(dest, options) {
-  var source = this;
-
-  function ondata(chunk) {
-    if (dest.writable) {
-      if (false === dest.write(chunk) && source.pause) {
-        source.pause();
-      }
-    }
-  }
-
-  source.on('data', ondata);
-
-  function ondrain() {
-    if (source.readable && source.resume) {
-      source.resume();
-    }
-  }
-
-  dest.on('drain', ondrain);
-
-  // If the 'end' option is not supplied, dest.end() will be called when
-  // source gets the 'end' or 'close' events.  Only dest.end() once.
-  if (!dest._isStdio && (!options || options.end !== false)) {
-    source.on('end', onend);
-    source.on('close', onclose);
-  }
-
-  var didOnEnd = false;
-  function onend() {
-    if (didOnEnd) return;
-    didOnEnd = true;
-
-    dest.end();
-  }
-
-
-  function onclose() {
-    if (didOnEnd) return;
-    didOnEnd = true;
-
-    if (typeof dest.destroy === 'function') dest.destroy();
-  }
-
-  // don't leave dangling pipes when there are errors.
-  function onerror(er) {
-    cleanup();
-    if (EE.listenerCount(this, 'error') === 0) {
-      throw er; // Unhandled stream error in pipe.
-    }
-  }
-
-  source.on('error', onerror);
-  dest.on('error', onerror);
-
-  // remove all the event listeners that were added.
-  function cleanup() {
-    source.removeListener('data', ondata);
-    dest.removeListener('drain', ondrain);
-
-    source.removeListener('end', onend);
-    source.removeListener('close', onclose);
-
-    source.removeListener('error', onerror);
-    dest.removeListener('error', onerror);
-
-    source.removeListener('end', cleanup);
-    source.removeListener('close', cleanup);
-
-    dest.removeListener('close', cleanup);
-  }
-
-  source.on('end', cleanup);
-  source.on('close', cleanup);
-
-  dest.on('close', cleanup);
-
-  dest.emit('pipe', source);
-
-  // Allow for unix-like usage: A.pipe(B).pipe(C)
-  return dest;
-};
+// export
+module.exports = Stream;
diff --git a/node.gyp b/node.gyp
@@ -61,6 +61,7 @@
       'lib/_stream_transform.js',
       'lib/_stream_passthrough.js',
       'lib/_stream_wrap.js',
+      'lib/_stream_legacy.js',
       'lib/string_decoder.js',
       'lib/sys.js',
       'lib/timers.js',