stream: combine properties using defineProperties

nodejs · Feb 10, 2020 · 36c6219 · 36c6219
1 parent d0413ae
commit 36c6219
Showing 1 changed file with 61 additions and 94 deletions.
diff --git a/lib/_stream_writable.js b/lib/_stream_writable.js
@@ -30,6 +30,7 @@ const {
   Boolean,
   FunctionPrototype,
   ObjectDefineProperty,
+  ObjectDefineProperties,
   ObjectSetPrototypeOf,
   Symbol,
   SymbolHasInstance,
@@ -336,46 +337,6 @@ Writable.prototype.setDefaultEncoding = function setDefaultEncoding(encoding) {
   return this;
 };
 
-ObjectDefineProperty(Writable.prototype, 'writableBuffer', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get: function() {
-    return this._writableState && this._writableState.getBuffer();
-  }
-});
-
-ObjectDefineProperty(Writable.prototype, 'writableEnded', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get: function() {
-    return this._writableState ? this._writableState.ending : false;
-  }
-});
-
-ObjectDefineProperty(Writable.prototype, 'writableHighWaterMark', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get: function() {
-    return this._writableState && this._writableState.highWaterMark;
-  }
-});
-
-ObjectDefineProperty(Writable.prototype, 'writableCorked', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get: function() {
-    return this._writableState ? this._writableState.corked : 0;
-  }
-});
-
 // If we're already writing something, then just put this
 // in the queue, and wait our turn.  Otherwise, call _write
 // If we return false, then we need a drain event, so set that flag.
@@ -652,16 +613,6 @@ Writable.prototype.end = function(chunk, encoding, cb) {
   return this;
 };
 
-ObjectDefineProperty(Writable.prototype, 'writableLength', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get() {
-    return this._writableState.length;
-  }
-});
-
 function needFinish(state) {
   return (state.ending &&
           state.length === 0 &&
@@ -774,59 +725,75 @@ function onFinished(stream, state, cb) {
   stream.prependListener('error', onerror);
 }
 
-ObjectDefineProperty(Writable.prototype, 'writable', {
-  get() {
-    const w = this._writableState;
-    if (!w) return false;
-    if (w.writable !== undefined) return w.writable;
-    return Boolean(!w.destroyed && !w.errored && !w.ending);
+ObjectDefineProperties(Writable.prototype, {
+
+  destroyed: {
+    get() {
+      return this._writableState ? this._writableState.destroyed : false;
+    },
+    set(value) {
+      // Backward compatibility, the user is explicitly managing destroyed
+      if (this._writableState) {
+        this._writableState.destroyed = value;
+      }
+    }
+  },
+
+  writable: {
+    get() {
+      const w = this._writableState;
+      if (!w) return false;
+      if (w.writable !== undefined) return w.writable;
+      return Boolean(!w.destroyed && !w.errored && !w.ending);
+    },
+    set(val) {
+      // Backwards compatible.
+      if (this._writableState) {
+        this._writableState.writable = !!val;
+      }
+    }
   },
-  set(val) {
-    // Backwards compat.
-    if (this._writableState) {
-      this._writableState.writable = !!val;
+
+  writableFinished: {
+    get() {
+      return this._writableState ? this._writableState.finished : false;
     }
-  }
-});
+  },
 
-ObjectDefineProperty(Writable.prototype, 'destroyed', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get() {
-    if (this._writableState === undefined) {
-      return false;
+  writableObjectMode: {
+    get() {
+      return this._writableState ? this._writableState.objectMode : false;
     }
-    return this._writableState.destroyed;
   },
-  set(value) {
-    // We ignore the value if the stream
-    // has not been initialized yet
-    if (!this._writableState) {
-      return;
+
+  writableBuffer: {
+    get() {
+      return this._writableState && this._writableState.getBuffer();
     }
+  },
 
-    // Backward compatibility, the user is explicitly
-    // managing destroyed
-    this._writableState.destroyed = value;
-  }
-});
+  writableEnded: {
+    get() {
+      return this._writableState ? this._writableState.ending : false;
+    }
+  },
 
-ObjectDefineProperty(Writable.prototype, 'writableObjectMode', {
-  enumerable: false,
-  get() {
-    return this._writableState ? this._writableState.objectMode : false;
-  }
-});
+  writableHighWaterMark: {
+    get() {
+      return this._writableState && this._writableState.highWaterMark;
+    }
+  },
 
-ObjectDefineProperty(Writable.prototype, 'writableFinished', {
-  // Making it explicit this property is not enumerable
-  // because otherwise some prototype manipulation in
-  // userland will fail
-  enumerable: false,
-  get() {
-    return this._writableState ? this._writableState.finished : false;
+  writableCorked: {
+    get() {
+      return this._writableState ? this._writableState.corked : 0;
+    }
+  },
+
+  writableLength: {
+    get() {
+      return this._writableState && this._writableState.length;
+    }
   }
 });