Problem

Design an EventEmitter class. This interface is similar (but with some differences) to the one found in Node.js or the Event Target interface of the DOM. The EventEmitter should allow for subscribing to events and emitting them.

Your EventEmitter class should have the following two methods:

  • subscribe - This method takes in two arguments: the name of an event as a string and a callback function. This callback function will later be called when the event is emitted. An event should be able to have multiple listeners for the same event. When emitting an event with multiple callbacks, each should be called in the order in which they were subscribed. An array of results should be returned. You can assume no callbacks passed to subscribe are referentially identical. The subscribe method should also return an object with an unsubscribe method that enables the user to unsubscribe. When it is called, the callback should be removed from the list of subscriptions and undefined should be returned.

  • emit - This method takes in two arguments: the name of an event as a string and an optional array of arguments that will be passed to the callback(s). If there are no callbacks subscribed to the given event, return an empty array. Otherwise, return an array of the results of all callback calls in the order they were subscribed.

Examples

Example 1

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
Input: 
actions = ["EventEmitter", "emit", "subscribe", "subscribe", "emit"], 
values = [[], ["firstEvent"], ["firstEvent", "function cb1() { return 5; }"],  ["firstEvent", "function cb1() { return 6; }"], ["firstEvent"]]
Output: [[],["emitted",[]],["subscribed"],["subscribed"],["emitted",[5,6]]]
Explanation: 
const emitter = new EventEmitter();
emitter.emit("firstEvent"); // [], no callback are subscribed yet
emitter.subscribe("firstEvent", function cb1() { return 5; });
emitter.subscribe("firstEvent", function cb2() { return 6; });
emitter.emit("firstEvent"); // [5, 6], returns the output of cb1 and cb2

Example 2

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
Input: 
actions = ["EventEmitter", "subscribe", "emit", "emit"], 
values = [[], ["firstEvent", "function cb1(...args) { return args.join(','); }"], ["firstEvent", [1,2,3]], ["firstEvent", [3,4,6]]]
Output: [[],["subscribed"],["emitted",["1,2,3"]],["emitted",["3,4,6"]]]
Explanation: Note that the emit method should be able to accept an OPTIONAL array of arguments.

const emitter = new EventEmitter();
emitter.subscribe("firstEvent, function cb1(...args) { return args.join(','); });
emitter.emit("firstEvent", [1, 2, 3]); // ["1,2,3"]
emitter.emit("firstEvent", [3, 4, 6]); // ["3,4,6"]

Example 3

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
Input: 
actions = ["EventEmitter", "subscribe", "emit", "unsubscribe", "emit"], 
values = [[], ["firstEvent", "(...args) => args.join(',')"], ["firstEvent", [1,2,3]], [0], ["firstEvent", [4,5,6]]]
Output: [[],["subscribed"],["emitted",["1,2,3"]],["unsubscribed",0],["emitted",[]]]
Explanation:
const emitter = new EventEmitter();
const sub = emitter.subscribe("firstEvent", (...args) => args.join(','));
emitter.emit("firstEvent", [1, 2, 3]); // ["1,2,3"]
sub.unsubscribe(); // undefined
emitter.emit("firstEvent", [4, 5, 6]); // [], there are no subscriptions


#### Example 4

Input: 
actions = ["EventEmitter", "subscribe", "subscribe", "unsubscribe", "emit"], 
values = [[], ["firstEvent", "x => x + 1"], ["firstEvent", "x => x + 2"], [0], ["firstEvent", [5]]]
Output: [[],["subscribed"],["subscribed"],["unsubscribed",0],["emitted",[7]]]
Explanation:
const emitter = new EventEmitter();
const sub1 = emitter.subscribe("firstEvent", x => x + 1);
const sub2 = emitter.subscribe("firstEvent", x => x + 2);
sub1.unsubscribe(); // undefined
emitter.emit("firstEvent", [5]); // [7]

Constraints

  • 1 <= actions.length <= 10
  • values.length === actions.length
  • All test cases are valid, e.g. you don’t need to handle scenarios when unsubscribing from a non-existing subscription.
  • There are only 4 different actions: EventEmitter, emit, subscribe, and unsubscribe.
  • The EventEmitter action doesn’t take any arguments.
  • The emit action takes between either 1 or 2 arguments. The first argument is the name of the event we want to emit, and the 2nd argument is passed to the callback functions.
  • The subscribe action takes 2 arguments, where the first one is the event name and the second is the callback function.
  • The unsubscribe action takes one argument, which is the 0-indexed order of the subscription made before.

Solution

Method 1 – Map of Event Listeners with Unsubscribe

Intuition

We use a map to store event names and their corresponding arrays of listeners. Each call to subscribe adds a listener and returns an object with an unsubscribe method that removes the listener. Emitting an event calls all listeners in order and returns their results.

Approach

  1. Maintain a map from event names to arrays of listener functions.
  2. subscribe(event, cb):
    • Add the callback to the event’s array.
    • Return an object with an unsubscribe method that removes this callback from the array.
  3. emit(event, args):
    • If the event has listeners, call each with the provided arguments and collect results.
    • Return the array of results.

Code

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class EventEmitter {
    constructor() {
        this.events = new Map();
    }
    subscribe(event, cb) {
        if (!this.events.has(event)) this.events.set(event, []);
        const listeners = this.events.get(event);
        listeners.push(cb);
        const idx = listeners.length - 1;
        let unsubscribed = false;
        return {
            unsubscribe: () => {
                if (!unsubscribed) {
                    listeners[idx] = null;
                    unsubscribed = true;
                }
                return undefined;
            }
        };
    }
    emit(event, args = []) {
        if (!this.events.has(event)) return [];
        const listeners = this.events.get(event);
        const res = [];
        for (let i = 0; i < listeners.length; ++i) {
            const fn = listeners[i];
            if (fn) res.push(fn(...args));
        }
        return res;
    }
}
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class EventEmitter {
    private events: Map<string, ((...args: any[]) => any | null)[]> = new Map();
    subscribe(event: string, cb: (...args: any[]) => any) {
        if (!this.events.has(event)) this.events.set(event, []);
        const listeners = this.events.get(event)!;
        listeners.push(cb);
        const idx = listeners.length - 1;
        let unsubscribed = false;
        return {
            unsubscribe: () => {
                if (!unsubscribed) {
                    listeners[idx] = null;
                    unsubscribed = true;
                }
                return undefined;
            }
        };
    }
    emit(event: string, args: any[] = []): any[] {
        if (!this.events.has(event)) return [];
        const listeners = this.events.get(event)!;
        const res: any[] = [];
        for (let i = 0; i < listeners.length; ++i) {
            const fn = listeners[i];
            if (fn) res.push(fn(...args));
        }
        return res;
    }
}

Complexity

  • ⏰ Time complexity: O(k) per emit, where k is the number of listeners for the event.
  • 🧺 Space complexity: O(n) for the total number of listeners.