|
| 1 | +// 1. 事件监听机制 |
| 2 | +function createEventBus() { |
| 3 | + this.listeners = new Map() |
| 4 | +} |
| 5 | + |
| 6 | +// 只有_内部函数才能操作 listeners |
| 7 | +// 在内部用于添加回调函数的工具函数 |
| 8 | +// 通过单调递增的 id 确保不会因为删除而有重复 id |
| 9 | +createEventBus.prototype._add = (event, handler, once) => { |
| 10 | + if (!this.listeners.has(event)) { // 初始化 |
| 11 | + this.listeners.set(event, [0]) |
| 12 | + } |
| 13 | + const pre = this.listeners.get(event) |
| 14 | + const newId = pre[0] + 1 |
| 15 | + const curObj = { |
| 16 | + id: newId, |
| 17 | + handler: () => handler(), |
| 18 | + flag: once // 是否只执行一次就移除 |
| 19 | + } |
| 20 | + const new = [...pre, curObj] |
| 21 | + new [0] = newId |
| 22 | + this.listeners.set(event, new) |
| 23 | + return newId // 返回 newId 用于删除 |
| 24 | +} |
| 25 | + |
| 26 | +// 在内部用于删除回调函数的工具函数 |
| 27 | +createEventBus.prototype._delete = (event, id) => { |
| 28 | + const cur = this.listeners.get(event) |
| 29 | + if (cur[0] < id) return false // 如果最大的 id 都比当前输入的小,那么肯定输错了 |
| 30 | + let targetIdx = -1 |
| 31 | + for (let i = 0; i < cur.length; i++) { |
| 32 | + if (cur[i].id === id) { |
| 33 | + targetIdx = i |
| 34 | + break |
| 35 | + } |
| 36 | + } |
| 37 | + if (targetIdx === -1) return false |
| 38 | + const new = cur.splice(targetIdx, 1) |
| 39 | + this.listeners.set(event, new) |
| 40 | + return true // 删除成功 |
| 41 | +} |
| 42 | + |
| 43 | +createEventBus.prototype.on = function (event, handler) { |
| 44 | + if (!handler) return false // 特殊情况处理,防止执行空导致报错 |
| 45 | + const newId = this._add(event, handler, false) |
| 46 | + return () => this._delete(event, newId) |
| 47 | +} |
| 48 | + |
| 49 | +createEventBus.prototype.once = function (event, handler) { |
| 50 | + if (!handler) return false |
| 51 | + const newId = this._add(event, handler, true) |
| 52 | + return () => this._delete(event, newId) |
| 53 | +} |
| 54 | + |
| 55 | +createEventBus.prototype.emit = function (event, ...args) { |
| 56 | + const arr = this.listeners.get(event) |
| 57 | + for (const item of arr) { |
| 58 | + item.handler(args) |
| 59 | + if (item.once) { |
| 60 | + this._delete(event, item.id) |
| 61 | + } |
| 62 | + } |
| 63 | +} |
| 64 | +// 思路:内部通过listeners映射维护监听者数组,并且 |
| 65 | +// 只有_内部函数才能操作 listeners |
| 66 | +// 在内部用于添加回调函数的工具函数 |
| 67 | +// 通过单调递增的 id 确保不会因为删除而有重复 id |
0 commit comments