JS实现Event(event bus)

原文参考： https://www.cxymsg.com/guide/jsWritten.html#%E5%AE%9E%E7%8E%B0event-event-bus

class EventEmitter {
  constructor() {
    this._event = this._event || new Map() // 储存事件
    this._maxListeners = this._maxListeners || 10 // 设置监听上限
  }
}

EventEmitter.prototype.emit = function (type, ...args) {
  let handler = null
  handler = this._event.get(type)

  if (Array.isArray(handler)) {
    // 有多个监听者,需要依次触发
    for (let i = 0; i < handler.length; i++) {
      if (args.length > 0) {
        handler[i].apply(this, args)
      } else {
        handler[i].call(this)
      }
    }
  } else if (handler && typeof handler === 'function') {
    if (args.length > 0) {
      handler.apply(this, args)
    } else {
      handler.call(this)
    }
  }

  return true
}

EventEmitter.prototype.addListener =  function (type, fn) {
  let handler = this._event.get(type)

  if (!handler) {
    this._event.set(type, fn)
  } else if (handler && typeof handler === 'function') {
    // 如果handler是函数，说明目前已经存在一个监听者
    this._event.set(type, [handler, fn])
  } else {
    // 已经有多个监听者，直接push
    handler.push(fn)
  }
}

EventEmitter.prototype.removeListener = function (type, fn) {
  let handler = this._event.get(type)

  if (handler && typeof handler === 'function') {
    // 只有一个监听者，直接删除
    this._event.delete(type, fn)
  } else if (Array.isArray(handler)) {
    // 是数组，说明被监听多次，要找到对应的函数
    let position = -1

    for(let i = 0; i < handler.length; i++) {
      if (handler[i] === fn) {
        position = i
      }
    }

    // 如果匹配，从数组中移除
    if (position !== -1) {
      handler.splice(position, 1)

      // 移除后，如果监听只剩一个，那么取消数组，以函数形式保存
      if (handler.length === 1) {
        this._event.set(type, handler[0])
      }
    } else {
      return this
    }
  }
}