fixuP

nodejs · Oct 27, 2023 · 4e46fdf · 4e46fdf
1 parent 321c41c
commit 4e46fdf
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/lib/events.js b/lib/events.js
@@ -37,6 +37,7 @@ const {
   ObjectDefineProperty,
   ObjectDefineProperties,
   ObjectGetPrototypeOf,
+  ObjectKeys,
   ObjectSetPrototypeOf,
   Promise,
   PromiseReject,
@@ -87,6 +88,7 @@ const {
 
 const kCapture = Symbol('kCapture');
 const kErrorMonitor = Symbol('events.errorMonitor');
+const kShapeMode = Symbol('shapeMode');
 const kMaxEventTargetListeners = Symbol('events.maxEventTargetListeners');
 const kMaxEventTargetListenersWarned =
   Symbol('events.maxEventTargetListenersWarned');
@@ -344,6 +346,9 @@ EventEmitter.init = function(opts) {
       this._events === ObjectGetPrototypeOf(this)._events) {
     this._events = { __proto__: null };
     this._eventsCount = 0;
+    this[kShapeMode] = false;
+  } else {
+    this[kShapeMode] = true;
   }
 
   this._maxListeners = this._maxListeners || undefined;
@@ -686,9 +691,13 @@ EventEmitter.prototype.removeListener =
         return this;
 
       if (list === listener || list.listener === listener) {
-        if (--this._eventsCount === 0)
+        this._eventsCount -= 1;
+
+        if (this[kShapeMode]) {
+          events[type] = undefined;
+        } else if (this._eventsCount === 0) {
           this._events = { __proto__: null };
-        else {
+        } else {
           delete events[type];
           if (events.removeListener)
             this.emit('removeListener', type, list.listener || listener);
@@ -750,6 +759,7 @@ EventEmitter.prototype.removeAllListeners =
           else
             delete events[type];
         }
+        this[kShapeMode] = false;
         return this;
       }
 
@@ -762,6 +772,7 @@ EventEmitter.prototype.removeAllListeners =
         this.removeAllListeners('removeListener');
         this._events = { __proto__: null };
         this._eventsCount = 0;
+        this[kShapeMode] = false;
         return this;
       }
 

diff --git a/lib/internal/streams/duplex.js b/lib/internal/streams/duplex.js
@@ -76,6 +76,8 @@ function Duplex(options) {
     readable: undefined,
     pipe: undefined,
     unpipe: undefined,
+    [destroyImpl.kConstruct]: undefined,
+    [destroyImpl.kDestroy]: undefined,
   };
 
   this._readableState = new Readable.ReadableState(options, this, true);

diff --git a/lib/internal/streams/readable.js b/lib/internal/streams/readable.js
@@ -326,6 +326,8 @@ function Readable(options) {
     readable: undefined,
     pipe: undefined,
     unpipe: undefined,
+    [destroyImpl.kConstruct]: undefined,
+    [destroyImpl.kDestroy]: undefined,
   };
 
   this._readableState = new ReadableState(options, this, false);

diff --git a/lib/internal/streams/writable.js b/lib/internal/streams/writable.js
@@ -388,6 +388,8 @@ function Writable(options) {
     prefinish: undefined,
     finish: undefined,
     drain: undefined,
+    [destroyImpl.kConstruct]: undefined,
+    [destroyImpl.kDestroy]: undefined,
   };
 
   this._writableState = new WritableState(options, this, false);