@blackglory/structures 0.5.1 → 0.6.0

package/dist/es2015/index.umd.js

@@ -32,25 +32,32 @@
         get [Symbol.toStringTag]() {
             return this.constructor.name;
         }
-        /**
-         * The same handler will only be registered once.
-         */
-        on(event, handler) {
+        on(event, listener) {
             if (!this.map.has(event)) {
                 this.map.set(event, new Set());
             }
             const set = this.map.get(event);
             set.add(handler);
-        }
-        off(event, handler) {
-            if (!this.map.has(event))
-                return;
-            const handlers = this.map.get(event);
-            handlers.delete(handler);
-            if (handlers.size === 0) {
-                this.map.delete(event);
+            return () => {
+                if (!this.map.has(event))
+                    return;
+                const handlers = this.map.get(event);
+                handlers.delete(handler);
+                if (handlers.size === 0) {
+                    this.map.delete(event);
+                }
+            };
+            function handler(...args) {
+                listener(...args);
             }
         }
+        once(event, listener) {
+            const removeListener = this.on(event, (...args) => {
+                listener(...args);
+                removeListener();
+            });
+            return removeListener;
+        }
         emit(event, ...args) {
             if (!this.map.has(event))
                 return;